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

Refactor ShouldRunNext component to handle boolean response properly

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-29 23:09:42 -03:00
commit 603bca019c
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/components/utilities/ShouldRunNext.py
View file

@ -1,5 +1,6 @@
# Implement ShouldRunNext component # Implement ShouldRunNext component
from typing import Text from typing import Text
from langchain_core.prompts import PromptTemplate from langchain_core.prompts import PromptTemplate
from langflow import CustomComponent from langflow import CustomComponent
@ -42,8 +43,10 @@ class ShouldRunNext(CustomComponent):
result = result.get("response") result = result.get("response")
if result.lower() not in ["true", "false"]: if result.lower() not in ["true", "false"]:
raise ValueError("The prompt should generate a boolean response (True or False).") raise ValueError(
"The prompt should generate a boolean response (True or False)."
)
# The string should be the words true or false # The string should be the words true or false
# if not raise an error # if not raise an error
bool_result = result.lower() == "true" bool_result = result.lower() == "true"
return {"condition": bool_result, "result": kwargs} return {"path": bool_result, "result": kwargs}