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

Another round of simplification and clarity for models/shields/memory_banks stuff

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-09 19:19:26 -07:00
parent 73a0a34e39
commit b55034c0de
27 changed files with 454 additions and 444 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/distribution/routers/__init__.py
View file

@ -4,7 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import Any, List
from typing import Any
from llama_stack.distribution.datatypes import * # noqa: F403
from .routing_tables import (
@ -16,7 +16,6 @@ from .routing_tables import (
async def get_routing_table_impl(
api: Api,
registry: List[RoutableObject],
impls_by_provider_id: Dict[str, RoutedProtocol],
_deps,
) -> Any:
@ -28,7 +27,7 @@ async def get_routing_table_impl(
if api.value not in api_to_tables:
raise ValueError(f"API {api.value} not found in router map")
impl = api_to_tables[api.value](registry, impls_by_provider_id)
impl = api_to_tables[api.value](impls_by_provider_id)
await impl.initialize()
return impl