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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.39
+Version: 0.1.41
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -220,7 +220,32 @@ from agent_lab_sdk.storage import store_file_in_sd_asset
 store_file_in_storage("my-agent-name-filename.png", file_b64, "giga-agents")
 ```
 
-### 4.2 AGW Checkpointer
+### 4.2 V2 File Upload
+
+Новый v2 API для загрузки файлов через Agent Gateway с поддержкой бинарных данных и автоматическим выбором сервиса хранения.
+
+```python
+from agent_lab_sdk.storage import upload_file, FileUploadResponse
+
+# Загрузка из байтов
+with open("document.pdf", "rb") as f:
+    file_bytes = f.read()
+result: FileUploadResponse = upload_file("document.pdf", file_bytes)
+
+# Результат - Pydantic модель с информацией о файле
+print(f"File ID: {result.id}")
+print(f"Absolute Path: {result.absolute_path}")
+print(f"Storage: {result.storage}")
+```
+
+#### Переменные окружения для V2 Upload
+
+| Переменная                 | Описание                           | Значение по умолчанию |
+| -------------------------- | ---------------------------------- | --------------------- |
+| `AGENT_SERVICE_NAME`       | Имя сервиса агента (обязательно)   | -                     |
+| `STORAGE_PROVIDER_AGW_URL` | URL Agent Gateway                  | `http://localhost`    |
+
+### 4.3 AGW Checkpointer
 
 AGW поддерживает langgraph checkpoint API и в SDK представлен `AsyncAGWCheckpointSaver`, который позволяет сохранять состояние графа в AGW напрямую. 
 

{agent_lab_sdk-0.1.39 → agent_lab_sdk-0.1.41}/README.md

@@ -190,7 +190,32 @@ from agent_lab_sdk.storage import store_file_in_sd_asset
 store_file_in_storage("my-agent-name-filename.png", file_b64, "giga-agents")
 ```
 
-### 4.2 AGW Checkpointer
+### 4.2 V2 File Upload
+
+Новый v2 API для загрузки файлов через Agent Gateway с поддержкой бинарных данных и автоматическим выбором сервиса хранения.
+
+```python
+from agent_lab_sdk.storage import upload_file, FileUploadResponse
+
+# Загрузка из байтов
+with open("document.pdf", "rb") as f:
+    file_bytes = f.read()
+result: FileUploadResponse = upload_file("document.pdf", file_bytes)
+
+# Результат - Pydantic модель с информацией о файле
+print(f"File ID: {result.id}")
+print(f"Absolute Path: {result.absolute_path}")
+print(f"Storage: {result.storage}")
+```
+
+#### Переменные окружения для V2 Upload
+
+| Переменная                 | Описание                           | Значение по умолчанию |
+| -------------------------- | ---------------------------------- | --------------------- |
+| `AGENT_SERVICE_NAME`       | Имя сервиса агента (обязательно)   | -                     |
+| `STORAGE_PROVIDER_AGW_URL` | URL Agent Gateway                  | `http://localhost`    |
+
+### 4.3 AGW Checkpointer
 
 AGW поддерживает langgraph checkpoint API и в SDK представлен `AsyncAGWCheckpointSaver`, который позволяет сохранять состояние графа в AGW напрямую. 
 

