File size: 816 Bytes
fd2b77d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 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 "" |