youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-02-27 13:07:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2 transform

Browse Source
This commit is contained in:
jyong
2025-07-18 19:04:46 +08:00
parent dd59cea085
commit dc359c6442
2 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
api/core/workflow/system_variable.py
View File

@ -1,5 +1,5 @@
from collections.abc import Sequence
from typing import Any
from typing import Any, Mapping
from pydantic import AliasChoices, BaseModel, ConfigDict, Field, model_validator
@ -43,6 +43,12 @@ class SystemVariable(BaseModel):
query: str | None = None
conversation_id: str | None = None
dialogue_count: int | None = None
document_id: str | None = None
dataset_id: str | None = None
batch: str | None = None
datasource_type: str | None = None
datasource_info: Mapping[str, Any] | None = None
invoke_from: str | None = None
@model_validator(mode="before")
@classmethod
@ -86,4 +92,16 @@ class SystemVariable(BaseModel):
d[SystemVariableKey.CONVERSATION_ID] = self.conversation_id
if self.dialogue_count is not None:
d[SystemVariableKey.DIALOGUE_COUNT] = self.dialogue_count
if self.document_id is not None:
d[SystemVariableKey.DOCUMENT_ID] = self.document_id
if self.dataset_id is not None:
d[SystemVariableKey.DATASET_ID] = self.dataset_id
if self.batch is not None:
d[SystemVariableKey.BATCH] = self.batch
if self.datasource_type is not None:
d[SystemVariableKey.DATASOURCE_TYPE] = self.datasource_type
if self.datasource_info is not None:
d[SystemVariableKey.DATASOURCE_INFO] = self.datasource_info
if self.invoke_from is not None:
d[SystemVariableKey.INVOKE_FROM] = self.invoke_from
return d