Spaces:
Sleeping
Sleeping
Charles Grandjean
commited on
Commit
·
6bde5c1
1
Parent(s):
8ef630e
adjust namings
Browse files- langraph_agent.py +1 -1
langraph_agent.py
CHANGED
|
@@ -130,7 +130,7 @@ class CyberLegalAgent:
|
|
| 130 |
logger.info(f"📝 Created Messages stack: {messages}")
|
| 131 |
if not has_system_message:
|
| 132 |
messages.insert(0, SystemMessage(content=SYSTEM_PROMPT))
|
| 133 |
-
answer_prompt = RESPONSE_FORMATTING_PROMPT.format(
|
| 134 |
messages.append(HumanMessage(content=answer_prompt))
|
| 135 |
|
| 136 |
response = await self.llm.ainvoke(messages)
|
|
|
|
| 130 |
logger.info(f"📝 Created Messages stack: {messages}")
|
| 131 |
if not has_system_message:
|
| 132 |
messages.insert(0, SystemMessage(content=SYSTEM_PROMPT))
|
| 133 |
+
answer_prompt = RESPONSE_FORMATTING_PROMPT.format(context=context, user_query=state["user_query"])
|
| 134 |
messages.append(HumanMessage(content=answer_prompt))
|
| 135 |
|
| 136 |
response = await self.llm.ainvoke(messages)
|