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

agent_lab_sdk/storage/__init__.py

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

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

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

@@ -12,10 +12,11 @@ agent_lab_sdk/metrics/metrics.py,sha256=_XTT9vMG7T0u_D2pL371wm8GoBU5fodJ45D2RACn
 agent_lab_sdk/schema/__init__.py,sha256=bHSyXQYkcB9fWBlziWodXR_IzC5nKrdKzrCpyVWNY9o,521
 agent_lab_sdk/schema/input_types.py,sha256=e75nRW7Dz_RHk5Yia8DkFfbqMafsLQsQrJPfzQhpIBw,9123
 agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCvyFF5Mw,638
-agent_lab_sdk/storage/__init__.py,sha256=ik1_v1DMTwehvcAEXIYxuvLuCjJCa3y5qAuJqoQpuSA,81
+agent_lab_sdk/storage/__init__.py,sha256=HAtUoqg3k0irqPMewayadVA9aXJOmYSxRr6a5J1scT0,174
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
-agent_lab_sdk-0.1.39.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.39.dist-info/METADATA,sha256=UdFdawrHLbhJShQ_DcUfVJzVRp5pdrTlZiZDSUsq4xk,17911
-agent_lab_sdk-0.1.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.39.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.39.dist-info/RECORD,,
+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,,