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

provider_type -> provider_id ... less confusing

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-16 12:09:53 -07:00
parent ce6c868499
commit 6f5d9a3df8
11 changed files with 25 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama_toolchain/core/distribution.py
View file

@ -14,14 +14,7 @@ from llama_toolchain.memory.api import Memory
from llama_toolchain.safety.api import Safety
from llama_toolchain.telemetry.api import Telemetry
from .datatypes import (
Api,
ApiEndpoint,
DistributionSpec,
InlineProviderSpec,
ProviderSpec,
remote_provider_spec,
)
from .datatypes import Api, ApiEndpoint, ProviderSpec, remote_provider_spec
# These are the dependencies needed by the distribution server.
# `llama-toolchain` is automatically installed by the installation script.
@ -77,7 +70,7 @@ def api_providers() -> Dict[Api, Dict[str, ProviderSpec]]:
module = importlib.import_module(f"llama_toolchain.{name}.providers")
ret[api] = {
"remote": remote_provider_spec(api),
**{a.provider_type: a for a in module.available_providers()},
**{a.provider_id: a for a in module.available_providers()},
}
return ret