Spaces:
Running
Running
| #!/usr/bin/env python | |
| """Tool-Using Agent — ask a question and watch the AI call real, live tools to answer it: | |
| a calculator, the current date/time, live weather, and Wikipedia. It runs the ReAct loop | |
| (decide → call tool → read result → repeat) and shows every tool call it made. | |
| All tools are keyless (Open-Meteo + MediaWiki), so the only secret is ANTHROPIC_API_KEY. | |
| pip install flask requests anthropic | |
| python app.py # http://127.0.0.1:7860 | |
| Project #9 of the "30 Projects in 15 Days" challenge — GritAI. | |
| UI: GritAI engineered/industrial design system (dark, grit-orange, IBM Plex + Archivo). | |
| """ | |
| import os, ast, json, math, operator, datetime, urllib.parse | |
| from flask import Flask, request, jsonify, Response | |
| import requests | |
| PORT = int(os.environ.get("PORT", "7860")) | |
| ANTHROPIC_KEY = os.environ.get("ANTHROPIC_API_KEY") | |
| ANTHROPIC_MODEL = os.environ.get("ANTHROPIC_MODEL", "claude-haiku-4-5") | |
| MAX_STEPS = 6 | |
| UA = {"User-Agent": "GritAI-ToolAgent/1.0 (https://gritai.solutions)"} | |
| # --------------------------------------------------------------------------- tools | |
| _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul, | |
| ast.Div: operator.truediv, ast.Pow: operator.pow, ast.Mod: operator.mod, | |
| ast.FloorDiv: operator.floordiv, ast.USub: operator.neg, ast.UAdd: operator.pos} | |
| _FUNCS = {"sqrt": math.sqrt, "abs": abs, "round": round, "log": math.log, "log10": math.log10, | |
| "sin": math.sin, "cos": math.cos, "tan": math.tan, "pi": math.pi, "e": math.e, | |
| "floor": math.floor, "ceil": math.ceil, "exp": math.exp, "min": min, "max": max} | |
| def _safe_eval(node): | |
| if isinstance(node, ast.Constant): | |
| if isinstance(node.value, (int, float)): | |
| return node.value | |
| raise ValueError("only numbers") | |
| if isinstance(node, ast.BinOp): | |
| return _OPS[type(node.op)](_safe_eval(node.left), _safe_eval(node.right)) | |
| if isinstance(node, ast.UnaryOp): | |
| return _OPS[type(node.op)](_safe_eval(node.operand)) | |
| if isinstance(node, ast.Name): | |
| if node.id in _FUNCS: | |
| return _FUNCS[node.id] | |
| raise ValueError(f"unknown name {node.id}") | |
| if isinstance(node, ast.Call): | |
| fn = _FUNCS.get(getattr(node.func, "id", "")) | |
| if not callable(fn): | |
| raise ValueError("unknown function") | |
| return fn(*[_safe_eval(a) for a in node.args]) | |
| raise ValueError("unsupported expression") | |
| def tool_calculator(expression): | |
| try: | |
| val = _safe_eval(ast.parse(str(expression), mode="eval").body) | |
| return f"{expression} = {val}" | |
| except Exception as e: | |
| return f"Could not evaluate '{expression}': {e}" | |
| def tool_datetime(timezone="UTC"): | |
| now = datetime.datetime.now(datetime.timezone.utc) | |
| return f"Current UTC date/time: {now.strftime('%A, %d %B %Y, %H:%M UTC')}" | |
| _WMO = {0: "clear sky", 1: "mainly clear", 2: "partly cloudy", 3: "overcast", 45: "fog", 48: "rime fog", | |
| 51: "light drizzle", 53: "drizzle", 55: "dense drizzle", 61: "light rain", 63: "rain", | |
| 65: "heavy rain", 71: "light snow", 73: "snow", 75: "heavy snow", 80: "rain showers", | |
| 81: "rain showers", 82: "violent rain showers", 95: "thunderstorm", 96: "thunderstorm w/ hail"} | |
| def tool_weather(location): | |
| try: | |
| g = requests.get("https://geocoding-api.open-meteo.com/v1/search", | |
| params={"name": location, "count": 1}, headers=UA, timeout=15).json() | |
| if not g.get("results"): | |
| return f"No location found for '{location}'." | |
| r = g["results"][0] | |
| lat, lon = r["latitude"], r["longitude"] | |
| place = ", ".join(filter(None, [r.get("name"), r.get("admin1"), r.get("country")])) | |
| w = requests.get("https://api.open-meteo.com/v1/forecast", | |
| params={"latitude": lat, "longitude": lon, | |
| "current": "temperature_2m,relative_humidity_2m,weather_code,wind_speed_10m"}, | |
| headers=UA, timeout=15).json() | |
| c = w.get("current", {}) | |
| desc = _WMO.get(c.get("weather_code"), "unknown") | |
| return (f"Weather in {place}: {c.get('temperature_2m')}°C, {desc}, " | |
| f"humidity {c.get('relative_humidity_2m')}%, wind {c.get('wind_speed_10m')} km/h.") | |
| except Exception as e: | |
| return f"Weather lookup failed: {e}" | |
| def tool_wikipedia(query): | |
| try: | |
| s = requests.get("https://en.wikipedia.org/w/api.php", headers=UA, timeout=15, | |
| params={"action": "query", "list": "search", "srsearch": query, | |
| "format": "json", "srlimit": 1}).json() | |
| hits = s.get("query", {}).get("search", []) | |
| if not hits: | |
| return f"No Wikipedia article found for '{query}'." | |
| title = hits[0]["title"] | |
| e = requests.get("https://en.wikipedia.org/w/api.php", headers=UA, timeout=15, | |
| params={"action": "query", "prop": "extracts", "exintro": 1, "explaintext": 1, | |
| "titles": title, "format": "json"}).json() | |
| pages = e.get("query", {}).get("pages", {}) | |
| extract = next(iter(pages.values()), {}).get("extract", "") | |
| return f"Wikipedia — {title}: {extract[:800]}" if extract else f"Found '{title}' but no summary." | |
| except Exception as e: | |
| return f"Wikipedia lookup failed: {e}" | |
| TOOLS_SPEC = [ | |
| {"name": "calculator", "description": "Evaluate an arithmetic expression. Supports + - * / ** % and " | |
| "sqrt, log, sin, cos, tan, floor, ceil, abs, round, pi, e.", | |
| "input_schema": {"type": "object", "properties": {"expression": {"type": "string"}}, | |
| "required": ["expression"]}}, | |
| {"name": "current_datetime", "description": "Get the current date and time (UTC).", | |
| "input_schema": {"type": "object", "properties": {"timezone": {"type": "string"}}}}, | |
| {"name": "weather", "description": "Get the current weather for a city/place name (live).", | |
| "input_schema": {"type": "object", "properties": {"location": {"type": "string"}}, | |
| "required": ["location"]}}, | |
| {"name": "wikipedia", "description": "Look up a topic on Wikipedia and get a summary (live).", | |
| "input_schema": {"type": "object", "properties": {"query": {"type": "string"}}, | |
| "required": ["query"]}}, | |
| ] | |
| EXECUTORS = {"calculator": tool_calculator, "current_datetime": tool_datetime, | |
| "weather": tool_weather, "wikipedia": tool_wikipedia} | |
| def run_agent(question): | |
| import anthropic | |
| client = anthropic.Anthropic(api_key=ANTHROPIC_KEY) | |
| sysp = ("You are a helpful assistant that answers using the provided tools whenever they help. " | |
| "Call tools to get live/precise data instead of guessing. You may call multiple tools. " | |
| "When you have enough, give a clear, concise final answer.") | |
| msgs = [{"role": "user", "content": question}] | |
| trace = [] | |
| for _ in range(MAX_STEPS): | |
| r = client.messages.create(model=ANTHROPIC_MODEL, max_tokens=1024, system=sysp, | |
| tools=TOOLS_SPEC, messages=msgs) | |
| if r.stop_reason == "tool_use": | |
| msgs.append({"role": "assistant", "content": r.content}) | |
| results = [] | |
| for b in r.content: | |
| if b.type == "tool_use": | |
| out = EXECUTORS.get(b.name, lambda **k: "unknown tool")(**b.input) | |
| trace.append({"tool": b.name, "input": b.input, "result": out[:400]}) | |
| results.append({"type": "tool_result", "tool_use_id": b.id, "content": out}) | |
| msgs.append({"role": "user", "content": results}) | |
| else: | |
| text = "".join(b.text for b in r.content if b.type == "text") | |
| return text.strip(), trace | |
| return "I reached the step limit before finishing — try a simpler question.", trace | |
| app = Flask(__name__) | |
| app.config["MAX_CONTENT_LENGTH"] = 512 * 1024 | |
| def home(): | |
| return Response(PAGE, mimetype="text/html") | |
| def api_ask(): | |
| q = (request.get_json(force=True).get("question") or "").strip()[:1000] | |
| if not q: | |
| return jsonify(error="Ask a question."), 400 | |
| if not ANTHROPIC_KEY: | |
| return jsonify(error="Set ANTHROPIC_API_KEY on the server to run the agent."), 200 | |
| try: | |
| answer, trace = run_agent(q) | |
| except Exception as e: | |
| import sys; print("agent error:", type(e).__name__, file=sys.stderr, flush=True) | |
| return jsonify(error="The agent hit an error — please try again."), 200 | |
| return jsonify(answer=answer, trace=trace) | |
| # ===================== GritAI engineered/industrial design system ===================== | |
| PAGE = """<!doctype html><html lang="en"><head><meta charset="utf-8"> | |
| <title>GritAI · Tool-Using Agent</title> | |
| <meta name="viewport" content="width=device-width,initial-scale=1"> | |
| <link rel="preconnect" href="https://fonts.googleapis.com"> | |
| <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin> | |
| <link href="https://fonts.googleapis.com/css2?family=Archivo:wght@600;700;800;900&family=IBM+Plex+Mono:wght@400;500;600&family=IBM+Plex+Sans:wght@400;500;600&display=swap" rel="stylesheet"> | |
| <style> | |
| :root{--bg:#0a0b0e;--bg2:#0e1015;--panel:#111318;--ink:#eef0f4;--dim:#9aa1b0;--dim2:#5f6675;--line:#20242e;--line2:#2b303c; | |
| --grit:#ff5a2a;--grit2:#ff7d54;--grit-dim:#c9451f; | |
| --disp:'Archivo',system-ui,sans-serif;--body:'IBM Plex Sans',system-ui,sans-serif;--mono:'IBM Plex Mono',ui-monospace,monospace} | |
| *{box-sizing:border-box}html,body{height:100%} | |
| body{margin:0;background:var(--bg);color:var(--ink);font-family:var(--body);font-size:15px;line-height:1.6;-webkit-font-smoothing:antialiased;position:relative;overflow-x:hidden} | |
| body::before{content:"";position:fixed;inset:0;z-index:0;pointer-events:none; | |
| background:linear-gradient(0deg,transparent 95%,rgba(255,90,42,.04) 100%), | |
| repeating-linear-gradient(90deg,transparent 0 63px,rgba(255,255,255,.018) 63px 64px), | |
| repeating-linear-gradient(0deg,transparent 0 63px,rgba(255,255,255,.018) 63px 64px), | |
| radial-gradient(120% 80% at 50% -10%,#141821 0%,var(--bg) 55%)} | |
| body::after{content:"";position:fixed;inset:0;z-index:0;pointer-events:none;opacity:.05;mix-blend-mode:overlay; | |
| background-image:url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' width='140' height='140'%3E%3Cfilter id='n'%3E%3CfeTurbulence type='fractalNoise' baseFrequency='.9' numOctaves='2'/%3E%3C/filter%3E%3Crect width='100%25' height='100%25' filter='url(%23n)'/%3E%3C/svg%3E")} | |
| .wrap{position:relative;z-index:1;max-width:960px;margin:0 auto;padding:0 28px 70px} | |
| header{display:flex;align-items:center;gap:16px;padding:22px 0 18px;border-bottom:1px solid var(--line);animation:fade .6s .02s both} | |
| .mark{width:38px;height:38px;flex:none}.mark svg{display:block} | |
| .brand{display:flex;flex-direction:column;line-height:1} | |
| .brand .co{font-family:var(--disp);font-weight:900;font-size:19px;letter-spacing:.5px}.brand .co b{color:var(--grit)} | |
| .brand .sub{font-family:var(--mono);font-size:10.5px;letter-spacing:.28em;text-transform:uppercase;color:var(--dim2);margin-top:5px} | |
| .tags{margin-left:auto;display:flex;gap:8px} | |
| .tag{font-family:var(--mono);font-size:10px;letter-spacing:.14em;text-transform:uppercase;color:var(--dim);border:1px solid var(--line2);border-radius:2px;padding:5px 9px} | |
| .tag.hot{color:var(--grit);border-color:var(--grit-dim)} | |
| .hero{padding:34px 0 26px;animation:fade .6s .12s both} | |
| .hero h1{font-family:var(--disp);font-weight:800;font-size:clamp(30px,5.4vw,52px);line-height:1.02;letter-spacing:-.02em;margin:0;max-width:16ch} | |
| .hero h1 em{font-style:normal;color:var(--grit)} | |
| .hero p{color:var(--dim);max-width:56ch;margin:16px 0 0;font-size:15.5px} | |
| .sec{display:flex;align-items:center;gap:14px;margin:30px 0 14px} | |
| .sec .num{font-family:var(--mono);font-size:11px;font-weight:600;color:var(--grit);letter-spacing:.1em} | |
| .sec .name{font-family:var(--mono);font-size:11px;font-weight:500;letter-spacing:.26em;text-transform:uppercase;color:var(--dim)} | |
| .sec .rule{flex:1;height:1px;background:linear-gradient(90deg,var(--line2),transparent)} | |
| .ask{animation:fade .6s .2s both} | |
| .field{display:flex;align-items:stretch;border:1px solid var(--line2);border-radius:3px;background:var(--panel);overflow:hidden;transition:border-color .18s} | |
| .field:focus-within{border-color:var(--grit-dim)} | |
| .field .car{font-family:var(--mono);color:var(--grit);padding:16px 0 16px 16px;font-size:15px;user-select:none} | |
| .field input{flex:1;background:transparent;border:0;outline:none;color:var(--ink);font-family:var(--body);font-size:15.5px;padding:16px 14px} | |
| .field input::placeholder{color:var(--dim2)} | |
| .run{border:0;background:var(--grit);color:#140a06;font-family:var(--mono);font-weight:600;font-size:13px;letter-spacing:.14em;text-transform:uppercase;padding:0 24px;cursor:pointer;transition:.16s;box-shadow:-1px 0 0 var(--grit-dim)} | |
| .run:hover{background:var(--grit2);box-shadow:0 0 26px -4px rgba(255,90,42,.55)} | |
| .run:disabled{opacity:.55;cursor:default;box-shadow:none} | |
| .ex{display:flex;flex-wrap:wrap;gap:8px;margin-top:12px} | |
| .chip{font-family:var(--mono);font-size:12px;color:var(--dim);background:var(--panel);border:1px solid var(--line);border-radius:2px;padding:8px 11px;cursor:pointer;transition:.15s} | |
| .chip:hover{color:var(--grit);border-color:var(--grit-dim)} | |
| .chip .pt{color:var(--dim2);margin-right:7px} | |
| .out{display:none}.out.show{display:block} | |
| .trace{border:1px solid var(--line);border-radius:3px;background:linear-gradient(180deg,var(--panel),var(--bg2))} | |
| .log{font-family:var(--mono);font-size:13px;line-height:1.75} | |
| .row{display:grid;grid-template-columns:22px 100px 1fr;gap:0 12px;padding:13px 18px;border-bottom:1px solid var(--line);animation:slidein .45s both} | |
| .row:last-child{border-bottom:0} | |
| .row .mk{color:var(--grit);font-weight:600} | |
| .row .tn{color:var(--grit2);text-transform:uppercase;letter-spacing:.06em;font-weight:500} | |
| .row .rd{color:var(--ink);word-break:break-word}.row .rd .arg{color:var(--dim2)} | |
| .row.none .rd{color:var(--dim)} | |
| .ans{border:1px solid var(--line2);border-left:2px solid var(--grit);border-radius:3px;background:var(--panel);padding:22px 24px} | |
| .ans p{margin:0 0 12px;font-size:16px;line-height:1.65}.ans p:last-child{margin-bottom:0} | |
| .spin{display:inline-flex;gap:6px;padding:16px 18px}.spin span{width:8px;height:8px;border-radius:50%;background:var(--grit);opacity:.5;animation:bd 1.2s infinite} | |
| .spin span:nth-child(2){animation-delay:.18s}.spin span:nth-child(3){animation-delay:.36s} | |
| footer{margin-top:34px;padding-top:16px;border-top:1px solid var(--line);font-family:var(--mono);font-size:11px;letter-spacing:.06em;color:var(--dim2);display:flex;gap:16px;flex-wrap:wrap;animation:fade .6s .5s both} | |
| footer .b{color:var(--dim)}footer .sp{margin-left:auto} | |
| footer a{color:var(--dim);text-decoration:none}footer a:hover{color:var(--grit)} | |
| @keyframes fade{from{opacity:0;transform:translateY(10px)}to{opacity:1;transform:none}} | |
| @keyframes slidein{from{opacity:0;transform:translateX(-8px)}to{opacity:1;transform:none}} | |
| @keyframes bd{0%,60%,100%{transform:translateY(0);opacity:.35}30%{transform:translateY(-6px);opacity:1}} | |
| </style></head> | |
| <body><div class="wrap"> | |
| <header> | |
| <div class="mark"><svg width="38" height="38" viewBox="0 0 38 38" fill="none"> | |
| <rect x="1" y="1" width="36" height="36" rx="2" stroke="#2b303c"/> | |
| <path d="M27 12.5A9 9 0 1 0 28 22H19" stroke="#ff5a2a" stroke-width="2.4" stroke-linecap="square"/> | |
| <rect x="10.5" y="10.5" width="3" height="3" fill="#ff5a2a"/> | |
| </svg></div> | |
| <div class="brand"><div class="co">GRIT<b>AI</b></div><div class="sub">Tool-Using Agent</div></div> | |
| <div class="tags"><span class="tag hot">▸ Live</span><span class="tag">SDVOSB</span><span class="tag">Lawton OK</span></div> | |
| </header> | |
| <section class="hero"> | |
| <h1>Ask it anything. It <em>calls real tools</em> to answer.</h1> | |
| <p>A calculator, a clock, live weather, and Wikipedia — the agent decides what to call, runs it, and shows its work. Not a chatbot guessing. An operator.</p> | |
| </section> | |
| <div class="sec"><span class="num">01</span><span class="name">Ask</span><span class="rule"></span></div> | |
| <div class="ask"> | |
| <div class="field"> | |
| <span class="car">>_</span> | |
| <input id="q" placeholder="What's a 15% tip on $47.80, and the current weather in Paris?" autofocus> | |
| <button class="run" id="go" onclick="ask()">Run ▸</button> | |
| </div> | |
| <div class="ex" id="ex"></div> | |
| </div> | |
| <div class="out" id="out"> | |
| <div class="sec"><span class="num">02</span><span class="name">Trace — live tool calls</span><span class="rule"></span></div> | |
| <div class="trace"><div class="log" id="trace"></div></div> | |
| <div class="sec"><span class="num">03</span><span class="name">Answer</span><span class="rule"></span></div> | |
| <div class="ans" id="answer"></div> | |
| </div> | |
| <footer> | |
| <span class="b">ReAct loop</span><span>·</span><span>keyless tools — Open-Meteo + Wikipedia</span> | |
| <span class="sp"><a href="https://gritai.solutions">GRITAI SOLUTIONS</a> · 30-in-15 №9</span> | |
| </footer> | |
| </div><script> | |
| const EXAMPLES=["Who invented the World Wide Web, and what's the square root of 4096?", | |
| "What's the weather in Tokyo, and the current UTC time?", | |
| "Summarize what an octopus is, then give me a 12% tip on $86."]; | |
| const $=id=>document.getElementById(id); | |
| const q=$('q'),go=$('go'),out=$('out'); | |
| function esc(s){return String(s==null?'':s).replace(/[&<>]/g,c=>({'&':'&','<':'<','>':'>'}[c]));} | |
| $('ex').innerHTML=EXAMPLES.map(e=>'<span class="chip"><span class="pt">↳</span>'+esc(e)+'</span>').join(''); | |
| $('ex').querySelectorAll('.chip').forEach(c=>c.onclick=()=>{q.value=c.textContent.replace(/^↳\s*/,'');ask();}); | |
| function argStr(o){try{return Object.values(o).join(', ');}catch(e){return '';}} | |
| function ask(){if(go.disabled)return;const text=q.value.trim();if(!text){q.focus();return;} | |
| go.disabled=true;go.textContent='…';out.classList.add('show'); | |
| $('trace').innerHTML='<div class="spin"><span></span><span></span><span></span></div>';$('answer').innerHTML=''; | |
| fetch('/api/ask',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({question:text})}) | |
| .then(r=>r.json()).then(d=>{ | |
| if(d.error){$('trace').innerHTML='<div class="row none"><span class="mk">!</span><span class="tn">error</span><span class="rd">'+esc(d.error)+'</span></div>';return;} | |
| const t=d.trace||[]; | |
| $('trace').innerHTML=t.length?t.map(s=>'<div class="row"><span class="mk">»</span><span class="tn">'+esc(s.tool)+'</span><span class="rd"><span class="arg">'+esc(argStr(s.input))+'</span> → '+esc(s.result)+'</span></div>').join(''): | |
| '<div class="row none"><span class="mk">·</span><span class="tn">direct</span><span class="rd">answered directly — no tools needed</span></div>'; | |
| $('answer').innerHTML=String(d.answer||'').split(/\\n\\n+/).map(p=>'<p>'+esc(p).replace(/\\n/g,'<br>')+'</p>').join(''); | |
| }).catch(()=>{$('trace').innerHTML='<div class="row none"><span class="mk">!</span><span class="tn">error</span><span class="rd">Something went wrong — please try again.</span></div>';}) | |
| .finally(()=>{go.disabled=false;go.textContent='Run ▸';}); | |
| } | |
| q.addEventListener('keydown',e=>{if(e.key==='Enter')ask();}); | |
| </script></body></html>""" | |
| if __name__ == "__main__": | |
| backend = "Claude API" if ANTHROPIC_KEY else "NO KEY (set ANTHROPIC_API_KEY)" | |
| print(f"Tool-Using Agent on http://127.0.0.1:{PORT} [backend: {backend}]", flush=True) | |
| app.run(host="0.0.0.0", port=PORT, threaded=True) | |