1c0f18f897
🔧 fix(endpoints.py): update import statement to reflect changes in module structure 🔧 fix(flows.py): update import statements to reflect changes in module structure 🔧 fix(login.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(custom_component.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(base.py): update import statement to reflect changes in module structure 🔧 fix(process.py): update import statement for get_session_service in langflow.processing.process module to reflect correct module location 🔧 fix(utils.py): update import statement for get_session and get_settings_service in langflow.services.auth.utils module to reflect correct module location 🔧 fix(manager.py): update import statement for ServiceType.CACHE_MANAGER in langflow.services.chat.manager module to reflect correct module location 🔧 fix(manager.py): update import statement for ServiceType.DATABASE_MANAGER in langflow.services.chat.manager module to reflect correct module location 🔧 fix(utils.py): update import statement for get_settings_service in langflow.services.database.manager module to reflect correct module location 🔧 fix(utils.py): update import statement for get_settings_service in langflow.services.database.utils module to reflect correct module location 🔧 fix(getters.py): update import statement for DatabaseManager, SettingsManager, ChatManager in langflow.services.getters module to reflect correct module location 🔧 fix(getters.py): update import statement for get_settings_manager in langflow.services.getters module to reflect correct function name change 🔧 fix(getters.py): update import statement for get_db_manager in langflow.services.getters module to reflect correct function name change 🔧 fix(getters.py): update import statement for get_chat_manager in langflow.services.getters module to reflect correct function name change 🔧 fix(getters.py): update import statement for get_settings_manager in langflow.services.getters module to reflect correct function name change 🔧 fix(getters.py): update import statement for get_db_manager in langflow.services.getters module to reflect correct function name change 🔧 fix(getters.py): update import statement for get_session in langflow.services.getters module to reflect correct function name change 🔧 fix(getters.py): update import statement for get_chat_manager in langflow.services.getters module to reflect correct function name change 🔧 fix(langfuse.py): update import statement for get_settings_service in langflow.services.plugins.langfuse module to reflect correct module location 🔧 fix(schema.py): update import statement for ServiceType.AUTH_MANAGER, ServiceType.CACHE_MANAGER, ServiceType.SETTINGS_MANAGER, ServiceType.DATABASE_MANAGER, ServiceType.CHAT_MANAGER, ServiceType.SESSION_MANAGER, ServiceType.TASK_MANAGER in langflow.services.schema module to reflect correct module location 🔧 fix
235 lines
7.2 KiB
Python
235 lines
7.2 KiB
Python
from langflow.processing.process import process_tweaks
|
|
from langflow.services.getters import get_session_service
|
|
|
|
|
|
def test_no_tweaks():
|
|
graph_data = {
|
|
"data": {
|
|
"nodes": [
|
|
{
|
|
"id": "node1",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 1},
|
|
"param2": {"value": 2},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
"id": "node2",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 3},
|
|
"param2": {"value": 4},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
tweaks = {}
|
|
result = process_tweaks(graph_data, tweaks)
|
|
assert result == graph_data
|
|
|
|
|
|
def test_single_tweak():
|
|
graph_data = {
|
|
"data": {
|
|
"nodes": [
|
|
{
|
|
"id": "node1",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 1},
|
|
"param2": {"value": 2},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
"id": "node2",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 3},
|
|
"param2": {"value": 4},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
tweaks = {"node1": {"param1": 5}}
|
|
expected_result = {
|
|
"data": {
|
|
"nodes": [
|
|
{
|
|
"id": "node1",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 5},
|
|
"param2": {"value": 2},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
"id": "node2",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 3},
|
|
"param2": {"value": 4},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
result = process_tweaks(graph_data, tweaks)
|
|
assert result == expected_result
|
|
|
|
|
|
def test_multiple_tweaks():
|
|
graph_data = {
|
|
"data": {
|
|
"nodes": [
|
|
{
|
|
"id": "node1",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 1},
|
|
"param2": {"value": 2},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
"id": "node2",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 3},
|
|
"param2": {"value": 4},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
tweaks = {
|
|
"node1": {"param1": 5, "param2": 6},
|
|
"node2": {"param1": 7},
|
|
}
|
|
expected_result = {
|
|
"data": {
|
|
"nodes": [
|
|
{
|
|
"id": "node1",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 5},
|
|
"param2": {"value": 6},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
"id": "node2",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 7},
|
|
"param2": {"value": 4},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
result = process_tweaks(graph_data, tweaks)
|
|
assert result == expected_result
|
|
|
|
|
|
def test_tweak_not_in_template():
|
|
graph_data = {
|
|
"data": {
|
|
"nodes": [
|
|
{
|
|
"id": "node1",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 1},
|
|
"param2": {"value": 2},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
"id": "node2",
|
|
"data": {
|
|
"node": {
|
|
"template": {
|
|
"param1": {"value": 3},
|
|
"param2": {"value": 4},
|
|
}
|
|
}
|
|
},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
tweaks = {"node1": {"param3": 5}}
|
|
result = process_tweaks(graph_data, tweaks)
|
|
assert result == graph_data
|
|
|
|
|
|
def test_load_langchain_object_with_cached_session(client, basic_graph_data):
|
|
# Provide a non-existent session_id
|
|
session_service = get_session_service()
|
|
session_id1 = "non-existent-session-id"
|
|
graph1, artifacts1 = session_service.load_session(session_id1, basic_graph_data)
|
|
# Use the new session_id to get the langchain_object again
|
|
graph2, artifacts2 = session_service.load_session(session_id1, basic_graph_data)
|
|
|
|
assert graph1 == graph2
|
|
assert artifacts1 == artifacts2
|
|
|
|
|
|
def test_load_langchain_object_with_no_cached_session(client, basic_graph_data):
|
|
# Provide a non-existent session_id
|
|
session_service = get_session_service()
|
|
session_id1 = "non-existent-session-id"
|
|
session_id = session_service.build_key(session_id1, basic_graph_data)
|
|
graph1, artifacts1 = session_service.load_session(session_id, basic_graph_data)
|
|
# Clear the cache
|
|
session_service.clear_session(session_id)
|
|
# Use the new session_id to get the langchain_object again
|
|
graph2, artifacts2 = session_service.load_session(session_id, basic_graph_data)
|
|
|
|
assert id(graph1) != id(graph2)
|
|
# Since the cache was cleared, objects should be different
|
|
|
|
|
|
def test_load_langchain_object_without_session_id(client, basic_graph_data):
|
|
# Provide a non-existent session_id
|
|
session_service = get_session_service()
|
|
session_id1 = None
|
|
graph1, artifacts1 = session_service.load_session(session_id1, basic_graph_data)
|
|
# Use the new session_id to get the langchain_object again
|
|
graph2, artifacts2 = session_service.load_session(session_id1, basic_graph_data)
|
|
|
|
assert graph1 == graph2
|