g

app.py

@@ -24,6 +24,16 @@ def merge_dicts(old: Dict, new: Dict) -> Dict:
     """Merge two dictionaries, with *new* values taking precedence."""
     return {**old, **new}
 
+def tighten(q: str) -> str:
+    """
+    Strip long GAIA questions down to quoted phrases and capitalised words.
+    Falls back to the original text if we strip too much.
+    """
+    quoted = re.findall(r'"([^"]+)"', q)
+    caps   = re.findall(r'\b([A-Z0-9][\w-]{2,})', q)
+    short  = " ".join(quoted + caps)
+    return short or q
+
 # -------------------------
 # Environment & constants
 # -------------------------

tools.py

@@ -13,7 +13,7 @@ def tighten(q: str) -> str:
 def _raw_search(query: str, max_results: int = 5) -> List[str]:
     """Internal function that performs the actual DuckDuckGo search."""
     with DDGS() as ddgs:
-        raw = list(ddgs.text(keywords=query, max_results=max_results))
+        raw = list(ddgs.text(query, max_results=max_results))
     out = []
     for r in raw:
         try: