fix linter errors

This commit is contained in:
Dinesh Yeduguru 2024-11-04 12:16:24 -08:00
parent 52344391b9
commit 8ff1b3af13
4 changed files with 9 additions and 11 deletions

View file

@ -147,7 +147,6 @@ a default SQLite store will be used.""",
)
class BuildConfig(BaseModel):
version: str = LLAMA_STACK_BUILD_CONFIG_VERSION
name: str

View file

@ -283,9 +283,7 @@ def main(
app = FastAPI()
# instantiate kvstore for storing and retrieving distribution metadata
if config.distribution_registry_store:
dist_kvstore = asyncio.run(
kvstore_impl(config.distribution_registry_store)
)
dist_kvstore = asyncio.run(kvstore_impl(config.distribution_registry_store))
else:
dist_kvstore = asyncio.run(
kvstore_impl(

View file

@ -18,6 +18,7 @@ from llama_stack.providers.utils.kvstore import KVStore
class DistributionRegistry(Protocol):
async def get(self, identifier: str) -> [RoutableObjectWithProvider]: ...
# The current data structure allows multiple objects with the same identifier but different providers.
# This is not ideal - we should have a single object that can be served by multiple providers,
# suggesting a data structure like (obj: Obj, providers: List[str]) rather than List[RoutableObjectWithProvider].