cohere

type Config struct {
	APIKey     string
	Model      string
	BaseURL    string
	Timeout    time.Duration
	MaxRetries int
	BatchSize  int
}

type EmbeddingData struct {
	Embedding []float32 `json:"embedding"`
	Index     int       `json:"index"`
}

type EmbeddingRequest struct {
	Model     string   `json:"model"`
	Texts     []string `json:"texts"`
	InputType string   `json:"input_type,omitempty"`
	Truncate  string   `json:"truncate,omitempty"`
}

type EmbeddingResponse struct {
	ID         string          `json:"id"`
	Embeddings []EmbeddingData `json:"embeddings"`
	Model      string          `json:"model"`
	Texts      []string        `json:"texts"`
}

type ErrorResponse struct {
	Message string `json:"message"`
	Code    int    `json:"code"`
}

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