瀏覽代碼

fix: missing few routes of model

takatost 10 月之前
父節點
當前提交
5251bdb92b

+ 4 - 0
internal/server/http_server.go

@@ -58,13 +58,17 @@ func (app *App) pluginDispatchGroup(group *gin.RouterGroup, config *app.Config)
 	group.POST("/tool/invoke", controllers.InvokeTool(config))
 	group.POST("/tool/validate_credentials", controllers.ValidateToolCredentials(config))
 	group.POST("/llm/invoke", controllers.InvokeLLM(config))
+	group.POST("/llm/num_tokens", controllers.GetLLMNumTokens(config))
 	group.POST("/text_embedding/invoke", controllers.InvokeTextEmbedding(config))
+	group.POST("/text_embedding/num_tokens", controllers.GetTextEmbeddingNumTokens(config))
 	group.POST("/rerank/invoke", controllers.InvokeRerank(config))
 	group.POST("/tts/invoke", controllers.InvokeTTS(config))
+	group.POST("/tts/model/voices", controllers.GetTTSModelVoices(config))
 	group.POST("/speech2text/invoke", controllers.InvokeSpeech2Text(config))
 	group.POST("/moderation/invoke", controllers.InvokeModeration(config))
 	group.POST("/model/validate_provider_credentials", controllers.ValidateProviderCredentials(config))
 	group.POST("/model/validate_model_credentials", controllers.ValidateModelCredentials(config))
+	group.POST("/model/schema", controllers.GetAIModelSchema(config))
 }
 
 func (app *App) remoteDebuggingGroup(group *gin.RouterGroup, config *app.Config) {

+ 1 - 1
internal/types/entities/model_entities/ai_model.go

@@ -3,5 +3,5 @@ package model_entities
 import "github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
 
 type GetModelSchemasResponse struct {
-	AIModels *plugin_entities.ModelDeclaration `json:"ai_models" validate:"omitempty,dive"`
+	ModelSchema *plugin_entities.ModelDeclaration `json:"model_schema" validate:"omitempty"`
 }

+ 4 - 2
internal/types/entities/requests/model.go

@@ -126,6 +126,8 @@ type RequestGetLLMNumTokens struct {
 }
 
 type RequestGetAIModelSchema struct {
-	Model       string         `json:"model" validate:"required"`
-	Credentials map[string]any `json:"credentials" validate:"omitempty,dive,is_basic_type"`
+	Provider    string                   `json:"provider" validate:"required"`
+	ModelType   model_entities.ModelType `json:"model_type"  validate:"required,model_type"`
+	Model       string                   `json:"model" validate:"required"`
+	Credentials map[string]any           `json:"credentials" validate:"omitempty,dive,is_basic_type"`
 }