embed

type DocumentRequest struct {
	Ctx            context.Context `json:"-"`
	Model          Model           `json:"model"`
	DocumentChunks []string        `json:"document_chunks"`
}

type DocumentResponse struct {
	Embeddings [][]float64     `json:"embeddings"`
	Metadata   models.Metadata `json:"metadata,omitempty"`
}

type Embeder interface {
	Provider() string
	Embed(req *Request) (*Response, error)
	EmbedDocument(req *DocumentRequest) (*DocumentResponse, error)
}

type Model struct {
	Provider string `json:"provider"`
	Name     string `json:"name"`

	Type Type `json:"type,omitempty"`

	Description string `json:"description,omitempty"`

	InputMaxTokens   int `json:"input_max_tokens,omitempty"`
	OutputDimensions int `json:"output_dimensions,omitempty"`

	Config map[string]any `json:"config,omitempty"`
}

type Request struct {
	Ctx   context.Context `json:"-"`
	Model Model           `json:"model"`
	Texts []string        `json:"texts"`
}

type Response struct {
	Embeddings [][]float64     `json:"embeddings"`
	Metadata   models.Metadata `json:"metadata,omitempty"`
}

type Type string