Shaukat39 commited on
Commit
ccb47f7
·
verified ·
1 Parent(s): 524ee6f

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +10 -4
agent.py CHANGED
@@ -130,6 +130,7 @@ vector_store = SupabaseVectorStore(
130
  table_name="documents",
131
  query_name="match_documents_langchain"
132
  )
 
133
 
134
  retriever_tool = create_retriever_tool(
135
  retriever=vector_store.as_retriever(),
@@ -165,10 +166,15 @@ def build_graph(provider: str = "groq"):
165
  return {"messages": [sys_msg] + state["messages"] + [example_msg]}
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)
 
130
  table_name="documents",
131
  query_name="match_documents_langchain"
132
  )
133
+ print("🔍 Retrieved vector documents:", similar)
134
 
135
  retriever_tool = create_retriever_tool(
136
  retriever=vector_store.as_retriever(),
 
166
  return {"messages": [sys_msg] + state["messages"] + [example_msg]}
167
 
168
  def assistant(state: MessagesState):
169
+ try:
170
+ result = llm_with_tools.invoke(state["messages"])
171
+ print("🤖 Raw LLM result:", repr(result))
172
+ final_response = result.content.strip()
173
+ return {"messages": [HumanMessage(content=final_response)]}
174
+ except Exception as e:
175
+ print(f"🔥 Error in assistant node: {e}")
176
+ return {"messages": [HumanMessage(content=f"AGENT ERROR: {e}")] }
177
+
178
 
179
 
180
  builder = StateGraph(MessagesState)