Update agents.py

agents.py

@@ -5,8 +5,10 @@ import requests
 import yaml
 import pandas as pd
 
+
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, WikipediaSearchTool, Tool, OpenAIServerModel, SpeechToTextTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, WikipediaSearchTool, Tool, SpeechToTextTool
+from langchain_groq import ChatGroq
 
 class GetTaskFileTool(Tool):
     name = "get_task_file_tool"
@@ -55,12 +57,18 @@ prompts["system_prompt"] = ("You are a general AI assistant. I will ask you a qu
                             + prompts["system_prompt"])
 
 def agentRudra():
-    gemini_model = OpenAIServerModel(
-            model_id="gemini-2.0-flash",
-            api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
-            api_key=os.getenv("API_KEY"),
-            temperature=0.7
+    # gemini_model = OpenAIServerModel(
+    #         model_id="gemini-2.0-flash",
+    #         api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+    #         api_key=os.getenv("API_KEY"),
+    #         temperature=0.7
+    # )
+    model = ChatGroq(
+    model_name="llama3-8b-8192",
+    api_key=groq_api_key,
+    temperature=0
     )
+    
     agent = CodeAgent(
         tools=[
             DuckDuckGoSearchTool(),
@@ -71,7 +79,7 @@ def agentRudra():
             LoadXlsxFileTool(),
             LoadTextFileTool()
         ],
-        model=gemini_model,
+        model=model,
         prompt_templates=prompts,
         max_steps=15,
         additional_authorized_imports = ["pandas"]