Spaces:
Running
Running
| import httpx | |
| import json | |
| import time | |
| URL = "https://leon4gr45-needle.hf.space" | |
| # Optional mode shouldn't strictly require this for basic tests, but let's see | |
| # The API requires a valid key for privileged mutation in optional mode. But for read it's fine. | |
| def run(): | |
| client = httpx.Client(timeout=30.0) | |
| traces = [] | |
| def trace_req(method, path, payload=None, headers=None): | |
| t0 = time.time() | |
| try: | |
| if payload: | |
| res = client.request(method, f"{URL}{path}", json=payload, headers=headers) | |
| else: | |
| res = client.request(method, f"{URL}{path}", headers=headers) | |
| t1 = time.time() | |
| traces.append({ | |
| "method": method, | |
| "path": path, | |
| "payload": payload, | |
| "status": res.status_code, | |
| "response": res.text[:500] if len(res.text) > 500 else res.json() if "{" in res.text else res.text, | |
| "latency_ms": round((t1 - t0) * 1000, 2), | |
| "timestamp": t0 | |
| }) | |
| print(f"{method} {path} -> {res.status_code}") | |
| except Exception as e: | |
| print(f"Error {method} {path}: {e}") | |
| # 1. Health | |
| trace_req("GET", "/health") | |
| # 2. Ready | |
| trace_req("GET", "/ready") | |
| # 3. Info | |
| trace_req("GET", "/v1/info") | |
| # 4. Tool registry retrieval | |
| trace_req("GET", "/v1/tools") | |
| # 5. Tool candidate retrieval | |
| trace_req("POST", "/v1/tools/retrieve", payload={"query": "test query", "max_candidates": 5}) | |
| # 6. /v1/route | |
| trace_req("POST", "/v1/route", payload={"input": "test", "tools": [{"name": "test", "description": "desc", "parameters": {"type": "object"}}]}) | |
| # 7. Hierarchical gate routing | |
| trace_req("POST", "/v1/gates/route", payload={"query": "test", "tools": [{"name": "test", "description": "desc", "parameters": {"type": "object"}}], "max_candidates": 2}) | |
| # 8. OpenAI compatible endpoint | |
| trace_req("POST", "/v1/chat/completions", payload={"model": "needle3", "messages": [{"role": "user", "content": "test"}], "tools": [{"name": "test", "description": "desc", "parameters": {"type": "object"}}]}) | |
| # 9. Metrics | |
| trace_req("GET", "/v1/metrics") | |
| with open("tests/live/evidence/api-traces.jsonl", "w") as f: | |
| for t in traces: | |
| f.write(json.dumps(t) + "\n") | |
| if __name__ == '__main__': | |
| run() | |