Shaukat39 commited on
Commit
dac6c57
·
verified ·
1 Parent(s): fa1833e

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -166,10 +166,11 @@ def build_graph(provider: str = "groq"):
166
 
167
  def assistant(state: MessagesState):
168
  result = llm_with_tools.invoke(state["messages"])
169
- # Extract only the raw string answer
170
  final_response = result.content
171
  return {"messages": [HumanMessage(content=final_response.strip())]}
172
 
 
173
  builder = StateGraph(MessagesState)
174
  builder.add_node("retriever", retriever)
175
  builder.add_node("assistant", assistant)
 
166
 
167
  def assistant(state: MessagesState):
168
  result = llm_with_tools.invoke(state["messages"])
169
+ print("🤖 LLM result content →", repr(result.content))
170
  final_response = result.content
171
  return {"messages": [HumanMessage(content=final_response.strip())]}
172
 
173
+
174
  builder = StateGraph(MessagesState)
175
  builder.add_node("retriever", retriever)
176
  builder.add_node("assistant", assistant)