agent-lab-sdk 0.1.40__tar.gz → 0.1.41__tar.gz

{agent_lab_sdk-0.1.40 → agent_lab_sdk-0.1.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.40
+Version: 0.1.41
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved

{agent_lab_sdk-0.1.40 → agent_lab_sdk-0.1.41}/agent_lab_sdk/langgraph/checkpoint/agw_saver.py

@@ -220,24 +220,33 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
                 self._client = None
 
     # ----------------------- universal dump/load ---------------------
-    # def _safe_dump(self, obj: Any) -> Any:
-    #     """self.serde.dump → гарантированная JSON-строка."""
-    #     dumped = self.serde.dumps(obj)
-    #     if isinstance(dumped, (bytes, bytearray)):
-    #         return base64.b64encode(dumped).decode()  # str
-    #     return dumped  # уже json-совместимо
-
     def _safe_dump(self, obj: Any) -> Any:
-        """bytes → python-object; fallback base64 для реально бинарных данных."""
-        dumped = self.serde.dumps(obj)
+        """
+        JSON-first сериализация:
+        1) Пытаемся через self.serde.dumps(obj).
+           - Если вернул bytes: пробуем декодировать в JSON-строку и распарсить.
+           - Если не JSON/не UTF-8 — заворачиваем как base64-строку.
+           - Если вернул dict/list/scalar — они уже JSON-совместимы.
+        2) Если self.serde.dumps(obj) бросает исключение (например, для Send),
+           делаем типизированный фолбэк {"type": str, "blob": base64 | None}.
+        """
+        try:
+            dumped = self.serde.dumps(obj)
+        except Exception:
+            # typed fallback (как рекомендуют в LangGraph для нетривиальных типов)
+            # https://langchain-ai.github.io/langgraph/reference/checkpoints/
+            try:
+                t, b = self.serde.dumps_typed(obj)
+            except Exception:
+                # крайний случай: строковое представление
+                t, b = type(obj).__name__, str(obj).encode()
+            return {"type": t, "blob": base64.b64encode(b).decode() if b is not None else None}
+
         if isinstance(dumped, (bytes, bytearray)):
             try:
-                # 1) bytes → str
                 s = dumped.decode()
-                # 2) str JSON → python (list/dict/scalar)
                 return orjson.loads(s)
             except (UnicodeDecodeError, orjson.JSONDecodeError):
-                # не UTF-8 или не JSON → base64
                 return base64.b64encode(dumped).decode()
         return dumped
 
@@ -308,16 +317,6 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         except Exception:
             return obj
 
-    # def _safe_load(self, obj: Any) -> Any:
-    #     """Обратная операция к _safe_dump."""
-    #     if isinstance(obj, str):
-    #         try:
-    #             return self.serde.load(base64.b64decode(obj))
-    #         except Exception:
-    #             # не base64 — обычная строка
-    #             return self.serde.load(obj)
-    #     return self.serde.load(obj)
-
     # ----------------------- config <-> api --------------------------
     def _to_api_config(self, cfg: RunnableConfig | None) -> Dict[str, Any]:
         if not cfg:

{agent_lab_sdk-0.1.40 → agent_lab_sdk-0.1.41}/agent_lab_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.40
+Version: 0.1.41
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved

{agent_lab_sdk-0.1.40 → agent_lab_sdk-0.1.41}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agent-lab-sdk"
-version = "0.1.40"
+version = "0.1.41"
 description = "SDK для работы с Agent Lab"
 readme = "README.md"
 license = { text = "Proprietary and Confidential — All Rights Reserved" }