Update agent.py

agent.py

@@ -706,27 +706,25 @@ tools = [
 
 
 # Build graph function
-def build_graph(provider: str = "groq"):
+def build_graph(provider: str = "google"):
     """Build the graph"""
     # Load environment variables from .env file
-    if provider == "groq":
+    if provider == "google":
+        # Google Gemini
+        llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    elif provider == "groq":
         # Groq https://console.groq.com/docs/models
-        llm = ChatGroq(model="qwen/qwen3-32b", temperature=0)
+        llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
         # TODO: Add huggingface endpoint
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(
-                repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
-                task="text-generation",  # for chat‐style use “text-generation”
-                max_new_tokens=1024,
-                do_sample=False,
-                repetition_penalty=1.03,
+                url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
                 temperature=0,
             ),
-            verbose=True,
         )
     else:
-        raise ValueError("Invalid provider. Choose 'groq' or 'huggingface'.")
+        raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     # Bind tools to LLM
     llm_with_tools = llm.bind_tools(tools)