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

ApiSurface -> Api

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-05 12:44:56 -07:00
parent 7890921e5c
commit 125fdb1b2a
13 changed files with 76 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

15
llama_toolchain/cli/distribution/configure.py
View file

@ -85,24 +85,21 @@ def configure_llama_distribution(dist: "Distribution", conda_env: str):
existing_config = yaml.safe_load(fp)
adapter_configs = {}
for api_surface, adapter in dist.adapters.items():
for api, adapter in dist.adapters.items():
if isinstance(adapter, PassthroughApiAdapter):
adapter_configs[api_surface.value] = adapter.dict()
adapter_configs[api.value] = adapter.dict()
else:
cprint(
f"Configuring API surface: {api_surface.value}", "white", attrs=["bold"]
)
cprint(f"Configuring API surface: {api.value}", "white", attrs=["bold"])
config_type = instantiate_class_type(adapter.config_class)
config = prompt_for_config(
config_type,
(
config_type(**existing_config["adapters"][api_surface.value])
if existing_config
and api_surface.value in existing_config["adapters"]
config_type(**existing_config["adapters"][api.value])
if existing_config and api.value in existing_config["adapters"]
else None
),
)
adapter_configs[api_surface.value] = {
adapter_configs[api.value] = {
"adapter_id": adapter.adapter_id,
**config.dict(),
}