mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-27 19:12:00 +00:00
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
from collections.abc import AsyncGenerator
|
|
|
|
import pytest
|
|
|
|
from llama_stack.providers.remote.files.object.s3.config import S3FilesImplConfig
|
|
from llama_stack.providers.remote.files.object.s3.s3_files import S3FilesAdapter
|
|
from llama_stack.providers.utils.kvstore import KVStore, kvstore_impl
|
|
from llama_stack.providers.utils.kvstore.config import SqliteKVStoreConfig
|
|
|
|
|
|
@pytest.fixture
|
|
def s3_config():
|
|
"""Create S3 configuration for MinIO."""
|
|
return S3FilesImplConfig(
|
|
aws_access_key_id="ROOTNAME",
|
|
aws_secret_access_key="CHANGEME123",
|
|
region_name="us-east-1",
|
|
endpoint_url="http://localhost:9000",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
async def kvstore() -> AsyncGenerator[KVStore, None]:
|
|
"""Create a SQLite KV store for testing."""
|
|
config = SqliteKVStoreConfig(
|
|
path=":memory:" # Use in-memory SQLite for tests
|
|
)
|
|
store = await kvstore_impl(config)
|
|
await store.initialize()
|
|
yield store
|
|
|
|
|
|
@pytest.fixture
|
|
async def s3_files(s3_config, kvstore) -> AsyncGenerator[S3FilesAdapter, None]:
|
|
"""Create S3FilesAdapter instance for testing."""
|
|
adapter = S3FilesAdapter(s3_config, kvstore)
|
|
await adapter.initialize()
|
|
yield adapter
|