wehub-resource-sync
65 lines
1.9 KiB
Bash
Executable File
65 lines
1.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Publish model catalog JSON files as GitHub Release assets.
|
|
#
|
|
# Usage:
|
|
# bash dev/publish_model_catalog.sh [--repo OWNER/REPO] [--tag TAG] [--catalog-dir DIR]
|
|
#
|
|
# Defaults:
|
|
# --repo mlflow/mlflow
|
|
# --tag model-catalog/latest
|
|
# --catalog-dir mlflow/utils/model_catalog
|
|
#
|
|
# The script creates the release if it doesn't exist, then uploads
|
|
# every *.json file in the catalog directory as a release asset,
|
|
# overwriting any existing asset with the same name.
|
|
|
|
set -euo pipefail
|
|
|
|
REPO="mlflow/mlflow"
|
|
TAG="model-catalog/latest"
|
|
CATALOG_DIR="mlflow/utils/model_catalog"
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--repo) REPO="$2"; shift 2 ;;
|
|
--tag) TAG="$2"; shift 2 ;;
|
|
--catalog-dir) CATALOG_DIR="$2"; shift 2 ;;
|
|
*) echo "Unknown option: $1" >&2; exit 1 ;;
|
|
esac
|
|
done
|
|
|
|
echo "Repo: $REPO"
|
|
echo "Tag: $TAG"
|
|
echo "Catalog dir: $CATALOG_DIR"
|
|
|
|
# Count JSON files
|
|
file_count=$(find "$CATALOG_DIR" -maxdepth 1 -name '*.json' | wc -l | tr -d ' ')
|
|
if [[ "$file_count" -eq 0 ]]; then
|
|
echo "Error: no JSON files found in $CATALOG_DIR" >&2
|
|
exit 1
|
|
fi
|
|
echo "Files: $file_count"
|
|
|
|
# Create release if it doesn't exist
|
|
if ! gh release view "$TAG" --repo "$REPO" >/dev/null; then
|
|
echo "Creating release $TAG ..."
|
|
gh release create "$TAG" \
|
|
--repo "$REPO" \
|
|
--title "Model Catalog" \
|
|
--notes "Per-provider model catalog files. Updated weekly by CI." \
|
|
--latest=false
|
|
fi
|
|
|
|
# Upload all JSON files as release assets.
|
|
# --clobber overwrites existing assets (delete + re-upload per file).
|
|
# There is a brief window where each asset is missing during the overwrite,
|
|
# but clients fall back to the bundled catalog so this is safe.
|
|
echo "Uploading $file_count files to release $TAG ..."
|
|
gh release upload "$TAG" \
|
|
--repo "$REPO" \
|
|
--clobber \
|
|
"$CATALOG_DIR"/*.json
|
|
|
|
echo "Done. Assets available at:"
|
|
echo " https://github.com/$REPO/releases/tag/$(echo "$TAG" | sed 's|/|%2F|g')"
|