agent-lab-sdk 0.1.40__py3-none-any.whl → 0.1.41__py3-none-any.whl

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.dist-info → agent_lab_sdk-0.1.41.dist-info}/METADATA

@@ -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.dist-info → agent_lab_sdk-0.1.41.dist-info}/RECORD

@@ -1,6 +1,6 @@
 agent_lab_sdk/__init__.py,sha256=1Dlmv-wuz1QuciymKtYtX7jXzr_fkeGTe7aENfEDl3E,108
 agent_lab_sdk/langgraph/checkpoint/__init__.py,sha256=DnKwR1LwbaQ3qhb124lE-tnojrUIVcCdNzHEHwgpL5M,86
-agent_lab_sdk/langgraph/checkpoint/agw_saver.py,sha256=Srf8RYcW34_u2s54ABl0Jqm-_Z1gBH97gKqVY7QrKOQ,25631
+agent_lab_sdk/langgraph/checkpoint/agw_saver.py,sha256=NFoRDTo-keZ0IGOJCWs2nuhPAPEfPOBFdhhNtv4Z0gQ,25874
 agent_lab_sdk/langgraph/checkpoint/serde.py,sha256=UTSYbTbhBeL1CAr-XMbaH3SSIx9TeiC7ak22duXvqkw,5175
 agent_lab_sdk/llm/__init__.py,sha256=Yo9MbYdHS1iX05A9XiJGwWN1Hm4IARGav9mNFPrtDeA,376
 agent_lab_sdk/llm/agw_token_manager.py,sha256=_bPPI8muaEa6H01P8hHQOJHiiivaLd8N_d3OT9UT_80,4787
@@ -15,8 +15,8 @@ agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCv
 agent_lab_sdk/storage/__init__.py,sha256=HAtUoqg3k0irqPMewayadVA9aXJOmYSxRr6a5J1scT0,174
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
 agent_lab_sdk/storage/storage_v2.py,sha256=ONseynX59xzWK17dfzxZvnii2rpz3Oo2Zo9Ck-lcGnw,1997
-agent_lab_sdk-0.1.40.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.40.dist-info/METADATA,sha256=CnbGjkYor6CXT0xxc4PBCd1n14yNLmGeyGYaw_IAYVQ,19099
-agent_lab_sdk-0.1.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.40.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.40.dist-info/RECORD,,
+agent_lab_sdk-0.1.41.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.41.dist-info/METADATA,sha256=ksFSStSOlPIsT5Xg-ATauthhOfeXWFsVCkfsPfoSOoc,19099
+agent_lab_sdk-0.1.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.41.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.41.dist-info/RECORD,,