Revert c1f87282cf594871f2100e200934ea2113f4e139

CustomAgent.py

@@ -6,27 +6,6 @@ from BasicAgent import BasicAgent
 import os
 
 
-def websearch(query: str) -> str:
-        """
-        Performs a web search using an external web search API. Note that better results may be obtained from varying the wording of the query.
-        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__()
@@ -40,7 +19,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.7, 
@@ -56,7 +35,7 @@ Question: {question}
         @prompt_chain(
             template=self._system_prompt,
             model=self._model,
-            functions=[websearch],
+            functions=[self.websearch],
         )
         def _magic(question: str) -> FunctionCall | str:
             """
@@ -67,3 +46,24 @@ 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. Note that better results may be obtained from varying the wording of the query.
+        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}")

app.py

@@ -21,7 +21,7 @@ Note: Once clicking on the "submit button, it can take quite some time (this is
 ---
 **Implementation:**
 
-CustomAgent is a subclass of BasicAgent that uses a custom system prompt and integrates with an OpenAI-compatible model (llama-3.3-70b-versatile via Groq API). It leverages the magentic library for prompt chaining and function calling, allowing the agent to use external tools such as web search.
+CustomAgent is a subclass of BasicAgent that uses a custom system prompt and integrates with an OpenAI-compatible model (llama-3.1-8b-instant via Groq API). It leverages the magentic library for prompt chaining and function calling, allowing the agent to use external tools such as web search.
 """
     )
     gr.LoginButton()