add logs
Browse files- agent_manager/__init__.py +1 -0
- 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 |
)
|