Patch fb8b40665e5fb277f1d15cbe94df1856b1a62cad

CustomAgent.py

@@ -6,11 +6,32 @@ from BasicAgent import BasicAgent
 import os
 
 
+def websearch(query: str) -> str:
+        """
+        Performs a web search using an external web search API.
+        Args:
+            query (str): The search query string.
+        Returns:
+            str: The response from the web search API as a string.
+        """
+        try:
+            r = requests.post(
+                f"{os.environ['WEBSEARCH_API_URL']}/tools/websearch",
+                json={"query": query}
+            )
+            r.raise_for_status()
+            response = r.text
+        except Exception as e:
+            print("Error:", e)
+            response = "An error has occurred while using websearch, please inform the user of this"
+        print(f"Agent searched for {query}. Response: {response}")
+
+
 class CustomAgent(BasicAgent):
     def __init__(self):
         super().__init__()
         self._system_prompt = """\
-You are a general AI assistant. I will ask you a question. Report your thoughts and provide an answer using, if appropriate, the tools at your disposal.
+You are a general AI assistant. I will ask you a question. Report your thoughts and provide an answer.
 Your final answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. Do not provide anything except your final answer.
 If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
 If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
@@ -18,7 +39,7 @@ If you are asked for a comma separated list, apply the above rules depending of
 Question: {question}
 """ # System prompt, adapted from the GAIA team's example at https://huggingface.co/spaces/gaia-benchmark/leaderboard
         self._model = OpenaiChatModel(
-            model="llama-3.3-70b-versatile",
+            model="llama-3.1-8b-instant",
             api_key=os.environ["GROQ_API_KEY"],
             base_url="https://api.groq.com/openai/v1/", 
             temperature=0.2, 
@@ -30,11 +51,11 @@ Question: {question}
         """
         this is where the magic happens
         """
-        super().__call__(question)
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
         @prompt_chain(
             template=self._system_prompt,
             model=self._model,
-            functions=[self.websearch],
+            functions=[websearch],
         )
         def _magic(question: str) -> FunctionCall | str:
             """
@@ -45,24 +66,3 @@ Question: {question}
         result = _magic(question)
         print("Agent returning actual answer:", result)
         return result
-    
-    @staticmethod
-    def websearch(query: str) -> str:
-        """
-        Performs a web search using an external web search API.
-        Args:
-            query (str): The search query string.
-        Returns:
-            str: The response from the web search API as a string.
-        """
-        try:
-            r = requests.post(
-                f"{os.environ['WEBSEARCH_API_URL']}/tools/websearch",
-                json={"query": query}
-            )
-            r.raise_for_status()
-            response = r.text
-        except Exception as e:
-            print("Error:", e)
-            response = "An error has occurred while using websearch, please inform the user of this"
-        print(f"Agent searched for {query}. Response: {response}")