(deps commandDeps)
| 28 | } |
| 29 | |
| 30 | func newProviderModelsListCommand(deps commandDeps) *cobra.Command { |
| 31 | var sourceID string |
| 32 | var refresh bool |
| 33 | var includeStale bool |
| 34 | cmd := &cobra.Command{ |
| 35 | Use: "list [provider]", |
| 36 | Short: "List provider model catalog entries", |
| 37 | Args: optionalProviderArg(), |
| 38 | RunE: func(cmd *cobra.Command, args []string) error { |
| 39 | providerID := providerArgValue(args) |
| 40 | client, err := clientFromDeps(deps) |
| 41 | if err != nil { |
| 42 | return err |
| 43 | } |
| 44 | record, err := client.ListProviderModels(cmd.Context(), ProviderModelListQuery{ |
| 45 | ProviderID: providerID, |
| 46 | SourceID: sourceID, |
| 47 | Refresh: refresh, |
| 48 | IncludeStale: includeStale, |
| 49 | }) |
| 50 | if err != nil { |
| 51 | return err |
| 52 | } |
| 53 | return writeCommandOutput(cmd, providerModelListBundle(record)) |
| 54 | }, |
| 55 | } |
| 56 | cmd.Flags().StringVar(&sourceID, "source", "", "Filter by catalog source id") |
| 57 | cmd.Flags().BoolVar(&refresh, "refresh", false, "Refresh sources before listing models") |
| 58 | cmd.Flags().BoolVar(&includeStale, "include-stale", false, "Include stale source rows") |
| 59 | return cmd |
| 60 | } |
| 61 | |
| 62 | func newProviderModelsRefreshCommand(deps commandDeps) *cobra.Command { |
| 63 | var sourceID string |
no test coverage detected