chore: establish project baseline

2026-09-21 15:51:47 -03:00
commit 9dcc263217
459 changed files with 447890 additions and 0 deletions
@@ -0,0 +1,37 @@
+import httpx
+
+class HubClient:
+    def __init__(self, url, token, chunk_size=100): self.url, self.token, self.chunk_size = url.rstrip("/"), token, chunk_size
+    def _headers(self): return {"Authorization": f"Bearer {self.token}"}
+    def classify_batch(self, products):
+        results = []
+        for start in range(0, len(products), self.chunk_size):
+            chunk = products[start:start + self.chunk_size]
+            response = httpx.post(f"{self.url}/v1/classify/batch", json={"products": chunk}, headers=self._headers(), timeout=60)
+            response.raise_for_status(); results.extend(response.json()["results"])
+        return results
+    def get_json(self, path):
+        response = httpx.get(f"{self.url}{path}", headers=self._headers(), timeout=30)
+        response.raise_for_status(); return response.json()
+    def health(self):
+        response = httpx.get(f"{self.url}/health", timeout=10)
+        return response.status_code == 200
+    def heartbeat(self, status, schema_fingerprint=None, agent_version=None, last_job_id=None):
+        payload = {"status": status, "schema_fingerprint": schema_fingerprint, "agent_version": agent_version, "last_job_id": last_job_id}
+        response = httpx.post(f"{self.url}/v1/agents/heartbeat", json=payload, headers=self._headers(), timeout=15)
+        response.raise_for_status(); return response.json()
+    def next_job(self):
+        response = httpx.get(f"{self.url}/v1/jobs/next", headers=self._headers(), timeout=15)
+        response.raise_for_status(); return response.json()
+    def create_job(self, kind, payload=None):
+        response = httpx.post(f"{self.url}/v1/jobs", json={"kind": kind, "payload": payload or {}}, headers=self._headers(), timeout=15)
+        response.raise_for_status(); return response.json()
+    def get_job(self, job_id):
+        response = httpx.get(f"{self.url}/v1/jobs/{job_id}", headers=self._headers(), timeout=15)
+        response.raise_for_status(); return response.json()
+    def complete_job(self, job_id, result_summary):
+        response = httpx.post(f"{self.url}/v1/jobs/{job_id}/complete", json={"result_summary": result_summary}, headers=self._headers(), timeout=15)
+        response.raise_for_status(); return response.json()
+    def fail_job(self, job_id, error):
+        response = httpx.post(f"{self.url}/v1/jobs/{job_id}/fail", json={"error": str(error)}, headers=self._headers(), timeout=15)
+        response.raise_for_status(); return response.json()