Method GetModel

pkg/modelsdev/store.go:219–245

GetModel returns a specific model by ID. The ID must carry both a provider and a model component; pass the result of [NewID], [ParseID], or a provider's [ID] method.

(ctx context.Context, id ID)

// provider and a model component; pass the result of [NewID], [ParseID],
// or a provider's [ID] method.
func (s *Store) GetModel(ctx context.Context, id ID) (*Model, error) {
	if !id.IsValid() {
		return nil, fmt.Errorf("invalid model ID: %q", id.String())
	}

	provider, err := s.getProvider(ctx, id.Provider)
	if err != nil {
		return nil, err
	}

	model, exists := provider.Models[id.Model]

	// For amazon-bedrock, try stripping region/inference profile prefixes.
	// Bedrock uses prefixes for cross-region inference profiles,
	// but models.dev stores models without these prefixes.
	if !exists && id.Provider == "amazon-bedrock" {
		if prefix, after, ok := strings.Cut(id.Model, "."); ok && bedrockRegionPrefixes[prefix] {
			model, exists = provider.Models[after]
		}
	}

	if !exists {
		return nil, fmt.Errorf("model %q not found in provider %q", id.Model, id.Provider)
	}

	return &model, nil
}

// loadDatabase loads the database from the local cache file or
// falls back to fetching from the models.dev API.