import re, ast, json, operator, requests
UA={"User-Agent":"zero-loop-distill/1.0 (research notebook)"}
API="https://en.wikipedia.org/w/api.php"
_cache={}
def search(query, n=3):
    query=str(query).strip()[:200]
    if not query: raise ValueError("empty query")
    if query in _cache: return _cache[query]
    r=requests.get(API, params={"action":"query","list":"search","srsearch":query,"srlimit":n,"format":"json"}, headers=UA, timeout=20)
    r.raise_for_status()
    titles=[h["title"] for h in r.json()["query"]["search"]]
    if not titles: raise ValueError("no results")
    r2=requests.get(API, params={"action":"query","prop":"extracts","exintro":1,"explaintext":1,"exchars":400,"exlimit":n,"titles":"|".join(titles),"redirects":1,"format":"json"}, headers=UA, timeout=20)
    r2.raise_for_status()
    ex={p["title"]:" ".join(p.get("extract","").split()) for p in r2.json()["query"]["pages"].values()}
    if not any(ex.values()): raise ValueError("empty extracts")
    out="\n".join(f"{i+1}. {t}: {ex.get(t,'')}" for i,t in enumerate(titles))[:1300]
    _cache[query]=out
    return out
_ops={ast.Add:operator.add,ast.Sub:operator.sub,ast.Mult:operator.mul,ast.Div:operator.truediv,ast.FloorDiv:operator.floordiv,ast.Mod:operator.mod,ast.Pow:operator.pow,ast.USub:operator.neg,ast.UAdd:operator.pos}
def _ev(n):
    if isinstance(n,ast.Constant) and isinstance(n.value,(int,float)) and not isinstance(n.value,bool): return n.value
    if isinstance(n,ast.BinOp) and type(n.op) in _ops:
        a,b=_ev(n.left),_ev(n.right)
        if isinstance(n.op,ast.Pow) and abs(b)>100: raise ValueError("exponent too large")
        return _ops[type(n.op)](a,b)
    if isinstance(n,ast.UnaryOp) and type(n.op) in _ops: return _ops[type(n.op)](_ev(n.operand))
    raise ValueError("unsupported expression")
def calculator(expression):
    e=str(expression).replace("^","**").replace("\u00d7","*").replace("\u00f7","/").replace(",","").strip()
    if not e or len(e)>200: raise ValueError("bad expression")
    v=_ev(ast.parse(e,mode="eval").body)
    if isinstance(v,float):
        v=int(v) if (v==int(v) and abs(v)<1e15) else round(v,10)
    return str(v)
def run_tool(name, args):
    if name=="search": return search(args["query"])
    if name=="calculator": return calculator(args["expression"])
    raise KeyError(name)
CALL_RE=re.compile(r"<tool_call>\s*(\{.*?\})\s*</tool_call>", re.S)
def parse_call(text):
    m=CALL_RE.search(text)
    if not m: return None
    try: c=json.loads(m.group(1))
    except Exception: return None
    if not isinstance(c,dict) or c.get("name") not in ("search","calculator") or not isinstance(c.get("arguments"),dict): return None
    k="query" if c["name"]=="search" else "expression"
    if not isinstance(c["arguments"].get(k),str) or not c["arguments"][k].strip(): return None
    return {"name":c["name"],"arguments":{k:c["arguments"][k].strip()}}