fix for pre-commit

This commit is contained in:
Swapna Lekkala 2025-08-29 10:55:35 -07:00
parent a379b8d98c
commit 7160ef74c6
3 changed files with 9 additions and 9 deletions

View file

@ -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 not obj.provider_id or obj.provider_id == "__disabled__":
if hasattr(obj, "provider_id") and (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,

View file

@ -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,
) )
], ],

View file

@ -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