Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Config ¶
type Config struct {
APIKey string
BaseURL string
Model string
MaxRetries int
RetryDelay time.Duration
}
Config holds embedding service configuration
type EmbeddingResponse ¶
type EmbeddingResponse struct {
Object string `json:"object"`
Data []struct {
Object string `json:"object"`
Index int `json:"index"`
Embedding []float32 `json:"embedding"`
} `json:"data"`
Model string `json:"model"`
Usage struct {
PromptTokens int `json:"prompt_tokens"`
TotalTokens int `json:"total_tokens"`
} `json:"usage"`
}
EmbeddingResponse represents OpenAI embedding API response
type EmbeddingService ¶
type EmbeddingService struct {
// contains filtered or unexported fields
}
EmbeddingService provides text embedding generation
func DefaultService ¶
func DefaultService() *EmbeddingService
DefaultService returns default embedding service using system config
func NewEmbeddingService ¶
func NewEmbeddingService(cfg Config) *EmbeddingService
NewEmbeddingService creates a new embedding service
func NewServiceFromChannel ¶
func NewServiceFromChannel(channel *model.Channel) (*EmbeddingService, error)
NewServiceFromChannel creates an EmbeddingService from a Channel
func (*EmbeddingService) GetEmbedding ¶
GetEmbedding generates embedding for a single text
func (*EmbeddingService) GetEmbeddings ¶
GetEmbeddings generates embeddings for multiple texts
Click to show internal directories.
Click to hide internal directories.