jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(variable-handling): enhance variable and segment conversion (#10483)

Browse source
This commit is contained in:
-LAN- 2024-11-12 21:51:09 +08:00 • committed by GitHub
commit 70c2ec8ed5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 91 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

9
api/core/workflow/entities/variable_pool.py
View file

@ -95,13 +95,16 @@ class VariablePool(BaseModel):
if len(selector) < 2:
raise ValueError("Invalid selector")
if isinstance(value, Variable):
variable = value
if isinstance(value, Segment):
v = value
variable = variable_factory.segment_to_variable(segment=value, selector=selector)
else:
v = variable_factory.build_segment(value)
segment = variable_factory.build_segment(value)
variable = variable_factory.segment_to_variable(segment=segment, selector=selector)
hash_key = hash(tuple(selector[1:]))
self.variable_dictionary[selector[0]][hash_key] = v
self.variable_dictionary[selector[0]][hash_key] = variable
def get(self, selector: Sequence[str], /) -> Segment | None:
"""