agent-lab-sdk 0.1.11__py3-none-any.whl → 0.1.12__py3-none-any.whl

agent_lab_sdk/langgraph/checkpoint/agw_saver.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-import json
+import orjson
 from random import random
 from langgraph.checkpoint.serde.types import ChannelProtocol
 import asyncio
@@ -9,7 +9,7 @@ from contextlib import asynccontextmanager
 from typing import Any, AsyncIterator, Dict, Iterator, Optional, Sequence, Tuple
 import logging
 
-import requests
+import httpx
 from langchain_core.runnables import RunnableConfig
 
 from langgraph.checkpoint.base import (
@@ -40,7 +40,7 @@ except NameError:  # pragma: no cover
 
 
 class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
-    """Persist checkpoints in Agent-Gateway с помощью `requests` + threads."""
+    """Persist checkpoints in Agent-Gateway с помощью `httpx` async client."""
 
     # ---------------------------- init / ctx -------------------------
     def __init__(
@@ -51,13 +51,12 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         timeout: int | float = 10,
         api_key: str | None = None,
         extra_headers: Dict[str, str] | None = None,
+        verify: bool = True,
     ):
         super().__init__(serde=serde)
         self.base_url = base_url.rstrip("/")
         self.timeout = timeout
-        self._session = requests.Session()
-        self.loop = asyncio.get_running_loop()
-
+        
         self.headers: Dict[str, str] = {
             "Accept": "application/json",
             "Content-Type": "application/json",
@@ -66,12 +65,20 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             self.headers.update(extra_headers)
         if api_key:
             self.headers["Authorization"] = f"Bearer {api_key}"
+            
+        self._client = httpx.AsyncClient(
+            base_url=self.base_url,
+            headers=self.headers,
+            timeout=self.timeout,
+            verify=verify,
+            trust_env=True
+        )
 
     async def __aenter__(self):  # noqa: D401
         return self
 
     async def __aexit__(self, exc_type, exc, tb):  # noqa: D401
-        await asyncio.to_thread(self._session.close)
+        await self._client.aclose()
 
     # ----------------------- universal dump/load ---------------------
     # def _safe_dump(self, obj: Any) -> Any:
@@ -89,15 +96,15 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
                 # 1) bytes → str
                 s = dumped.decode()
                 # 2) str JSON → python (list/dict/scalar)
-                return json.loads(s)
-            except (UnicodeDecodeError, json.JSONDecodeError):
+                return orjson.loads(s)
+            except (UnicodeDecodeError, orjson.JSONDecodeError):
                 # не UTF-8 или не JSON → base64
                 return base64.b64encode(dumped).decode()
         return dumped
 
     def _safe_load(self, obj: Any) -> Any:
         if isinstance(obj, (dict, list)):          # уже распакованный JSON
-            return self.serde.loads(json.dumps(obj, ensure_ascii=False).encode())
+            return self.serde.loads(orjson.dumps(obj))
         if isinstance(obj, str):
             # сначала plain JSON-строка
             try:
@@ -195,18 +202,12 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         return self._safe_load(md)
 
     # ------------------------ HTTP wrapper ---------------------------
-    async def _http(self, m: str, path: str, **kw) -> requests.Response:
-        url = f"{self.base_url}{path}"
-        hdr = {**self.headers, **kw.pop("headers", {})}
-
+    async def _http(self, method: str, path: str, **kw) -> httpx.Response:
         if "json" in kw:
             payload = kw.pop("json")
-            kw["data"] = json.dumps(payload, ensure_ascii=False, separators=(",", ":"))
-            logger.info(kw["data"])
-
-        return await asyncio.to_thread(
-            self._session.request, m, url, headers=hdr, timeout=self.timeout, **kw
-        )
+            logger.info(orjson.dumps(payload).decode())
+            
+        return await self._client.request(method, path, **kw)
 
     # -------------------- api -> CheckpointTuple ----------------------
     def _to_tuple(self, node: Dict[str, Any]) -> CheckpointTuple:
@@ -378,4 +379,4 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         try:
             yield saver
         finally:
-            await asyncio.to_thread(saver._session.close)
+            await saver._client.aclose()

{agent_lab_sdk-0.1.11.dist-info → agent_lab_sdk-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.11
+Version: 0.1.12
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -23,6 +23,8 @@ Requires-Dist: langgraph~=0.4.1
 Requires-Dist: langchain_gigachat
 Requires-Dist: prometheus-client
 Requires-Dist: langchain
+Requires-Dist: httpx~=0.27.2
+Requires-Dist: orjson>=3.9.7
 Dynamic: license-file
 
 # Agent Lab SDK

{agent_lab_sdk-0.1.11.dist-info → agent_lab_sdk-0.1.12.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=ALhZW-7XUIVsbZVjCRVhZcMmcYj_f9k3UU-owaq5vxM,13797
+agent_lab_sdk/langgraph/checkpoint/agw_saver.py,sha256=0A5shpjAvkCTHcRlUWMrXJCOLpsXGIpg5FPqRZapNaI,13688
 agent_lab_sdk/llm/__init__.py,sha256=Yo9MbYdHS1iX05A9XiJGwWN1Hm4IARGav9mNFPrtDeA,376
 agent_lab_sdk/llm/agw_token_manager.py,sha256=_bPPI8muaEa6H01P8hHQOJHiiivaLd8N_d3OT9UT_80,4787
 agent_lab_sdk/llm/gigachat_token_manager.py,sha256=TPA8cb0ypdWtRTI5C7GItL9jbLt93vR-Ijf2yMrOytQ,7921
@@ -12,8 +12,8 @@ agent_lab_sdk/schema/__init__.py,sha256=s91PH3Fj1eOq75gtOTiObSDiQdtpixg5Nhs3isBm
 agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCvyFF5Mw,638
 agent_lab_sdk/storage/__init__.py,sha256=ik1_v1DMTwehvcAEXIYxuvLuCjJCa3y5qAuJqoQpuSA,81
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
-agent_lab_sdk-0.1.11.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.11.dist-info/METADATA,sha256=LdYKkRDXPhIsA-lkNsP-c_Eycq02y-nqceKDwrZv_ok,1749
-agent_lab_sdk-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.11.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.11.dist-info/RECORD,,
+agent_lab_sdk-0.1.12.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.12.dist-info/METADATA,sha256=cSLJxrOWJOZP9aDEHJFyyQMtKDiPsib6VKY0YLT_a_E,1807
+agent_lab_sdk-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.12.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.12.dist-info/RECORD,,