Spaces:
Sleeping
Sleeping
Charles Grandjean
commited on
Commit
Β·
08e396b
1
Parent(s):
1ec65d4
correct logger
Browse files- langraph_agent.py +2 -2
langraph_agent.py
CHANGED
|
@@ -107,7 +107,7 @@ class CyberLegalAgent:
|
|
| 107 |
history = state.get("conversation_history", [])
|
| 108 |
has_system_message = False
|
| 109 |
|
| 110 |
-
logger.
|
| 111 |
for msg in history:
|
| 112 |
if msg["role"] == "system":
|
| 113 |
messages.append(SystemMessage(content=msg["content"]))
|
|
@@ -122,7 +122,7 @@ class CyberLegalAgent:
|
|
| 122 |
answer_prompt = RESPONSE_FORMATTING_PROMPT.format(context=context, query=state["user_query"])
|
| 123 |
messages.append(HumanMessage(content=answer_prompt))
|
| 124 |
|
| 125 |
-
logger.
|
| 126 |
response = await self.llm.ainvoke(messages)
|
| 127 |
answer = response.content
|
| 128 |
state["final_response"] = answer
|
|
|
|
| 107 |
history = state.get("conversation_history", [])
|
| 108 |
has_system_message = False
|
| 109 |
|
| 110 |
+
logger.info(f"π Messages {messages}")
|
| 111 |
for msg in history:
|
| 112 |
if msg["role"] == "system":
|
| 113 |
messages.append(SystemMessage(content=msg["content"]))
|
|
|
|
| 122 |
answer_prompt = RESPONSE_FORMATTING_PROMPT.format(context=context, query=state["user_query"])
|
| 123 |
messages.append(HumanMessage(content=answer_prompt))
|
| 124 |
|
| 125 |
+
logger.info(f"π Messages after adding last one : {messages}")
|
| 126 |
response = await self.llm.ainvoke(messages)
|
| 127 |
answer = response.content
|
| 128 |
state["final_response"] = answer
|