From 6de92100a73821b0d0503dde3d748cc41674b3cc Mon Sep 17 00:00:00 2001 From: jsdio Date: Sun, 15 Oct 2023 09:01:33 +0300 Subject: [PATCH] Changes by ijaric --- src/assistant/lib/agent/services.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/assistant/lib/agent/services.py b/src/assistant/lib/agent/services.py index f23b715..66dcf70 100644 --- a/src/assistant/lib/agent/services.py +++ b/src/assistant/lib/agent/services.py @@ -73,13 +73,12 @@ class AgentService: request_chat_history = models.RequestChatHistory(session_id=session_id) chat_history_source = await self.chat_repository.get_messages_by_sid(request_chat_history) - chat_history.append(langchain.schema.HumanMessage(content="Hi there!")) for entry in chat_history_source: print("ENTRY: ", entry) - if entry.content["role"] == "user": - chat_history.append(langchain.schema.HumanMessage(content=entry.content["content"])) - elif entry.content["role"] == "agent": - chat_history.append(langchain.schema.AIMessage(content=entry.content["content"])) + if entry.role == "user": + chat_history.append(langchain.schema.HumanMessage(content=entry.content)) + elif entry.role == "agent": + chat_history.append(langchain.schema.AIMessage(content=entry.content)) # memory = langchain.memory.ConversationBufferMemory(memory_key=chat_history_name,chat_memory=chat_history) @@ -118,8 +117,10 @@ class AgentService: agent_executor = langchain.agents.AgentExecutor(agent=agent, tools=tools, verbose=True) print("CH:", type(chat_history), chat_history) + chat_history = [] # temporary disable chat_history response = await agent_executor.ainvoke({"input": request.text, "chat_history": chat_history}) print("AI RESPONSE:", response) + user_request = models.RequestChatMessage( session_id=session_id, user_id=request.user_id,