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

registry

Browse source
This commit is contained in:
Xi Yan 2024-12-26 18:34:00 -08:00
parent 27da763af9
commit 74de9bebd1
2 changed files with 25 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/distribution/store/tests/test_registry.py
View file

@ -8,11 +8,14 @@ import os
import pytest
import pytest_asyncio
from llama_stack.distribution.store import * # noqa F403
from llama_stack.apis.inference import Model
from llama_stack.apis.memory_banks import VectorMemoryBank
from llama_stack.distribution.store.registry import (
CachedDiskDistributionRegistry,
DiskDistributionRegistry,
)
from llama_stack.providers.utils.kvstore import kvstore_impl, SqliteKVStoreConfig
from llama_stack.distribution.datatypes import * # noqa F403
@pytest.fixture