Update agent.py

agent.py

@@ -3,6 +3,7 @@ from dotenv import load_dotenv
 from typing import TypedDict, List, Dict, Any, Optional
 from langchain.agents import create_tool_calling_agent, AgentExecutor, initialize_agent
 from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_groq import ChatGroq
 from langchain_core.tools import tool
 from langchain_core.messages import HumanMessage
 from langchain_core.prompts import ChatPromptTemplate
@@ -230,13 +231,22 @@ def divide(a: float, b: float) -> float:
 # ("human", f"Question: {question}\nReport to validate: {final_answer}")
 class BasicAgent:
     def __init__(self):
-        self.model = ChatGoogleGenerativeAI(
-            model="gemini-2.0-flash-lite",
+        # self.model = ChatGoogleGenerativeAI(
+        #     model="gemini-2.0-flash-lite",
+        #     temperature=0,
+        #     max_tokens=128,
+        #     timeout=None,
+        #     max_retries=2,
+        #     google_api_key=os.getenv("GEMINI_API_KEY"),
+        #     # other params...
+        # )
+        self.model = ChatGroq(
+            model="qwen-qwq-32b",
             temperature=0,
             max_tokens=128,
             timeout=None,
             max_retries=2,
-            google_api_key=os.getenv("GEMINI_API_KEY"),
+            groq_api_key=os.getenv("GROQ_API_KEY")
             # other params...
         )
         # System Prompt for few shot prompting
@@ -264,7 +274,7 @@ class BasicAgent:
                 
                 If Task ID is included in the question, remember to call the relevant read tools [ie. read_file, excel_read, csv_read, mp3_listen, image_caption]
         """
-        self.tools = [serper_websearch, duckduck_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, python_tool]
+        self.tools = [duckduck_websearch, serper_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, python_tool]
         self.prompt = ChatPromptTemplate.from_messages([
             ("system", self.sys_prompt),
             ("human", "{input}")