mirror of
https://github.com/langgenius/dify.git
synced 2025-07-10 02:31:37 +00:00
26 lines
669 B
Python
26 lines
669 B
Python
|
from uuid import uuid4
|
||
|
|
|
||
|
|
from core.app.segments import SegmentType, factory
|
||
|
|
from models import ConversationVariable
|
||
|
|
|
||
|
|
|
||
|
|
def test_from_variable_and_to_variable():
|
||
|
|
variable = factory.build_variable_from_mapping(
|
||
|
|
{
|
||
|
|
'id': str(uuid4()),
|
||
|
|
'name': 'name',
|
||
|
|
'value_type': SegmentType.OBJECT,
|
||
|
|
'value': {
|
||
|
|
'key': {
|
||
|
|
'key': 'value',
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
conversation_variable = ConversationVariable.from_variable(
|
||
|
|
app_id='app_id', conversation_id='conversation_id', variable=variable
|
||
|
|
)
|
||
|
|
|
||
|
|
assert conversation_variable.to_variable() == variable
|