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

refactor: Update TextOperatorComponent to use langflow.template and langflow.schema

Browse source 
The TextOperatorComponent in TextOperator.py has been refactored to use the langflow.template.Input, langflow.template.Output, and langflow.schema.Record classes for improved code structure and maintainability.

Note: The commit message has been generated based on the provided code changes and recent commits.
This commit is contained in:
ogabrielluiz 2024-06-04 12:56:10 -03:00
commit dd4b617b48
2 changed files with 62 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

102
src/backend/base/langflow/components/experimental/TextOperator.py
View file

@ -1,50 +1,64 @@
from typing import Optional, Union
from typing import Union
from langflow.custom import CustomComponent
from langflow.custom import Component
from langflow.field_typing import Text
from langflow.schema import Record
from langflow.template import Input, Output
class TextOperatorComponent(CustomComponent):
class TextOperatorComponent(Component):
display_name = "Text Operator"
description = "Compares two text inputs based on a specified condition such as equality or inequality, with optional case sensitivity."
def build_config(self) -> dict:
return {
"input_text": {
"display_name": "Input Text",
"info": "The primary text input for the operation.",
},
"match_text": {
"display_name": "Match Text",
"info": "The text input to compare against.",
},
"operator": {
"display_name": "Operator",
"info": "The operator to apply for comparing the texts.",
"options": ["equals", "not equals", "contains", "starts with", "ends with", "exists"],
},
"case_sensitive": {
"display_name": "Case Sensitive",
"info": "If true, the comparison will be case sensitive.",
"field_type": "bool",
"default": False,
},
"true_output": {
"display_name": "Output",
"info": "The output to return or display when the comparison is true.",
"input_types": ["Text", "Record"], # Allow both text and record types
},
}
inputs = [
Input(name="input_text", type=str, display_name="Input Text", info="The primary text input for the operation."),
Input(name="match_text", type=str, display_name="Match Text", info="The text input to compare against."),
Input(
name="operator",
type=str,
display_name="Operator",
info="The operator to apply for comparing the texts.",
options=["equals", "not equals", "contains", "starts with", "ends with", "exists"],
),
Input(
name="case_sensitive",
type=bool,
display_name="Case Sensitive",
info="If true, the comparison will be case sensitive.",
default=False,
),
Input(
name="true_output",
type=Union[str, Record],
display_name="True Output",
info="The output to return or display when the comparison is true.",
input_types=["Text", "Record"],
),
Input(
name="false_output",
type=Union[str, Record],
display_name="False Output",
info="The output to return or display when the comparison is false.",
input_types=["Text", "Record"],
),
]
outputs = [
Output(name="True Result", method="result_response"),
Output(name="False Result", method="result_response"),
]
def true_response(self) -> Union[Text, Record]:
return self.true_output if self.true_output else self.input_text
def false_response(self) -> Union[Text, Record]:
return self.false_output if self.false_output else self.input_text
def result_response(self) -> Union[Text, Record]:
input_text = self.input_text
match_text = self.match_text
operator = self.operator
case_sensitive = self.case_sensitive
def build(
self,
input_text: Text,
match_text: Text,
operator: Text,
case_sensitive: bool = False,
true_output: Optional[Text] = "",
) -> Union[Text, Record]:
if not input_text or not match_text:
raise ValueError("Both 'input_text' and 'match_text' must be provided and non-empty.")
@ -64,13 +78,9 @@ class TextOperatorComponent(CustomComponent):
elif operator == "ends with":
result = input_text.endswith(match_text)
output_record = true_output if true_output else input_text
if result:
self.status = output_record
return output_record
self.status = self.true_response()
return self.true_response()
else:
self.status = "Comparison failed, stopping execution."
self.stop()
return output_record
self.status = self.false_response()
return self.false_response()