Spaces:
Runtime error
Runtime error
Update agent.py
Browse files
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 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|