mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-05-06 10:17:49 +08:00
Fix Gitee embedding model URL error (#13553)
### What problem does this PR solve? As title ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
@ -204,6 +204,7 @@ func (h *LLMHandler) Factories(c *gin.Context) {
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"code": common.CodeSuccess,
|
||||
"data": filtered,
|
||||
})
|
||||
}
|
||||
|
||||
@ -25,13 +25,13 @@ import (
|
||||
|
||||
// ModelProvider represents a model provider configuration
|
||||
type ModelProvider struct {
|
||||
Name string `json:"name"`
|
||||
Logo string `json:"logo"`
|
||||
Tags string `json:"tags"`
|
||||
Status string `json:"status"`
|
||||
Rank string `json:"rank"`
|
||||
LLMs []LLM `json:"llm"`
|
||||
DefaultEmbeddingURL string `json:"default_embedding_url,omitempty"`
|
||||
Name string `json:"name"`
|
||||
Logo string `json:"logo"`
|
||||
Tags string `json:"tags"`
|
||||
Status string `json:"status"`
|
||||
Rank string `json:"rank"`
|
||||
LLMs []LLM `json:"llm"`
|
||||
DefaultURL string `json:"url,omitempty"`
|
||||
}
|
||||
|
||||
// LLM represents a language model within a provider
|
||||
|
||||
@ -86,10 +86,10 @@ func (p *ModelProviderImpl) GetEmbeddingModel(ctx context.Context, tenantID stri
|
||||
// Always get API base from model provider configuration
|
||||
providerDAO := dao.NewModelProviderDAO()
|
||||
providerConfig := providerDAO.GetProviderByName(provider)
|
||||
if providerConfig == nil || providerConfig.DefaultEmbeddingURL == "" {
|
||||
if providerConfig == nil || providerConfig.DefaultURL == "" {
|
||||
return nil, fmt.Errorf("no API base found for provider %s", provider)
|
||||
}
|
||||
apiBase := providerConfig.DefaultEmbeddingURL
|
||||
apiBase := fmt.Sprintf("%sembeddings/", providerConfig.DefaultURL)
|
||||
|
||||
return models.CreateEmbeddingModel(provider, *apiKey, apiBase, modelName, p.httpClient)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user