Yeuoly 10 kuukautta sitten
vanhempi
commit
72dee6ec68

+ 1 - 1
internal/core/plugin_daemon/access_types/access.go

@@ -24,6 +24,6 @@ const (
 	PLUGIN_ACCESS_ACTION_INVOKE_ENDPOINT               PluginAccessAction = "invoke_endpoint"
 	PLUGIN_ACCESS_ACTION_GET_TTS_MODEL_VOICES          PluginAccessAction = "get_tts_model_voices"
 	PLUGIN_ACCESS_ACTION_GET_TEXT_EMBEDDING_NUM_TOKENS PluginAccessAction = "get_text_embedding_num_tokens"
-	PLUGIN_ACCESS_ACTION_GET_AI_MODEL_SCHEMA           PluginAccessAction = "get_ai_model_schema"
+	PLUGIN_ACCESS_ACTION_GET_AI_MODEL_SCHEMAS          PluginAccessAction = "get_ai_model_schemas"
 	PLUGIN_ACCESS_ACTION_GET_LLM_NUM_TOKENS            PluginAccessAction = "get_llm_num_tokens"
 )

+ 1 - 1
internal/service/invoke_model.go

@@ -285,7 +285,7 @@ func GetAIModelSchema(
 	session, err := createSession(
 		r,
 		access_types.PLUGIN_ACCESS_TYPE_MODEL,
-		access_types.PLUGIN_ACCESS_ACTION_GET_AI_MODEL_SCHEMA,
+		access_types.PLUGIN_ACCESS_ACTION_GET_AI_MODEL_SCHEMAS,
 		ctx.GetString("cluster_id"),
 	)
 	if err != nil {

+ 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:"required,dive"`
+	AIModels *plugin_entities.ModelDeclaration `json:"ai_models" validate:"omitempty,dive"`
 }