youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 17:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge main

Browse Source
This commit is contained in:
Yeuoly
2024-09-14 02:47:01 +08:00
parent 87c746f6bb b6b1057a18
commit 00d1c45518
959 changed files with 25695 additions and 24057 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
api/core/plugin/entities/request.py
View File

@ -32,6 +32,7 @@ class RequestInvokeLLM(BaseRequestInvokeModel):
"""
Request to invoke LLM
"""
model_type: ModelType = ModelType.LLM
mode: str
model_parameters: dict[str, Any] = Field(default_factory=dict)
@ -40,19 +41,19 @@ class RequestInvokeLLM(BaseRequestInvokeModel):
stop: Optional[list[str]] = Field(default_factory=list)
stream: Optional[bool] = False
@field_validator('prompt_messages', mode='before')
def convert_prompt_messages(cls, v):
@field_validator("prompt_messages", mode="before")
def convert_prompt_messages(self, v):
if not isinstance(v, list):
raise ValueError('prompt_messages must be a list')
raise ValueError("prompt_messages must be a list")
for i in range(len(v)):
if v[i]['role'] == PromptMessageRole.USER.value:
if v[i]["role"] == PromptMessageRole.USER.value:
v[i] = UserPromptMessage(**v[i])
elif v[i]['role'] == PromptMessageRole.ASSISTANT.value:
elif v[i]["role"] == PromptMessageRole.ASSISTANT.value:
v[i] = AssistantPromptMessage(**v[i])
elif v[i]['role'] == PromptMessageRole.SYSTEM.value:
elif v[i]["role"] == PromptMessageRole.SYSTEM.value:
v[i] = SystemPromptMessage(**v[i])
elif v[i]['role'] == PromptMessageRole.TOOL.value:
elif v[i]["role"] == PromptMessageRole.TOOL.value:
v[i] = ToolPromptMessage(**v[i])
else:
v[i] = PromptMessage(**v[i])
@ -95,10 +96,12 @@ class RequestInvokeNode(BaseModel):
Request to invoke node
"""
class RequestInvokeApp(BaseModel):
"""
Request to invoke app
"""
app_id: str
inputs: dict[str, Any]
query: Optional[str] = None
@ -107,12 +110,14 @@ class RequestInvokeApp(BaseModel):
user: Optional[str] = None
files: list[dict] = Field(default_factory=list)
class RequestInvokeEncrypt(BaseModel):
"""
Request to encryption
"""
opt: Literal["encrypt", "decrypt"]
namespace: Literal["endpoint"]
identity: str
data: dict = Field(default_factory=dict)
config: Mapping[str, BasicProviderConfig] = Field(default_factory=Mapping)
config: Mapping[str, BasicProviderConfig] = Field(default_factory=Mapping)