Skip to content

Commit 86e4e68

Browse files
committed
ci: fix per-model-type API doc generation
1 parent 98554f8 commit 86e4e68

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.github/workflows/api-docs.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ jobs:
3838
python app/cli/cli.py export-model-apis --model-type anoncat --add-training-apis --add-evaluation-apis --add-previews-apis --exclude-metacat-training --exclude-unsupervised-training
3939
python app/cli/cli.py export-model-apis --model-type transformers_deid --add-training-apis --add-evaluation-apis --add-previews-apis --exclude-metacat-training --exclude-unsupervised-training
4040
python app/cli/cli.py export-model-apis --model-type huggingface_ner --add-training-apis --add-evaluation-apis --add-previews-apis --exclude-metacat-training
41-
python app/cli/cli.py export-model-apis --model-type huggingface_llm
41+
python app/cli/cli.py export-model-apis --model-type huggingface_llm --add-training-apis --add-evaluation-apis --exclude-metacat-training
4242
python app/cli/cli.py export-openapi-spec
4343
git fetch origin gh-pages:gh-pages
4444
git checkout gh-pages

app/cli/cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -451,9 +451,9 @@ def generate_api_doc_per_model(
451451
config.ENABLE_PREVIEWS_APIS = "true" if add_previews_apis else "false"
452452
config.AUTH_USER_ENABLED = "true" if add_user_authentication else "false"
453453

454-
model_service_dep = ModelServiceDep(model_type, config, model_name or model_type)
454+
model_service_dep = ModelServiceDep(model_type, config, model_name or model_type.value)
455455
cms_globals.model_service_dep = model_service_dep
456-
doc_name = f"{model_name or model_type}_model_apis.json"
456+
doc_name = f"{model_type.value}_model_apis.json"
457457

458458
if model_type == ModelType.HUGGINGFACE_LLM:
459459
app = get_generative_server(config)

0 commit comments

Comments
 (0)