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

refactor(run.py): remove unused process_graph function

Browse source 
refactor(test_graph.py): update get_result_and_steps to get_result_and_thought function name
This commit is contained in:
Gabriel Almeida 2023-05-10 11:31:06 -03:00
commit 31062f2a8c
2 changed files with 2 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_graph.py
View file

@ -14,7 +14,7 @@ from langflow.graph.nodes import (
ToolNode,
WrapperNode,
)
from langflow.interface.run import get_result_and_steps
from langflow.interface.run import get_result_and_thought
from langflow.utils.payload import get_root_node
# Test cases for the graph module
@ -335,7 +335,7 @@ def test_get_result_and_thought(basic_graph):
# now build again and check if FakeListLLM was used
# Get the result and thought
result, thought = get_result_and_steps(langchain_object, message)
result, thought = get_result_and_thought(langchain_object, message)
# The result should be a str
assert isinstance(result, str)
# The thought should be a Thought