embedding

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

type BedRockRequest struct {
	InputText  string `json:"inputText"`
	Dimensions int64  `json:"dimensions,omitempty"`
	Normalize  bool   `json:"normalize,omitempty"`
}

type BedRockResponse struct {
	Embedding           []float32 `json:"embedding"`
	InputTextTokenCount int       `json:"inputTextTokenCount"`
}

type BedrockClient interface {
	InvokeModel(ctx context.Context, params *bedrockruntime.InvokeModelInput, optFns ...func(*bedrockruntime.Options)) (*bedrockruntime.InvokeModelOutput, error)
}

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

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

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

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

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

type Runner interface {
	GetSchema() *schemapb.FunctionSchema
	GetOutputFields() []*schemapb.FieldSchema
	GetCollectionName() string
	GetFunctionTypeName() string
	GetFunctionName() string
	GetFunctionProvider() string
	Check(ctx context.Context) error

	MaxBatch() int
	ProcessInsert(ctx context.Context, inputs []*schemapb.FieldData) ([]*schemapb.FieldData, error)
	ProcessSearch(ctx context.Context, placeholderGroup *commonpb.PlaceholderGroup) (*commonpb.PlaceholderGroup, error)
	ProcessBulkInsert(ctx context.Context, inputs []storage.FieldData) (map[storage.FieldID]storage.FieldData, error)
}

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

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

type TextEmbeddingFunction struct {
	FunctionBase
	// contains filtered or unexported fields
}

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

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

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