usecase

package

v2.10.0 Latest Latest Go to latest Published: Dec 16, 2025 License: AGPL-3.0 Imports: 37 Imported by: 0

Details

This section is empty.

This section is empty.

func FilterModelsByType(models []domain.ModelListItem, req *domain.ModelListReq) []domain.ModelListItem

func NewOpenAI() domain.IModelProvider[domain.ModelMetadata]

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

func NewModelKit(logger *slog.Logger) *ModelKit

NewModelKit 创建一个新的ModelKit实例

func (m *ModelKit) CheckModel(ctx context.Context, req *domain.CheckModelReq) (*domain.CheckModelResp, error)

func (m *ModelKit) GetChatModel(ctx context.Context, md *domain.ModelMetadata) (model.BaseChatModel, error)

func (m *ModelKit) GetEmbedder(ctx context.Context, model *domain.ModelMetadata) (embedding.Embedder, error)

func (m *ModelKit) GetReranker(ctx context.Context, model *domain.ModelMetadata) (domain.Reranker, error)

func (m *ModelKit) ModelList(ctx context.Context, req *domain.ModelListReq) (*domain.ModelListResp, error)

func (m *ModelKit) UseEmbedder(ctx context.Context, e embedding.Embedder, texts []string) (*domain.EmbeddingsResponse, error)

type OpenAI struct{}

func (o *OpenAI) ListModel(subType string, provider string) ([]domain.ModelMetadata, error)