Upload agent.py

agent.py

@@ -290,7 +290,8 @@ def build_graph(provider: str = "anthropic"):
         # Google Gemini
         llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     elif provider == "groq":
-        llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
+        # Groq https://console.groq.com/docs/models
+            llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
         # TODO: Add huggingface endpoint
         llm = ChatHuggingFace(
@@ -306,9 +307,8 @@ def build_graph(provider: str = "anthropic"):
 
     # Node
     def assistant(state: MessagesState):
-        messages_with_sys = [sys_msg] + state["messages"]
-        return {"messages": [llm_with_tools.invoke(messages_with_sys)]}
-
+        """Assistant node"""
+        return {"messages": [llm_with_tools.invoke(state["messages"])]}
     
     def retriever(state: MessagesState):
         """Retriever node"""