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