Update agent.py

agent.py

@@ -728,7 +728,7 @@ def build_graph(provider: str = "huggingface"):
     # Node
     def assistant(state: MessagesState):
         """Assistant node"""      
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
         
     """
     def retriever(state: MessagesState):
@@ -746,11 +746,12 @@ def build_graph(provider: str = "huggingface"):
     
 
     builder = StateGraph(MessagesState)
-    builder.add_node("retriever", retriever)
+    #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_edge(START, "retriever")
+    builder.add_edge(START, "assistamt")
+    #builder.add_edge("retriever", "assistant")
     builder.add_conditional_edges(
         "assistant",
         tools_condition,