dify/api/tests/unit_tests/models/test_conversation_variable.py

26 lines
669 B
Python
Raw Normal View History

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