phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 09:53:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: use string annotations for S3Client type hints

Browse source 
Remove future annotations import and use quoted string annotations for S3Client to avoid import issues.

Changes:
o Remove __future__ annotations import
o Use "S3Client" string annotations in type hints

Signed-off-by: Derek Higgins <derekh@redhat.com>
This commit is contained in:
Derek Higgins 2025-12-01 11:55:34 +00:00
parent a7c7c72467
commit 13fcf75439
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/llama_stack/providers/remote/files/s3/files.py
View file

@ -4,8 +4,6 @@
# This source code is licensed under the terms described in the LICENSE file in # This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree. # the root directory of this source tree.
from __future__ import annotations
import uuid import uuid
from datetime import UTC, datetime from datetime import UTC, datetime
from typing import TYPE_CHECKING, Annotated, Any, cast from typing import TYPE_CHECKING, Annotated, Any, cast
@ -39,7 +37,7 @@ from .config import S3FilesImplConfig
# TODO: provider data for S3 credentials # TODO: provider data for S3 credentials
def _create_s3_client(config: S3FilesImplConfig) -> S3Client: def _create_s3_client(config: S3FilesImplConfig) -> "S3Client":
try: try:
s3_config = { s3_config = {
"region_name": config.region, "region_name": config.region,
@ -66,7 +64,7 @@ def _create_s3_client(config: S3FilesImplConfig) -> S3Client:
raise RuntimeError(f"Failed to initialize S3 client: {e}") from e raise RuntimeError(f"Failed to initialize S3 client: {e}") from e
async def _create_bucket_if_not_exists(client: S3Client, config: S3FilesImplConfig) -> None: async def _create_bucket_if_not_exists(client: "S3Client", config: S3FilesImplConfig) -> None:
try: try:
client.head_bucket(Bucket=config.bucket_name) client.head_bucket(Bucket=config.bucket_name)
except ClientError as e: except ClientError as e:
@ -192,7 +190,7 @@ class S3FilesImpl(Files):
pass pass
@property @property
def client(self) -> S3Client: def client(self) -> "S3Client":
assert self._client is not None, "Provider not initialized" assert self._client is not None, "Provider not initialized"
return self._client return self._client