models

package

v1.2.0 Latest Latest Go to latest Published: May 2, 2026 License: MIT Imports: 8 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Model struct {
	ID          string `json:"id"`
	DisplayName string `json:"display_name"`
	Provider    string `json:"provider"`
}

Model represents an available LLM model.

type Registry struct {
	// contains filtered or unexported fields
}

Registry discovers and caches available models from provider APIs.

func NewRegistry(claudeAPIKey, mistralAPIKey string, log *slog.Logger) *Registry

NewRegistry creates a model registry with API keys for each provider.

func (r *Registry) ListAllModels(ctx context.Context) []Model

ListAllModels returns models from all configured providers, tagged with provider name.

func (r *Registry) ListModels(ctx context.Context, provider string) ([]Model, error)

ListModels returns available models for a provider, using a cached result if fresh.