agent-lab-sdk 0.1.11__tar.gz → 0.1.12__tar.gz

{agent_lab_sdk-0.1.11 → agent_lab_sdk-0.1.12}/PKG-INFO

@@ -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 → agent_lab_sdk-0.1.12}/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 → agent_lab_sdk-0.1.12}/agent_lab_sdk.egg-info/PKG-INFO

@@ -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 → agent_lab_sdk-0.1.12}/agent_lab_sdk.egg-info/requires.txt

@@ -3,3 +3,5 @@ langgraph~=0.4.1
 langchain_gigachat
 prometheus-client
 langchain
+httpx~=0.27.2
+orjson>=3.9.7

{agent_lab_sdk-0.1.11 → agent_lab_sdk-0.1.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agent-lab-sdk"
-version = "0.1.11"
+version = "0.1.12"
 description = "SDK для работы с Agent Lab"
 readme = "README.md"
 license = { text = "Proprietary and Confidential — All Rights Reserved" }
@@ -13,7 +13,7 @@ authors = [
 ]
 requires-python = ">=3.11,<4.0"
 dependencies = [
-  "requests", "langgraph~=0.4.1", "langchain_gigachat", "prometheus-client", "langchain"
+  "requests", "langgraph~=0.4.1", "langchain_gigachat", "prometheus-client", "langchain", "httpx~=0.27.2", "orjson>=3.9.7"
 ]
 keywords = ["agent", "lab", "sdk"]
 classifiers = [