Update agent.py

agent.py

@@ -226,6 +226,24 @@ execute_code_multilang_tool = Tool(
 ### ======================================== DOCUMENT PROCESSING TOOLS ======================================== ###
 
 
+### ======================================== RETRIEVER TOOLS ======================================== ###
+
+embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
+supabase: Client = create_client(
+    os.environ.get("SUPABASE_URL"), 
+    os.environ.get("SUPABASE_SERVICE_KEY"))
+vector_store = SupabaseVectorStore(
+    client=supabase,
+    embedding= embeddings,
+    table_name="documents",
+    query_name="match_documents_langchain",
+)
+create_retriever_tool = create_retriever_tool(
+    retriever=vector_store.as_retriever(),
+    name="Question Search",
+    description="A tool to retrieve similar questions from a vector store.",
+)
+
 
 # load the system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
@@ -293,14 +311,6 @@ def build_graph(provider: str = "huggingface"):
     def assistant(state: MessagesState):
         """Assistant node"""
         return {"messages": [chat_with_tools.invoke(state["messages"])]}
-    
-    # def retriever(state: MessagesState):
-      #  """Retriever node"""
-       # similar_question = vector_store.similarity_search(state["messages"][0].content)
-        #example_msg = HumanMessage(
-         #   content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
-       # )
-       # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     from langchain_core.messages import AIMessage
 
@@ -316,30 +326,34 @@ def build_graph(provider: str = "huggingface"):
 
         return {"messages": [AIMessage(content=answer)]}
 
-   # builder = StateGraph(MessagesState)
-    #builder.add_node("retriever", retriever)
-    #builder.add_node("assistant", assistant)
-    #builder.add_node("tools", ToolNode(tools))
-    #builder.add_edge(START, "retriever")
-    #builder.add_edge("retriever", "assistant")
-    #builder.add_conditional_edges(
-     #   "assistant",
-      #  tools_condition,
-    #)
+    """
+    Graph with retriever and tools
+    
+    builder = StateGraph(MessagesState)
+    builder.add_node("retriever", retriever)
+    builder.add_node("assistant", assistant)
+    builder.add_node("tools", ToolNode(tools))
+    builder.add_edge(START, "retriever")
+    builder.add_edge("retriever", "assistant")
+    builder.add_conditional_edges(
+        "assistant",
+        tools_condition,
+    )
     #builder.add_edge("tools", "assistant")
+    """
 
-    #builder = StateGraph(MessagesState)
-    #builder.add_node("retriever", retriever)
-
+    builder = StateGraph(MessagesState)
+    builder.add_node("retriever", retriever)
     # Retriever ist Start und Endpunkt
-    #builder.set_entry_point("retriever")
-    #builder.set_finish_point("retriever")
-
+    builder.set_entry_point("retriever")
+    builder.set_finish_point("retriever")
     # Compile graph
-    #return builder.compile()
-
+    return builder.compile()
 
 
+    """
+    Graph with tools conditions
+    
     builder = StateGraph(MessagesState)
     # Define nodes: these do the work
     builder.add_node("assistant", assistant)
@@ -353,6 +367,6 @@ def build_graph(provider: str = "huggingface"):
         tools_condition,
     )
     builder.add_edge("tools", "assistant")
-
     # Compile graph
-    return builder.compile()
+    return builder.compile()
+    """