imperal-sdk 6.3.2__tar.gz → 6.3.3__tar.gz

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/CHANGELOG.md

@@ -2,6 +2,12 @@
 
 All notable changes to `imperal-sdk` are documented here.
 
+## 6.3.3 — 2026-10-20
+
+### Fixed
+- `apply_batches` sizes each part as it is sent (the blob's text and facts, its chunks' vectors) against
+  `APPLY_BATCH_BYTES` = 4 MB, half the vault's request cap: a first sync of a large repo no longer gets 413.
+
 ## 6.3.2 — 2026-10-20
 
 ### Security

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: imperal-sdk
-Version: 6.3.2
+Version: 6.3.3
 Summary: SDK for building Imperal Cloud extensions
 Author: Valentin Scerbacov, Imperal, Inc.
 License-Expression: Apache-2.0

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/sdk-reference.json

@@ -1,5 +1,5 @@
 {
-  "sdk_version": "6.3.2",
+  "sdk_version": "6.3.3",
   "symbols": {
     "ActionResult": {
       "action_vocab_safe": false,

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/sdk_claims.json

@@ -1,5 +1,5 @@
 {
-  "_sdk_version": "6.3.2",
+  "_sdk_version": "6.3.3",
   "constants": {
     "max_call_depth": {
       "counts_root": true,

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/src/imperal_sdk/__init__.py

@@ -1,7 +1,7 @@
 """Imperal Cloud SDK — build extensions for the Imperal platform."""
 from typing import TYPE_CHECKING
 
-__version__ = "6.3.2"
+__version__ = "6.3.3"
 
 # 5.2.2 (2026-06-11): the package root resolves its public surface lazily
 # (PEP 562). The eager imports pulled the HTTP transport (Context / client

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/src/imperal_sdk/code_index.py

@@ -1610,7 +1610,7 @@ def embed_query(text: str) -> bytes:
 # normalised and hashed, only blobs the repo does not hold are extracted (and their chunks embedded),
 # and the apply goes in size-bounded parts -- the last one stitches the links.
 
-APPLY_BATCH_BYTES = 8_000_000
+APPLY_BATCH_BYTES = 4_000_000      # one apply body as sent (JSON) -- half the vault's 8 MiB request cap
 READ_BATCH = 200
 
 
@@ -1618,11 +1618,17 @@ def apply_batches(files: list, blobs: list, chunks: list, limit: int = APPLY_BAT
     """One sync's apply split by size (a big first sync never goes as one HTTP body): each part carries
     its blobs, the files that point at them and their chunks; files of blobs the repo already holds go
     first. ``[{files, blobs, chunks}]`` -- the last part stitches."""
+    import json as _json
     parts: list = [{"files": [], "blobs": [], "chunks": []}]
     where: dict = {}
     size = 0
+    by_blob: dict = {}
+    for c in chunks:
+        by_blob.setdefault(c.get("blob"), []).append(c)
     for b in blobs:
-        n = len(str(b.get("text") or "").encode("utf-8"))
+        # the size as sent: the blob with its text and facts, and its chunks with their vectors
+        n = len(_json.dumps(b, default=str).encode("utf-8")) + sum(
+            len(_json.dumps(c, default=str).encode("utf-8")) for c in by_blob.get(b["hash"], ()))
         if parts[-1]["blobs"] and size + n > limit:
             parts.append({"files": [], "blobs": [], "chunks": []})
             size = 0

{imperal_sdk-6.3.2 → imperal_sdk-6.3.3}/tests/test_code_index_indexer.py

@@ -393,3 +393,19 @@ def test_a_remote_never_carries_credentials_however_it_is_written():
     assert clean_remote("bob:secret@host:r.git") == "host:r.git"
     assert clean_remote("https://[::1]:9/r") == "https://[::1]:9/r" and clean_remote("http://h:bad/r") == ""
     assert clean_remote("") == "" and clean_remote("a b") == ""
+
+
+def test_every_apply_part_fits_the_vault_body_cap_as_sent():
+    """Live 2026-10-20: a first sync of the kernel repo hit the vault's 8 MiB body cap (413) -- the parts
+    were sized by file text alone while each blob also carries its facts and its chunks' vectors."""
+    import json
+    from imperal_sdk import code_index as ci
+    blobs = [{"hash": f"h{i}", "text": "x" * 200_000,
+              "facts": {"defs": [{"name": f"d{j}", "qualname": f"q{j}" * 20} for j in range(4000)]}} for i in range(40)]
+    files = [{"path": f"f{i}.py", "blob": f"h{i}"} for i in range(40)]
+    chunks = [{"blob": f"h{i}", "vector": "A" * 2000, "start_line": k} for i in range(40) for k in range(50)]
+    parts = ci.apply_batches(files, blobs, chunks)
+    assert len(parts) > 1 and sum(len(p["blobs"]) for p in parts) == 40
+    for p in parts:
+        assert len(json.dumps(p).encode("utf-8")) <= ci.APPLY_BATCH_BYTES + 64_000
+    assert ci.APPLY_BATCH_BYTES <= 4_200_000