INNER CODE UNIT · Go

main

kaito-project/kaito · presets/workspace/generator/update_model_catalog/main.go:36

func main() {
	repos := flag.String("repos", "", "Comma-separated list of repos to add or update (default: all existing)")
	dryRun := flag.Bool("dry-run", false, "Show changes without writing to file")
	token := flag.String("token", os.Getenv("HF_TOKEN"), "HuggingFace API token (default: $HF_TOKEN)")
	catalog := flag.String("catalog", filepath.Join("presets", "workspace", "models", "model_catalog.yaml"), "Path to model_catalog.yaml")
	flag.Parse()

	var repoList []string
	if *repos != "" {
		for _, r := range strings.Split(*repos, ",") {
			r = strings.TrimSpace(r)
			if r != "" {
				repoList = append(repoList, r)
			}
		}
	}

	if err := updateModelCatalog(repoList, *dryRun, *token, *catalog); err != nil {

View source record →