소스 검색

feat: support auto generate and template

Yeuoly 7 달 전
부모
커밋
b34d815883
1개의 변경된 파일13개의 추가작업 그리고 0개의 파일을 삭제
  1. 13 0
      api/core/plugin/entities/parameters.py

+ 13 - 0
api/core/plugin/entities/parameters.py

@@ -40,11 +40,24 @@ class PluginParameterType(enum.StrEnum):
     SYSTEM_FILES = CommonParameterType.SYSTEM_FILES.value
     SYSTEM_FILES = CommonParameterType.SYSTEM_FILES.value
 
 
 
 
+class PluginParameterAutoGenerate(BaseModel):
+    class Type(enum.StrEnum):
+        PROMPT_INSTRUCTION = "prompt_instruction"
+
+    type: Type
+
+
+class PluginParameterTemplate(BaseModel):
+    enabled: bool = Field(default=False, description="Whether the parameter is jinja enabled")
+
+
 class PluginParameter(BaseModel):
 class PluginParameter(BaseModel):
     name: str = Field(..., description="The name of the parameter")
     name: str = Field(..., description="The name of the parameter")
     label: I18nObject = Field(..., description="The label presented to the user")
     label: I18nObject = Field(..., description="The label presented to the user")
     placeholder: Optional[I18nObject] = Field(default=None, description="The placeholder presented to the user")
     placeholder: Optional[I18nObject] = Field(default=None, description="The placeholder presented to the user")
     scope: str | None = None
     scope: str | None = None
+    auto_generate: Optional[PluginParameterAutoGenerate] = None
+    template: Optional[PluginParameterTemplate] = None
     required: bool = False
     required: bool = False
     default: Optional[Union[float, int, str]] = None
     default: Optional[Union[float, int, str]] = None
     min: Optional[Union[float, int]] = None
     min: Optional[Union[float, int]] = None