kaeizen commited on
Commit
3cb9f47
·
1 Parent(s): 4adb3e1
Files changed (2) hide show
  1. agent_manager/__init__.py +1 -0
  2. api/views.py +1 -0
agent_manager/__init__.py CHANGED
@@ -146,6 +146,7 @@ STRUCTURED_CHAT = StructuredChatWrapper(CHAT)
146
  SESSION_AGENTS = {}
147
 
148
  def set_session_agent(session_key):
 
149
  memory = InMemorySaver()
150
  agent = create_agent(
151
  model=STRUCTURED_CHAT,
 
146
  SESSION_AGENTS = {}
147
 
148
  def set_session_agent(session_key):
149
+ print(f"New session created: {session_key}")
150
  memory = InMemorySaver()
151
  agent = create_agent(
152
  model=STRUCTURED_CHAT,
api/views.py CHANGED
@@ -31,6 +31,7 @@ def chat(request):
31
  tone = request.data.get("tone")
32
  messages = get_message_list(mode, tone, message)
33
 
 
34
  result = agent.invoke({ "messages": messages },
35
  config={ "configurable": {"thread_id": request.session.session_key } }
36
  )
 
31
  tone = request.data.get("tone")
32
  messages = get_message_list(mode, tone, message)
33
 
34
+ print("Message:", message, "Session Key:", request.session.session_key)
35
  result = agent.invoke({ "messages": messages },
36
  config={ "configurable": {"thread_id": request.session.session_key } }
37
  )