{agent_lab_sdk-0.1.39 → 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.41/agent_lab_sdk/storage/__init__.py

@@ -0,0 +1,4 @@
+from .storage import store_file_in_sd_asset
+from .storage_v2 import upload_file, FileUploadResponse
+
+__all__ = ["store_file_in_sd_asset", "upload_file", "FileUploadResponse"]

agent_lab_sdk-0.1.41/agent_lab_sdk/storage/storage_v2.py

@@ -0,0 +1,66 @@
+import os
+from typing import Optional
+import requests
+from pydantic import BaseModel
+
+
+class FileUploadResponse(BaseModel):
+    """Ответ от сервиса загрузки файлов"""
+    id: Optional[str] = None
+    bucket: Optional[str] = None
+    key: Optional[str] = None
+    storage: Optional[str] = None
+    absolute_path: str
+
+
+def upload_file(filename: str, file_bytes: bytes) -> Optional[FileUploadResponse]:
+    """
+    Загружает файл в бинарном формате через v2 API и возвращает информацию о загруженном файле.
+
+    Args:
+        filename: имя файла
+        file_bytes: содержимое файла в виде байтов
+
+    Returns:
+        FileUploadResponse с информацией о файле или None в случае ошибки
+    """
+
+    ai_agent = os.getenv("AGENT_SERVICE_NAME")
+    if not ai_agent:
+        raise ValueError("AGENT_SERVICE_NAME environment variable is required")
+
+    gateway_url = os.getenv("STORAGE_PROVIDER_AGW_URL", "http://localhost")
+    url = f"{gateway_url}/files/v2/upload"
+
+    headers = {
+        "x-agent-id": ai_agent,
+        "Content-Type": "application/octet-stream"
+    }
+
+    params = {
+        "name": filename
+    }
+
+    try:
+        response = requests.post(
+            url,
+            headers=headers,
+            params=params,
+            data=file_bytes
+        )
+        response.raise_for_status()
+
+        if response.status_code == 200:
+            data = response.json()
+            return FileUploadResponse(
+                id=data.get("id", ""),
+                bucket=data.get("bucket", ""),
+                key=data.get("key", ""),
+                storage=data.get("storage", ""),
+                absolute_path=data.get("absolutePath", "")
+            )
+
+    except requests.exceptions.RequestException as e:
+        print(f"Ошибка при загрузке файла через v2 API: {e}")
+
+    return None

{agent_lab_sdk-0.1.39 → 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.39
+Version: 0.1.41
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved
@@ -220,7 +220,32 @@ from agent_lab_sdk.storage import store_file_in_sd_asset
 store_file_in_storage("my-agent-name-filename.png", file_b64, "giga-agents")
 ```
 
-### 4.2 AGW Checkpointer
+### 4.2 V2 File Upload
+
+Новый v2 API для загрузки файлов через Agent Gateway с поддержкой бинарных данных и автоматическим выбором сервиса хранения.
+
+```python
+from agent_lab_sdk.storage import upload_file, FileUploadResponse
+
+# Загрузка из байтов
+with open("document.pdf", "rb") as f:
+    file_bytes = f.read()
+result: FileUploadResponse = upload_file("document.pdf", file_bytes)
+
+# Результат - Pydantic модель с информацией о файле
+print(f"File ID: {result.id}")
+print(f"Absolute Path: {result.absolute_path}")
+print(f"Storage: {result.storage}")
+```
+
+#### Переменные окружения для V2 Upload
+
+| Переменная                 | Описание                           | Значение по умолчанию |
+| -------------------------- | ---------------------------------- | --------------------- |
+| `AGENT_SERVICE_NAME`       | Имя сервиса агента (обязательно)   | -                     |
+| `STORAGE_PROVIDER_AGW_URL` | URL Agent Gateway                  | `http://localhost`    |
+
+### 4.3 AGW Checkpointer
 
 AGW поддерживает langgraph checkpoint API и в SDK представлен `AsyncAGWCheckpointSaver`, который позволяет сохранять состояние графа в AGW напрямую. 
 

{agent_lab_sdk-0.1.39 → agent_lab_sdk-0.1.41}/agent_lab_sdk.egg-info/SOURCES.txt

@@ -21,4 +21,5 @@ agent_lab_sdk/schema/__init__.py
 agent_lab_sdk/schema/input_types.py
 agent_lab_sdk/schema/log_message.py
 agent_lab_sdk/storage/__init__.py
-agent_lab_sdk/storage/storage.py
+agent_lab_sdk/storage/storage.py
+agent_lab_sdk/storage/storage_v2.py

{agent_lab_sdk-0.1.39 → 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.39"
+version = "0.1.41"
 description = "SDK для работы с Agent Lab"
 readme = "README.md"
 license = { text = "Proprietary and Confidential — All Rights Reserved" }

agent_lab_sdk-0.1.39/agent_lab_sdk/storage/__init__.py

@@ -1,3 +0,0 @@
-from .storage import store_file_in_sd_asset
-
-__all__ = ["store_file_in_sd_asset"]