phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-30 11:59:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

wip

Browse source
This commit is contained in:
Sébastien Han 2025-05-13 20:47:03 +02:00
parent 749cbcca31
commit 1afd33c21e
No known key found for this signature in database
10 changed files with 509 additions and 59 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/remote/files/object/s3/persistence.py
View file

@ -21,7 +21,8 @@ class UploadSessionInfo(BaseModel):
upload_id: str
bucket: str
key: str
key: str # Original key for file reading
s3_key: str # S3 key for S3 operations
mime_type: str
size: int
url: str
@ -31,12 +32,12 @@ class UploadSessionInfo(BaseModel):
class S3FilesPersistence:
def __init__(self, kvstore: KVStore):
self._kvstore = kvstore
self._store = None
self._store: KVStore | None = None
async def _get_store(self) -> KVStore:
"""Get the kvstore instance, initializing it if needed."""
if self._store is None:
self._store = await anext(self._kvstore)
self._store = self._kvstore
return self._store
async def store_upload_session(
@ -47,6 +48,7 @@ class S3FilesPersistence:
upload_id=session_info.id,
bucket=bucket,
key=key,
s3_key=key,
mime_type=mime_type,
size=size,
url=session_info.url,