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

feat: update if-else component (#8756)

Browse source 
* add ifelse component

* [autofix.ci] apply automated fixes

* fix test

* fix test

* fix test

* fix tests

* [autofix.ci] apply automated fixes

* fix tests

* fix test

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Lucas Democh <democh@datax.dev>
This commit is contained in:
Yuqi Tang 2025-07-02 15:56:57 -07:00 • committed by GitHub
commit 19c9514d54
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 119 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/backend/tests/unit/graph/graph/test_cycles.py
View file

@ -32,15 +32,16 @@ class Concatenate(Component):
def test_cycle_in_graph():
chat_input = ChatInput(_id="chat_input")
router = ConditionalRouterComponent(_id="router", default_route="true_result")
# Use router's message output instead of false_response
chat_input.set(input_value=router.message)
# Use router's true case message output instead of message
chat_input.set(input_value=router.true_case_message)
concat_component = Concatenate(_id="concatenate")
concat_component.set(text=chat_input.message_response)
router.set(
input_text=chat_input.message_response,
match_text="testtesttesttest",
operator="equals",
message=concat_component.concatenate,
true_case_message=concat_component.concatenate,
false_case_message=concat_component.concatenate,
)
text_output = TextOutputComponent(_id="text_output")
text_output.set(input_value=router.true_response)
@ -83,14 +84,16 @@ def test_cycle_in_graph():
def test_cycle_in_graph_max_iterations():
text_input = TextInputComponent(_id="text_input")
router = ConditionalRouterComponent(_id="router")
# Connect text_input to router's input
text_input.set(input_value=router.false_response)
concat_component = Concatenate(_id="concatenate")
concat_component.set(text=text_input.text_response)
# Connect concatenate output back to router's input to create cycle
router.set(
input_text=text_input.text_response,
match_text="testtesttesttest",
operator="equals",
message=concat_component.concatenate,
false_case_message=concat_component.concatenate,
)
text_output = TextOutputComponent(_id="text_output")
text_output.set(input_value=router.true_response)
@ -100,7 +103,7 @@ def test_cycle_in_graph_max_iterations():
graph = Graph(text_input, chat_output)
assert graph.is_cyclic is True
# Run queue should contain chat_input and not router
# Run queue should contain text_input and not router
assert "text_input" in graph._run_queue
assert "router" not in graph._run_queue
@ -109,24 +112,24 @@ def test_cycle_in_graph_max_iterations():
def test_that_outputs_cache_is_set_to_false_in_cycle():
chat_input = ChatInput(_id="chat_input")
text_input = TextInputComponent(_id="text_input")
router = ConditionalRouterComponent(_id="router")
# Use router's message output instead of false_response
chat_input.set(input_value=router.message)
concat_component = Concatenate(_id="concatenate")
concat_component.set(text=chat_input.message_response)
text_input.set(input_value=router.false_response)
concat_component.set(text=text_input.text_response)
router.set(
input_text=chat_input.message_response,
input_text=text_input.text_response,
match_text="testtesttesttest",
operator="equals",
message=concat_component.concatenate,
true_case_message=concat_component.concatenate,
false_case_message=concat_component.concatenate,
)
text_output = TextOutputComponent(_id="text_output")
text_output.set(input_value=router.true_response)
chat_output = ChatOutput(_id="chat_output")
chat_output.set(input_value=text_output.text_response)
graph = Graph(chat_input, chat_output)
graph = Graph(text_input, chat_output)
cycle_vertices = find_cycle_vertices(graph._get_edges_as_list_of_tuples())
cycle_outputs_lists = [
graph.vertex_map[vertex_id].custom_component._outputs_map.values() for vertex_id in cycle_vertices
@ -167,7 +170,8 @@ def test_updated_graph_with_prompts():
input_text=openai_component_1.text_response,
match_text=chat_input.message_response,
operator="contains",
message=openai_component_1.text_response,
true_case_message=openai_component_1.text_response,
false_case_message=openai_component_1.text_response,
)
# Second prompt: After the last try, provide a new hint
@ -236,7 +240,8 @@ def test_updated_graph_with_max_iterations():
input_text=openai_component_1.text_response,
match_text=chat_input.message_response,
operator="contains",
message=openai_component_1.text_response,
true_case_message=openai_component_1.text_response,
false_case_message=openai_component_1.text_response,
)
# Second prompt: After the last try, provide a new hint
@ -290,7 +295,8 @@ def test_conditional_router_max_iterations():
input_text=text_input.text_response,
match_text="bacon",
operator="equals",
message="This message should not be routed to true_result",
true_case_message="This message should not be routed to true_result",
false_case_message="This message should not be routed to false_result",
max_iterations=5,
default_route="true_result",
)