Update agent.py

agent.py

@@ -189,16 +189,32 @@ def build_graph(provider: str = "huggingface"):
     from langchain_core.messages import AIMessage
 
     def retriever(state: MessagesState):
-        query = state["messages"][-1].content
-        similar_doc = vector_store.similarity_search(query, k=1)[0]
-
-        content = similar_doc.page_content
+    query = state["messages"][-1].content
+    # 1. Embed the query to vector
+    query_embedding = embeddings.embed_query(query)  # list of floats
+
+    # 2. Call the RPC function directly
+    response = supabase.rpc(
+        'match_documents_langchain',
+        {
+            'match_count': 2,
+            'query_embedding': query_embedding
+        }
+    ).execute()
+
+    docs = response.data
+    if not docs or len(docs) == 0:
+        answer = "Sorry, I couldn't find an answer to your question."
+    else:
+        content = docs[0]['content']  # get content of the first matched doc
+        # Extract answer if it has 'Final answer :' pattern
         if "Final answer :" in content:
             answer = content.split("Final answer :")[-1].strip()
         else:
             answer = content.strip()
 
-        return {"messages": [AIMessage(content=answer)]}
+    return {"messages": [AIMessage(content=answer)]}
+
 
    # builder = StateGraph(MessagesState)
     #builder.add_node("retriever", retriever)