feat: add generation API endpoints

This commit is contained in:
QiuSW
2026-07-02 22:41:37 +08:00
parent 47e6aed7a8
commit 0f7798ec2f
17 changed files with 851 additions and 19 deletions
+8 -3
View File
@@ -23,8 +23,8 @@ REQUIRED_CAPABILITIES = {
}
def resolve_alias(operation_type: str, alias: str | None = None) -> ResolvedModel:
"""Resolve an external capability alias to a provider-ready model config."""
def resolve_model_alias(operation_type: str, alias: str | None = None) -> ModelAlias:
"""Resolve an external capability alias to an active ModelAlias row."""
required_capability = REQUIRED_CAPABILITIES.get(operation_type)
if required_capability is None:
raise AliasResolutionError(f"unsupported operation_type: {operation_type}")
@@ -52,4 +52,9 @@ def resolve_alias(operation_type: str, alias: str | None = None) -> ResolvedMode
f"alias {model_alias.alias} maps to model {ai_model.name} without "
f"{required_capability} capability"
)
return ai_model.to_resolved_model()
return model_alias
def resolve_alias(operation_type: str, alias: str | None = None) -> ResolvedModel:
"""Resolve an external capability alias to a provider-ready model config."""
return resolve_model_alias(operation_type, alias).ai_model.to_resolved_model()