Update agent.py

agent.py

@@ -193,7 +193,25 @@ tools_list.append(
     )
 )
 
-llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+# ----------------------------------------------------------
+# provider switcher
+# ----------------------------------------------------------
+def build_llm(provider: str = "groq"):
+    if provider == "google":
+        return ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    elif provider == "groq":
+        return ChatGroq(model="llama-3.3-70b-versatile", temperature=0)
+    elif provider == "huggingface":
+        return ChatHuggingFace(
+            llm=HuggingFaceEndpoint(
+                repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+                temperature=0,
+            )
+        )
+    else:
+        raise ValueError("provider must be 'google', 'groq', or 'huggingface'")
+
+llm = build_llm("google")  # or "groq", "huggingface"
 llm_with_tools = llm.bind_tools(tools_list)
 
 def assistant(state: MessagesState):