Update agent.py

agent.py

@@ -154,13 +154,13 @@ tools = [
 def build_graph(provider: str = "groq"):
     """Build the graph"""
     # Load environment variables from .env file
-    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-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
-    elif provider == "openai":
+    # 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-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
+    if provider == "openai":
         llm = ChatOpenAI(model="gpt-4",temperature=0)
     elif provider == "huggingface":
         # TODO: Add huggingface endpoint