Implement CustomAgent, using magentic

CustomAgent.py

@@ -1,7 +1,59 @@
+from magentic import FunctionCall, OpenaiChatModel, prompt_chain
+import requests
+
 from BasicAgent import BasicAgent
 
+import os
+
 
 class CustomAgent(BasicAgent):
     def __init__(self):
-        self._fixed_answer = "This is a custom default answer."
+        self._system_prompt = """\
+You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer like so: [YOUR FINAL ANSWER].
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+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.
+If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+""" # 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",
+            api_key=os.environ["GROQ_API_KEY"],
+            base_url="https://api.groq.com/openai/v1/", 
+            temperature=0.2, 
+            max_tokens=512
+        )
         print("CustomAgent initialized.")
+
+    def __call__(self, question: str) -> str:
+        """
+        this is where the magic happens
+        """
+        super().__call__(question)
+        @prompt_chain(
+            template=self._system_prompt,
+            model=self._model,
+            functions=[self._banana],
+        )
+        def _magic(question: str) -> FunctionCall | str:
+            """
+            An agent, using websearch to gather information and
+            answer a user's query.
+            """
+            ...
+        return _magic(question)
+    
+    def _banana(query: str) -> str:
+        """
+        ✨ banana ✨
+        """
+        try:
+            r = requests.post(
+                f"{os.environ["WEBSEARCH_API_URL"]}/tools/websearch",
+                json={"query": query}
+            )
+            r.raise_for_status()
+            response = r.json()
+        except Exception as e:
+            print(e)
+            response = f'{{"error": "An error has occurred while using websearch, please inform the user of this"}}'
+        print(f"Agent searched for {query}. Response: {response}")

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-requests
+magentic
+requests