Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
…panion into dev
  • Loading branch information
dhrumilp12 committed Jun 15, 2024
2 parents 88da73b + 944baff commit ea4eb6d
Showing 1 changed file with 9 additions and 6 deletions.
15 changes: 9 additions & 6 deletions server/agents/mental_health_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,13 +66,16 @@ def get_session_history(self, session_id: str) -> MongoDBChatMessageHistory:


def get_agent_memory(self, user_id, chat_id):
chat_history = self.get_chat_history(user_id, chat_id)
# chat_history = self.get_chat_history(user_id, chat_id)

memory = ConversationSummaryMemory.from_messages(
llm=self.llm,
chat_memory=chat_history,
return_messages=True
)
# memory = ConversationSummaryMemory.from_messages(
# llm=self.llm,
# chat_memory=chat_history,
# return_messages=True
# )

# TODO: Rewrite function or remove if not used anymore
memory = None

return memory

Expand Down

0 comments on commit ea4eb6d

Please sign in to comment.