cohere

type ChatOptions struct {
	// Thinking configures reasoning features.
	// See: https://docs.cohere.com/reference/chat#request.body.thinking
	Thinking *ThinkingConfig `json:"thinking,omitempty"`
}

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

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

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

type EmbeddingOptions struct {
	// InputType overrides the default "search_document". Values:
	// "search_document", "search_query", "classification", "clustering".
	InputType string `json:"inputType,omitempty"`

	// OutputDimension selects the embedding dimensionality. Only the
	// embed-v4 family supports this; Cohere accepts 256, 512, 1024,
	// 1536 (ai-sdk #0df64d6).
	OutputDimension int `json:"outputDimension,omitempty"`
}

type Options struct {
	// APIKey is the Cohere API key.
	APIKey string

	// BaseURL overrides the default API endpoint.
	BaseURL string

	// Headers are additional HTTP headers to send.
	Headers map[string]string
}

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

type ThinkingConfig struct {
	// Type can be "enabled" or "disabled". Default is "enabled".
	Type string `json:"type,omitempty"`

	// TokenBudget is the maximum number of tokens the model can use for thinking.
	// Must be a positive integer. The model will stop thinking if it reaches
	// the budget and proceed with the response.
	TokenBudget int `json:"tokenBudget,omitempty"`
}