memorysync 1.8.0__tar.gz → 1.9.0__tar.gz

{memorysync-1.8.0 → memorysync-1.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: memorysync
-Version: 1.8.0
+Version: 1.9.0
 Summary: Official Python client for the MemorySync API.
 Project-URL: Homepage, https://memorysync.io
 Project-URL: Documentation, https://docs.memorysync.io

memorysync-1.9.0/src/memorysync/_version.py

@@ -0,0 +1 @@
+__version__ = "1.9.0"

{memorysync-1.8.0 → memorysync-1.9.0}/src/memorysync/client.py

@@ -612,6 +612,81 @@ class MemorySyncClient:
             return skipped
         return Memory.from_dict(raw)
 
+    # ── Bulk import ──────────────────────────────────────────────────
+
+    def create_import(
+        self,
+        file: Union[bytes, "IO[bytes]"],
+        *,
+        filename: str,
+        end_user_id: Optional[str] = None,
+        resume: bool = False,
+        continue_on_error: bool = False,
+    ) -> Dict[str, Any]:
+        """Upload a payload and get an import job back, without waiting for it.
+
+        ``bulk_add`` accepts fifty records per call because each one runs the
+        extraction pipeline. For a migration out of another system, upload the
+        whole file here and poll :meth:`get_import`.
+
+        ``resume`` requires ``client_ref`` on every record and is what makes
+        re-running the same file safe: records a previous run already stored come
+        back counted as ``already_imported`` rather than stored again.
+
+        The payload is JSONL, or JSON when ``filename`` ends in ``.json``. The
+        extension is how the server picks the reader, so it is required.
+        """
+        if not filename or not filename.strip():
+            raise ValidationError("filename is required so the server can pick a reader")
+        form = _strip_none(
+            {
+                "end_user_id": end_user_id,
+                "resume": "true" if resume else "false",
+                "continue_on_error": "true" if continue_on_error else "false",
+            }
+        )
+        return (
+            self._request(
+                "POST",
+                "/imports",
+                files={"file": (filename, file)},
+                data=form,
+                end_user_override=end_user_id,
+            )
+            or {}
+        )
+
+    def get_import(self, job_id: str) -> Dict[str, Any]:
+        """One import job's status, progress and per-outcome counters.
+
+        ``progress_percentage`` stays below 100 until the job is genuinely
+        finished, so reaching 100 means done.
+        """
+        if not job_id or not str(job_id).strip():
+            raise ValidationError("job_id is required")
+        return self._request("GET", f"/imports/{job_id}") or {}
+
+    def list_imports(
+        self, *, status: Optional[str] = None, limit: Optional[int] = None
+    ) -> Dict[str, Any]:
+        """Recent import jobs, newest first."""
+        return (
+            self._request(
+                "GET", "/imports", params={"status": status, "limit": limit}
+            )
+            or {}
+        )
+
+    def cancel_import(self, job_id: str) -> Dict[str, Any]:
+        """Ask the worker to stop between batches.
+
+        Records already imported stay imported, and the counters report how far it
+        reached. Cancelling a finished job is a no-op rather than an error.
+        """
+        if not job_id or not str(job_id).strip():
+            raise ValidationError("job_id is required")
+        return self._request("POST", f"/imports/{job_id}/cancel") or {}
+
     # ── Bulk edit ────────────────────────────────────────────────────
 
     def batch_update(
@@ -1296,6 +1371,61 @@ class AsyncMemorySyncClient:
 
     # ── Bulk edit ────────────────────────────────────────────────────
 
+    # ── Bulk import ──────────────────────────────────────────────────
+
+    async def create_import(
+        self,
+        file: Union[bytes, "IO[bytes]"],
+        *,
+        filename: str,
+        end_user_id: Optional[str] = None,
+        resume: bool = False,
+        continue_on_error: bool = False,
+    ) -> Dict[str, Any]:
+        """Async twin of :meth:`MemorySyncClient.create_import`."""
+        if not filename or not filename.strip():
+            raise ValidationError("filename is required so the server can pick a reader")
+        form = _strip_none(
+            {
+                "end_user_id": end_user_id,
+                "resume": "true" if resume else "false",
+                "continue_on_error": "true" if continue_on_error else "false",
+            }
+        )
+        return (
+            await self._request(
+                "POST",
+                "/imports",
+                files={"file": (filename, file)},
+                data=form,
+                end_user_override=end_user_id,
+            )
+            or {}
+        )
+
+    async def get_import(self, job_id: str) -> Dict[str, Any]:
+        """Async twin of :meth:`MemorySyncClient.get_import`."""
+        if not job_id or not str(job_id).strip():
+            raise ValidationError("job_id is required")
+        return await self._request("GET", f"/imports/{job_id}") or {}
+
+    async def list_imports(
+        self, *, status: Optional[str] = None, limit: Optional[int] = None
+    ) -> Dict[str, Any]:
+        """Async twin of :meth:`MemorySyncClient.list_imports`."""
+        return (
+            await self._request(
+                "GET", "/imports", params={"status": status, "limit": limit}
+            )
+            or {}
+        )
+
+    async def cancel_import(self, job_id: str) -> Dict[str, Any]:
+        """Async twin of :meth:`MemorySyncClient.cancel_import`."""
+        if not job_id or not str(job_id).strip():
+            raise ValidationError("job_id is required")
+        return await self._request("POST", f"/imports/{job_id}/cancel") or {}
+
     async def batch_update(
         self,
         items: Sequence[Union[BatchUpdateItem, Dict[str, Any]]],

memorysync-1.8.0/src/memorysync/_version.py

@@ -1 +0,0 @@
-__version__ = "1.8.0"