Charles Grandjean commited on
Commit
1ec65d4
·
1 Parent(s): e7c5fb8

messages fix

Browse files
Files changed (1) hide show
  1. 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
- logger.messages(f"📝 Messages {messages}")
 
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