phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Allow passing provider_registry to resolve_impls()

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-28 11:58:16 -07:00
parent 8a3b64d1be
commit b7d2b83d55
3 changed files with 13 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/distribution/server/server.py
View file

@ -26,7 +26,10 @@ from pydantic import BaseModel, ValidationError
from termcolor import cprint
from typing_extensions import Annotated
from llama_stack.distribution.distribution import builtin_automatically_routed_apis
from llama_stack.distribution.distribution import (
builtin_automatically_routed_apis,
get_provider_registry,
)
from llama_stack.providers.utils.telemetry.tracing import (
end_trace,
@ -276,7 +279,7 @@ def main(
app = FastAPI()
impls = asyncio.run(resolve_impls(config))
impls = asyncio.run(resolve_impls(config, get_provider_registry()))
if Api.telemetry in impls:
setup_logger(impls[Api.telemetry])