AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 18:19:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api/core/app/segments): Allow to contains boolean value in object segment (#7047)

Browse Source
This commit is contained in:
-LAN- 2024-08-07 16:20:22 +08:00 committed by GitHub
parent 72c75b75cf
commit aad02113c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 7 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/app/segments/factory.py
View File

@ -75,8 +75,7 @@ def build_segment(value: Any, /) -> Segment:
return FloatSegment(value=value) return FloatSegment(value=value)
if isinstance(value, dict): if isinstance(value, dict):
# TODO: Limit the depth of the object # TODO: Limit the depth of the object
obj = {k: build_segment(v) for k, v in value.items()} return ObjectSegment(value=value)
return ObjectSegment(value=obj)
if isinstance(value, list): if isinstance(value, list):
# TODO: Limit the depth of the array # TODO: Limit the depth of the array
elements = [build_segment(v) for v in value] elements = [build_segment(v) for v in value]

5
api/core/app/segments/segments.py
View File

@ -85,7 +85,7 @@ class FileSegment(Segment):
class ObjectSegment(Segment): class ObjectSegment(Segment):
value_type: SegmentType = SegmentType.OBJECT value_type: SegmentType = SegmentType.OBJECT
value: Mapping[str, Segment] value: Mapping[str, Any]
@property @property
def text(self) -> str: def text(self) -> str:
@ -102,9 +102,6 @@ class ObjectSegment(Segment):
# TODO: Use markdown code block # TODO: Use markdown code block
return json.dumps(self.model_dump()['value'], ensure_ascii=False, indent=2) return json.dumps(self.model_dump()['value'], ensure_ascii=False, indent=2)
def to_object(self):
return {k: v.to_object() for k, v in self.value.items()}
class ArraySegment(Segment): class ArraySegment(Segment):
@property @property

2
api/tests/unit_tests/core/app/segments/test_factory.py
View File

@ -67,7 +67,7 @@ def test_build_a_object_variable_with_none_value():
} }
) )
assert isinstance(var, ObjectSegment) assert isinstance(var, ObjectSegment)
assert isinstance(var.value['key1'], NoneSegment) assert var.value['key1'] is None
def test_object_variable(): def test_object_variable():

16
api/tests/unit_tests/core/app/segments/test_variables.py
View File

@ -54,20 +54,10 @@ def test_object_variable_to_object():
var = ObjectVariable( var = ObjectVariable(
name='object', name='object',
value={ value={
'key1': ObjectVariable( 'key1': {
name='object', 'key2': 'value2',
value={
'key2': StringVariable(name='key2', value='value2'),
}, },
), 'key2': ['value5_1', 42, {}],
'key2': ArrayAnyVariable(
name='array',
value=[
StringVariable(name='key5_1', value='value5_1'),
IntegerVariable(name='key5_2', value=42),
ObjectVariable(name='key5_3', value={}),
],
),
}, },
) )