Update agent.py

agent.py

@@ -136,7 +136,11 @@ def build_graph(provider: str = "openai"):
     # Node
     def assistant(state: MessagesState):
         """Assistant node"""
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        # Add system message at the beginning of messages
+        messages = [sys_msg] + state["messages"]
+        response = llm_with_tools.invoke(messages)
+        # Return the response as is
+        return {"messages": state["messages"] + [response]}
 
     builder = StateGraph(MessagesState)
     builder.add_node("assistant", assistant)