Update agent.py

agent.py

@@ -195,27 +195,20 @@ def build_graph(provider: str = "openai"):
         response = llm_with_tools.invoke(messages)
         # Return new state with appended message
         return {"messages": messages + [response]}
-    
     def retriever(state: MessagesState):
         """Retriever node - provides context from vector store"""
-        # Get current messages
         messages = state["messages"]
-        # Last message is the user query
         query = messages[-1].content
         
         # Retrieve similar documents
         similar_docs = vector_store.similarity_search(query, k=1)
         
         if not similar_docs:
-            # Return original messages if no context found
             return {"messages": messages}
         
-        # Get context from first document
         context = similar_docs[0].page_content
-        # Create system message with context
         context_msg = SystemMessage(content=f"Reference context:\n{context}")
         
-        # Append context to messages
         return {"messages": messages + [context_msg]}
 
     # Build graph
@@ -235,18 +228,68 @@ def build_graph(provider: str = "openai"):
         "assistant",
         tools_condition,
         {
-            # Continue to tools if needed
-            "continue": "tools",
-            # End conversation if no tools needed
-            "end": END
+            "continue": "tools",  # If tools needed
+            "end": END            # If no tools needed
         }
     )
     
-    # After tools execute, return to assistant for response generation
+    # After tools execute, go back to assistant
     builder.add_edge("tools", "assistant")
     
-    # builder.add_finish_point(END)  # Explicitly declare END as finish point
+    # Compile graph
     return builder.compile()
+    
+    # def retriever(state: MessagesState):
+    #     """Retriever node - provides context from vector store"""
+    #     # Get current messages
+    #     messages = state["messages"]
+    #     # Last message is the user query
+    #     query = messages[-1].content
+        
+    #     # Retrieve similar documents
+    #     similar_docs = vector_store.similarity_search(query, k=1)
+        
+    #     if not similar_docs:
+    #         # Return original messages if no context found
+    #         return {"messages": messages}
+        
+    #     # Get context from first document
+    #     context = similar_docs[0].page_content
+    #     # Create system message with context
+    #     context_msg = SystemMessage(content=f"Reference context:\n{context}")
+        
+    #     # Append context to messages
+    #     return {"messages": messages + [context_msg]}
+
+    # # Build graph
+    # builder = StateGraph(MessagesState)
+    
+    # # Add nodes
+    # builder.add_node("retriever", retriever)
+    # builder.add_node("assistant", assistant)
+    # builder.add_node("tools", ToolNode(tools))
+    
+    # # Set up edges
+    # builder.set_entry_point("retriever")
+    # builder.add_edge("retriever", "assistant")
+    
+    # # Conditional tool usage
+    # builder.add_conditional_edges(
+    #     "assistant",
+    #     tools_condition,
+    #     {
+    #         # Continue to tools if needed
+    #         "continue": "tools",
+    #         # End conversation if no tools needed
+    #         "end": END
+    #     }
+    # )
+    
+    # # After tools execute, return to assistant for response generation
+    # builder.add_edge("tools", "assistant")
+    
+    # # builder.add_finish_point(END)  # Explicitly declare END as finish point
+    # return builder.compile()
 
 
 # def build_graph(provider: str = "openai"):