Charles Grandjean commited on
Commit
08e396b
Β·
1 Parent(s): 1ec65d4

correct logger

Browse files
Files changed (1) hide show
  1. 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.messages(f"πŸ“ Messages {messages}")
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.messages(f"πŸ“ Messages after adding last one : {messages}")
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