Update agent.py

agent.py

@@ -50,20 +50,13 @@ def build_graph(provider: str = "google"):
     # -------------------------------
     # LLM SELECTION
     # -------------------------------
-    if provider == "google":
-        llm = ChatGoogleGenerativeAI(
-            model="gemini-2.0-flash",
-            temperature=0
+    llm = ChatHuggingFace(
+        llm=HuggingFaceEndpoint(
+            repo_id="HuggingFaceH4/zephyr-7b-beta",
+            temperature=0,
+            max_new_tokens=512,
         )
-    elif provider == "groq":
-        llm = ChatGroq(
-            model="qwen-qwq-32b",
-            temperature=0
-        )
-    else:
-        raise ValueError("Invalid provider. Use 'google' or 'groq'.")
-
-    llm_with_tools = llm.bind_tools(TOOLS)
+    )
 
     # -------------------------------
     # ASSISTANT NODE