Upload 6 files

agent.py

@@ -56,8 +56,7 @@ def _select_model():
         hf_token = os.getenv("HF_API_KEY")
         return InferenceClientModel(
             model_id=hf_model_id,
-            token=hf_token,
-            system_prompt=SYSTEM_PROMPT
+            token=hf_token
         )
 
     if provider == "openai":
@@ -66,8 +65,7 @@ def _select_model():
         openai_token = os.getenv("OPENAI_API_KEY")
         return OpenAIServerModel(
             model_id=openai_model_id,
-            api_key=openai_token,
-            system_prompt=SYSTEM_PROMPT
+            api_key=openai_token
         )
 
     raise ValueError(
@@ -89,8 +87,12 @@ DEFAULT_TOOLS = [
 ]
 
 class GAIAAgent(CodeAgent):
-    def __init__(self, tools=None):
-        super().__init__(tools=tools or DEFAULT_TOOLS, model=_select_model())
+    def __init__(self, tools=None, *, system_prompt: str = SYSTEM_PROMPT):
+        super().__init__(
+            tools=tools or DEFAULT_TOOLS,
+            model=_select_model(),
+            system_prompt=system_prompt
+        )
 
     # Convenience so the object itself can be *called* directly
     def __call__(self, question: str, **kwargs: Any) -> str: