mp

type AppModelParams struct {
	ProviderOpenAICompatible AppModelParamsOpenAICompatible `json:"providerOpenAICompatible"` // OpenAI-API-compatible模型配置
	ProviderYuanJing         AppModelParamsYuanjing         `json:"providerYuanjing"`         // YuanJing模型配置
	ProviderHuoshan          AppModelParamsHuoshan          `json:"providerHuoshan"`
	ProviderQwen             AppModelParamsQwen             `json:"providerQwen"`
	ProviderOllama           AppModelParamsOllama           `json:"providerOllama"`
	ProviderInfini           AppModelParamsInfini           `json:"providerModelByInfini"`
}

type AppModelParamsHuoshan struct {
	LLM mp_huoshan.LLMParams `json:"llm"` // 大语言模型配置
}

type AppModelParamsInfini struct {
	LLM mp_infini.LLMParams `json:"llm"` // 大语言模型配置
}

type AppModelParamsOllama struct {
	LLM mp_ollama.LLMParams `json:"llm"` // 大语言模型配置
}

type AppModelParamsOpenAICompatible struct {
	LLM mp_openai_compatible.LLMParams `json:"llm"` // 大语言模型配置
}

type AppModelParamsQwen struct {
	LLM mp_qwen.LLMParams `json:"llm"` // 大语言模型配置
}

type AppModelParamsYuanjing struct {
	LLM mp_yuanjing.LLMParams `json:"llm"` // 大语言模型配置
}

type IEmbedding interface {
	Tags() []mp_common.Tag
	NewReq(req *mp_common.EmbeddingReq) (mp_common.IEmbeddingReq, error)
	Embeddings(ctx context.Context, req mp_common.IEmbeddingReq, headers ...mp_common.Header) (mp_common.IEmbeddingResp, error)
}

type IGui interface {
	Tags() []mp_common.Tag
	NewReq(req *mp_common.GuiReq) (mp_common.IGuiReq, error)
	Gui(ctx context.Context, req mp_common.IGuiReq, headers ...mp_common.Header) (mp_common.IGuiResp, error)
}

type ILLM interface {
	Tags() []mp_common.Tag
	NewReq(req *mp_common.LLMReq) (mp_common.ILLMReq, error)
	ChatCompletions(ctx context.Context, req mp_common.ILLMReq, headers ...mp_common.Header) (mp_common.ILLMResp, <-chan mp_common.ILLMResp, error)
}

type ILLMParams interface {
	GetParams() map[string]interface{}
}

type IOcr interface {
	Tags() []mp_common.Tag
	NewReq(req *mp_common.OcrReq) (mp_common.IOcrReq, error)
	Ocr(ctx *gin.Context, req mp_common.IOcrReq, headers ...mp_common.Header) (mp_common.IOcrResp, error)
}

type IPdfParser interface {
	Tags() []mp_common.Tag
	NewReq(req *mp_common.PdfParserReq) (mp_common.IPdfParserReq, error)
	PdfParser(ctx *gin.Context, req mp_common.IPdfParserReq, headers ...mp_common.Header) (mp_common.IPdfParserResp, error)
}

type IRerank interface {
	Tags() []mp_common.Tag
	NewReq(req *mp_common.RerankReq) (mp_common.IRerankReq, error)
	Rerank(ctx context.Context, req mp_common.IRerankReq, headers ...mp_common.Header) (mp_common.IRerankResp, error)
}

type ProviderModelByHuoshan struct {
	Llm       mp_huoshan.LLM       `json:"llm"`
	Embedding mp_huoshan.Embedding `json:"embedding"`
}

type ProviderModelByInfini struct {
	Llm       mp_infini.LLM       `json:"llm"`
	Rerank    mp_infini.Rerank    `json:"rerank"`
	Embedding mp_infini.Embedding `json:"embedding"`
}

type ProviderModelByOllama struct {
	Llm       mp_ollama.LLM       `json:"llm"`
	Embedding mp_ollama.Embedding `json:"embedding"`
}

type ProviderModelByOpenAICompatible struct {
	Llm       mp_openai_compatible.LLM       `json:"llm"`
	Rerank    mp_openai_compatible.Rerank    `json:"rerank"`
	Embedding mp_openai_compatible.Embedding `json:"embedding"`
}

type ProviderModelByQwen struct {
	Llm       mp_qwen.LLM       `json:"llm"`
	Rerank    mp_qwen.Rerank    `json:"rerank"`
	Embedding mp_qwen.Embedding `json:"embedding"`
}

type ProviderModelByYuanjing struct {
	Llm       mp_yuanjing.LLM       `json:"llm"`
	Rerank    mp_yuanjing.Rerank    `json:"rerank"`
	Embedding mp_yuanjing.Embedding `json:"embedding"`
	Ocr       mp_yuanjing.Ocr       `json:"ocr"`
	Gui       mp_yuanjing.Gui       `json:"gui"`
	PdfParser mp_yuanjing.PdfParser `json:"pdf-parser"`
}

type ProviderModelConfig struct {
	ProviderYuanJing         ProviderModelByYuanjing         `json:"providerYuanJing"`
	ProviderOpenAICompatible ProviderModelByOpenAICompatible `json:"providerOpenAICompatible"`
	ProviderHuoshan          ProviderModelByHuoshan          `json:"providerHuoshan"`
	ProviderQwen             ProviderModelByQwen             `json:"providerQwen"`
	ProviderOllama           ProviderModelByOllama           `json:"providerOllama"`
	ProviderInfini           ProviderModelByInfini           `json:"providerModelByInfini"`
}