commit

app.py

@@ -9,7 +9,6 @@ from typing import TypedDict
 
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import HumanMessage
-from langchain_community.tools import DuckDuckGoSearchRun
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -26,13 +25,6 @@ You MUST:
 - No extra text.
 """
 
-RULES = {
-    "number": "Answer with ONLY the number. No words.",
-    "date": "Answer with ONLY the year or date. No words.",
-    "yesno": "Answer with ONLY Yes or No.",
-    "fact": "Answer with ONLY the final answer."
-}
-
 def clean_answer(text: str) -> str:
     if not text:
         return ""
@@ -42,22 +34,8 @@ def clean_answer(text: str) -> str:
     s = s.strip('"\'`')
     if len(s) > 1 and s.endswith("."):
         s = s[:-1].strip()
-
-    if s.lower() in ["yes", "no"]:
-        s = s.capitalize()
-
     return s
 
-def classify_question(q: str) -> str:
-    ql = q.lower().strip()
-    if any(k in ql for k in ["how many", "how much", "number of", "population"]):
-        return "number"
-    if any(k in ql for k in ["what year", "when", "date"]):
-        return "date"
-    if ql.startswith(("is ", "are ", "was ", "were ", "did ", "does ")):
-        return "yesno"
-    return "fact"
-
 # -------------------------------
 # State
 # -------------------------------
@@ -68,8 +46,6 @@ class AgentState(TypedDict):
 # -------------------------------
 # Tools & LLM
 # -------------------------------
-# Search tool (무료)
-search_tool = DuckDuckGoSearchRun()
 
 # LLM (OpenAI – 이미 네 환경에서 동작 확인됨)
 llm = ChatOpenAI(
@@ -83,56 +59,32 @@ llm = ChatOpenAI(
 # -------------------------------
 class BasicAgent:
     def __init__(self):
-        print("Search-based GAIA Agent initialized.")
+        print("GAIA LLM-only Agent initialized.")
 
     def __call__(self, question: str) -> str:
         print(f"Question: {question[:80]}...")
 
-        qtype = classify_question(question)
-
-        # 🔹 1단계: 검색 없이 먼저 답 시도
-        base_prompt = f"""
-    You are solving a GAIA benchmark question.
-
-    Rules:
-    - Answer with ONLY the final answer
-    - No explanation
-    - No extra text
-
-    Question:
-    {question}
-    """.strip()
-
-        base_response = llm.invoke([HumanMessage(content=base_prompt)])
-        base_answer = clean_answer(base_response.content)
-
-        # 🔹 숫자/연도/YesNo가 명확하면 바로 반환
-        if qtype in ["number", "date", "yesno"] and base_answer:
-            print(f"Answer (no search): {base_answer}")
-            return base_answer
-
-        # 🔹 2단계: 그래도 애매하면 검색 사용
-        try:
-            search_result = search_tool.run(question)
-        except Exception:
-            search_result = ""
-
         prompt = f"""
-    {SYSTEM_PROMPT}
+You are solving a GAIA benchmark question.
 
-    Question:
-    {question}
+Rules:
+- Answer with ONLY the final answer.
+- No explanation.
+- No formatting.
+- No extra text.
+- If the answer is a number, output ONLY the number.
+- If the answer is Yes/No, output ONLY Yes or No.
 
-    Search Results:
-    {search_result}
-    """.strip()
+Question:
+{question}
+""".strip()
 
         response = llm.invoke([HumanMessage(content=prompt)])
         answer = clean_answer(response.content)
 
-        print(f"Answer (with search): {answer}")
+        print(f"Answer: {answer}")
         return answer
-
+    
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,