Update agents.py

agents.py

@@ -5,7 +5,6 @@ import requests
 import yaml
 import pandas as pd
 
-from types import SimpleNamespace  # NEW
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 groq_api_key = os.getenv("GROQ_API_KEY")
@@ -13,19 +12,6 @@ groq_api_key = os.getenv("GROQ_API_KEY")
 from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, WikipediaSearchTool, Tool, SpeechToTextTool
 from langchain_groq import ChatGroq
 
-# === WRAPPER CLASS TO FIX 'content' ERROR ===
-from langchain_core.messages import AIMessage
-
-class WrappedChatGroq(ChatGroq):
-    def invoke(self, *args, **kwargs):
-        result = super().invoke(*args, **kwargs)
-        if isinstance(result, dict) and "content" in result:
-            return AIMessage(content=result["content"])
-        return result
-
-    def __call__(self, *args, **kwargs):
-        return self.invoke(*args, **kwargs)
-
 
 # === TOOLS ===
 class GetTaskFileTool(Tool):
@@ -81,7 +67,7 @@ prompts["system_prompt"] = (
 
 # === AGENT FUNCTION ===
 def agentRudra():
-    model = WrappedChatGroq(  # << use the wrapper instead of original ChatGroq
+    model = ChatGroq(
         model_name="llama3-8b-8192",
         api_key=groq_api_key,
         temperature=0