Create agent.py

agent.py

@@ -0,0 +1,39 @@
+import os
+from huggingface_hub import InferenceClient
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+client = InferenceClient(
+    model="Qwen/Qwen2.5-7B-Instruct",
+    token=HF_TOKEN
+) if HF_TOKEN else None
+
+
+def agent(task):
+    """
+    Robust agent for GAIA-style evaluation.
+    Accepts dict OR string safely.
+    """
+
+    # Extract input safely
+    if isinstance(task, dict):
+        query = task.get("input") or task.get("question") or str(task)
+    else:
+        query = str(task)
+
+    if client is None:
+        return ""
+
+    try:
+        response = client.chat_completion(
+            messages=[
+                {"role": "user", "content": query}
+            ],
+            max_tokens=512,
+            temperature=0.1
+        )
+
+        return response.choices[0].message.content.strip()
+
+    except:
+        return ""