From 6dac6059ad88770d7a62038829d5f8be0133c9fe Mon Sep 17 00:00:00 2001 From: alvaro Date: Wed, 23 Jul 2025 12:15:05 +0200 Subject: [PATCH] Add prefix when getting objects from storage --- yepcode_run/api/yepcode_api.py | 4 ++-- yepcode_run/storage/yepcode_storage.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/yepcode_run/api/yepcode_api.py b/yepcode_run/api/yepcode_api.py index c514e27..72a0910 100644 --- a/yepcode_run/api/yepcode_api.py +++ b/yepcode_run/api/yepcode_api.py @@ -470,8 +470,8 @@ def create_module_version_alias( ) -> VersionedModuleAlias: return self._request("POST", f"/modules/{module_id}/aliases", {"data": data}) - def get_objects(self) -> List[StorageObject]: - response = self._request("GET", "/storage/objects") + def get_objects(self, params: Optional[Dict[str, Any]] = None) -> List[StorageObject]: + response = self._request("GET", "/storage/objects", {"params": params or {}}) return [StorageObject.from_dict(obj) for obj in response] def get_object(self, name: str) -> requests.Response: diff --git a/yepcode_run/storage/yepcode_storage.py b/yepcode_run/storage/yepcode_storage.py index 62e411e..6640d58 100644 --- a/yepcode_run/storage/yepcode_storage.py +++ b/yepcode_run/storage/yepcode_storage.py @@ -1,4 +1,4 @@ -from typing import List +from typing import List, Optional, Dict, Any from ..api.api_manager import YepCodeApiManager from ..api.types import CreateStorageObjectInput, StorageObject, YepCodeApiConfig @@ -25,5 +25,5 @@ def upload(self, name: str, file: bytes) -> StorageObject: def delete(self, name: str) -> None: return self._api.delete_object(name) - def list(self) -> List[StorageObject]: - return self._api.get_objects() + def list(self, **kwargs) -> List[StorageObject]: + return self._api.get_objects(kwargs if kwargs else None)