querymemory 0.1.0__tar.gz → 0.2.0__tar.gz

{querymemory-0.1.0 → querymemory-0.2.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: querymemory
-Version: 0.1.0
-Summary: Official Query Memory SDK for Knowledge, Agent, and Web Parsing APIs
+Version: 0.2.0
+Summary: Official Query Memory SDK for Knowledge, Agent, Web Parsing, and Extraction APIs
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.31.0
@@ -13,7 +13,7 @@ Official SDK for Query Memory Knowledge, Agent, and Web Parsing APIs.
 ## Install
 
 ```bash
-pip install querymemory-sdk
+pip install querymemory
 ```
 
 ## Quick Start

{querymemory-0.1.0 → querymemory-0.2.0}/README.md

@@ -5,7 +5,7 @@ Official SDK for Query Memory Knowledge, Agent, and Web Parsing APIs.
 ## Install
 
 ```bash
-pip install querymemory-sdk
+pip install querymemory
 ```
 
 ## Quick Start

{querymemory-0.1.0 → querymemory-0.2.0}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "querymemory"
-version = "0.1.0"
-description = "Official Query Memory SDK for Knowledge, Agent, and Web Parsing APIs"
+version = "0.2.0"
+description = "Official Query Memory SDK for Knowledge, Agent, Web Parsing, and Extraction APIs"
 readme = "README.md"
 requires-python = ">=3.9"
 dependencies = [

{querymemory-0.1.0 → querymemory-0.2.0}/querymemory.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: querymemory
-Version: 0.1.0
-Summary: Official Query Memory SDK for Knowledge, Agent, and Web Parsing APIs
+Version: 0.2.0
+Summary: Official Query Memory SDK for Knowledge, Agent, Web Parsing, and Extraction APIs
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.31.0
@@ -13,7 +13,7 @@ Official SDK for Query Memory Knowledge, Agent, and Web Parsing APIs.
 ## Install
 
 ```bash
-pip install querymemory-sdk
+pip install querymemory
 ```
 
 ## Quick Start

{querymemory-0.1.0 → querymemory-0.2.0}/querymemory_sdk/client.py

@@ -294,6 +294,76 @@ class WebApi:
         )
 
 
+class ExtractionsApi:
+    def __init__(self, http: _HttpClient):
+        self._http = http
+
+    def list_templates(self) -> Dict[str, Any]:
+        return self._http.request("GET", "/api/v1/extractions/templates", auth_mode="auto")
+
+    def get_template(self, template_id: str) -> Dict[str, Any]:
+        return self._http.request("GET", f"/api/v1/extractions/templates/{template_id}", auth_mode="auto")
+
+    def create_template(
+        self,
+        *,
+        name: str,
+        description: Optional[str] = None,
+        fields: Optional[list[Dict[str, Any]]] = None,
+    ) -> Dict[str, Any]:
+        return self._http.request(
+            "POST",
+            "/api/v1/extractions/templates",
+            auth_mode="auto",
+            json_body={
+                "name": name,
+                "description": description,
+                "fields": fields or [],
+            },
+        )
+
+    def update_template(self, template_id: str, **payload: Any) -> Dict[str, Any]:
+        return self._http.request("PUT", f"/api/v1/extractions/templates/{template_id}", auth_mode="auto", json_body=payload)
+
+    def delete_template(self, template_id: str) -> Dict[str, Any]:
+        return self._http.request("DELETE", f"/api/v1/extractions/templates/{template_id}", auth_mode="auto")
+
+    def run(self, *, template_id: str, knowledge_id: str) -> Dict[str, Any]:
+        return self._http.request(
+            "POST",
+            "/api/v1/extractions/run",
+            auth_mode="auto",
+            json_body={"template_id": template_id, "knowledge_id": knowledge_id},
+        )
+
+    def list_jobs(self) -> Dict[str, Any]:
+        return self._http.request("GET", "/api/v1/extractions/jobs", auth_mode="auto")
+
+    def get_job(self, job_id: str) -> Dict[str, Any]:
+        return self._http.request("GET", f"/api/v1/extractions/jobs/{job_id}", auth_mode="auto")
+
+    def wait_for_job(self, job_id: str, *, timeout_sec: int = 300, interval_sec: float = 3.0) -> Dict[str, Any]:
+        started = time.time()
+        while time.time() - started < timeout_sec:
+            result = self.get_job(job_id)
+            status = str(result.get("status", "")).lower()
+            if status == "completed":
+                return result
+            if status == "failed":
+                raise QueryMemoryError(result.get("error_message", "Extraction job failed"), 500, result)
+            time.sleep(interval_sec)
+        raise QueryMemoryError(f"Extraction job polling timed out after {timeout_sec}s", 408)
+
+    def download_csv(self, job_id: str) -> str:
+        resp = self._http.session.get(
+            f"{self._http.base_url}/api/v1/extractions/jobs/{job_id}/csv",
+            headers=self._http._headers(),
+            timeout=self._http.timeout,
+        )
+        resp.raise_for_status()
+        return resp.text
+
+
 class QueryMemoryClient:
     def __init__(
         self,
@@ -316,3 +386,4 @@ class QueryMemoryClient:
         self.knowledge = KnowledgeApi(http)
         self.agents = AgentsApi(http)
         self.web = WebApi(http)
+        self.extractions = ExtractionsApi(http)