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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.39
+Version: 0.1.40
 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.40}/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.40/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.40/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.40}/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.40
 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.40}/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.40}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agent-lab-sdk"
-version = "0.1.39"
+version = "0.1.40"
 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"]