🐛 fix(OpenAIConversationalAgent.py): change parameter name from model_name to model to improve clarity and consistency

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-09 21:48:40 -03:00
commit 5d226d43d1

View file

@ -43,7 +43,7 @@ class ConversationalAgent(CustomComponent):
system_message: Optional[SystemMessagePromptTemplate] = None,
max_token_limit: int = 2000,
) -> AgentExecutor:
llm = ChatOpenAI(model_name=model_name)
llm = ChatOpenAI(model=model_name)
if not memory:
memory_key = "chat_history"
memory = ConversationTokenBufferMemory(