Update agent.py

agent.py

@@ -1,19 +1,14 @@
 """
 HF Agents Course — Unit 4
-GAIA Level 1 Agent (LangGraph + Qwen3.5-35B-A3B)
+GAIA Level 1 Agent (Qwen3.5-35B-A3B, safe mode)
 """
 
 import os
+import re
+import requests
 from dotenv import load_dotenv
 from langchain_openai import ChatOpenAI
-from langgraph.graph import StateGraph, END
-from langchain_core.messages import HumanMessage
-from typing import TypedDict, Annotated
-import operator
 
-# =========================
-# 基础配置
-# =========================
 load_dotenv()
 
 AGICTO_API_KEY = os.getenv("AGICTO_API_KEY")
@@ -23,55 +18,59 @@ llm = ChatOpenAI(
     model="qwen3.5-35b-a3b",
     openai_api_key=AGICTO_API_KEY,
     openai_api_base=AGICTO_BASE_URL,
-    temperature=0.1,
-    max_tokens=1024,
+    temperature=0.0,
+    max_tokens=512,
 )
 
-# =========================
-# System Prompt（GAIA 专用）
-# =========================
 SYSTEM_PROMPT = (
     "You are a GAIA benchmark agent.\n"
-    "Think step-by-step internally.\n"
-    "Do NOT explain your reasoning.\n"
-    "Output ONLY the final exact answer.\n"
-    "No units, no punctuation, no extra words.\n"
-    "If the question includes a file URL, use its content."
+    "Answer directly.\n"
+    "Do NOT explain.\n"
+    "Do NOT output reasoning.\n"
+    "Do NOT output  tags.\n"
+    "Output ONLY the final answer.\n"
+    "No punctuation. No units. No extra words."
 )
 
-# =========================
-# LangGraph State
-# =========================
-class AgentState(TypedDict):
-    messages: Annotated[list, operator.add]
+# ---------- 工具 ----------
+def download_file(url: str) -> str:
+    try:
+        r = requests.get(url, timeout=10)
+        if r.ok:
+            return r.text[:8000]
+    except Exception:
+        pass
+    return ""
 
-def build_graph():
-    def run_agent(state: AgentState) -> AgentState:
-        response = llm.invoke(state["messages"])
-        return {"messages": [response]}
+def clean_output(text: str) -> str:
+    # 去除 Qwen3 思考链
+    text = re.sub(r".*?", "", text, flags=re.S)
+    text = text.strip()
 
-    graph = StateGraph(AgentState)
-    graph.add_node("agent", run_agent)
-    graph.set_entry_point("agent")
-    graph.add_edge("agent", END)
-    return graph.compile()
+    # 只保留第一行
+    text = text.splitlines()[0]
 
-graph = build_graph()
+    # 去除所有非必要字符
+    for ch in ".,;:!?()[]{}<>\"'":
+        text = text.replace(ch, "")
 
-# =========================
-# ✅ HF 官方接口
-# =========================
+    return text.strip() or "0"
+
+# ---------- ✅ HF 官方接口 ----------
 def agent(question: str, files: list[str] | None = None) -> str:
-    """
-    必须保留此函数签名，HF Unit 4 评分器会直接调用。
-    """
-    messages = [
-        {"role": "system", "content": SYSTEM_PROMPT},
-        {"role": "user", "content": question},
-    ]
+    try:
+        context = ""
+        if files:
+            for f in files:
+                context += download_file(f) + "\n"
+
+        messages = [
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": f"{context}{question}".strip()},
+        ]
 
-    result = graph.invoke({"messages": messages})
-    answer = result["messages"][-1].content
+        res = llm.invoke(messages)
+        return clean_output(res.content)
 
-    # ✅ GAIA 硬性规则
-    return answer.strip().splitlines()[0]
+    except Exception:
+        return "0"