mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-17 17:22:38 +00:00
fix for pre-commit
This commit is contained in:
parent
a379b8d98c
commit
7160ef74c6
3 changed files with 9 additions and 9 deletions
|
|
@ -105,12 +105,12 @@ async def register_resources(run_config: StackRunConfig, impls: dict[Api, Any]):
|
||||||
|
|
||||||
method = getattr(impls[api], register_method)
|
method = getattr(impls[api], register_method)
|
||||||
for obj in objects:
|
for obj in objects:
|
||||||
logger.debug(f"registering {rsrc.capitalize()} {obj} for provider {obj.provider_id}")
|
if hasattr(obj, "provider_id"):
|
||||||
|
|
||||||
# Do not register models on disabled providers
|
# Do not register models on disabled providers
|
||||||
if hasattr(obj, "provider_id") and (not obj.provider_id or obj.provider_id == "__disabled__"):
|
if not obj.provider_id or obj.provider_id == "__disabled__":
|
||||||
logger.debug(f"Skipping {rsrc.capitalize()} registration for disabled provider.")
|
logger.debug(f"Skipping {rsrc.capitalize()} registration for disabled provider.")
|
||||||
continue
|
continue
|
||||||
|
logger.debug(f"registering {rsrc.capitalize()} {obj} for provider {obj.provider_id}")
|
||||||
|
|
||||||
# we want to maintain the type information in arguments to method.
|
# we want to maintain the type information in arguments to method.
|
||||||
# instead of method(**obj.model_dump()), which may convert a typed attr to a dict,
|
# instead of method(**obj.model_dump()), which may convert a typed attr to a dict,
|
||||||
|
|
|
||||||
|
|
@ -43,7 +43,7 @@ def get_inference_providers() -> tuple[list[Provider], dict[str, list[ProviderMo
|
||||||
"openai",
|
"openai",
|
||||||
[
|
[
|
||||||
ProviderModelEntry(
|
ProviderModelEntry(
|
||||||
provider_model_id="openai/gpt-4o",
|
provider_model_id="gpt-4o",
|
||||||
model_type=ModelType.llm,
|
model_type=ModelType.llm,
|
||||||
)
|
)
|
||||||
],
|
],
|
||||||
|
|
|
||||||
|
|
@ -136,9 +136,9 @@ inference_store:
|
||||||
db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/open-benchmark}/inference_store.db
|
db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/open-benchmark}/inference_store.db
|
||||||
models:
|
models:
|
||||||
- metadata: {}
|
- metadata: {}
|
||||||
model_id: openai/gpt-4o
|
model_id: gpt-4o
|
||||||
provider_id: openai
|
provider_id: openai
|
||||||
provider_model_id: openai/gpt-4o
|
provider_model_id: gpt-4o
|
||||||
model_type: llm
|
model_type: llm
|
||||||
- metadata: {}
|
- metadata: {}
|
||||||
model_id: claude-3-5-sonnet-latest
|
model_id: claude-3-5-sonnet-latest
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue