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

Refactor RunnableExecutor.py to handle input keys and provide status information

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-27 17:59:19 -03:00
commit b3a621d1b2
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/base/langflow/components/experimental/RunnableExecutor.py
View file

@ -89,8 +89,19 @@ class RunnableExecComponent(CustomComponent):
input_key: str = "input", input_key: str = "input",
output_key: str = "output", output_key: str = "output",
) -> Text: ) -> Text:
input_dict = {}
status = ""
if hasattr(runnable, "input_keys"):
# Check if input_key is in the runnable's input_keys
if input_key in runnable.input_keys:
input_dict[input_key] = input_value
else:
input_dict = {k: input_value for k in runnable.input_keys}
status = f"Warning: The input key is not '{input_key}'. The input key is '{runnable.input_keys}'."
result = runnable.invoke({input_key: input_value}) result = runnable.invoke({input_key: input_value})
result_value, status = self.get_output(result, input_key, output_key) result_value, _status = self.get_output(result, input_key, output_key)
status += _status
status += f"\nOutput: {result_value}\n\nRaw Output: {result}" status += f"\nOutput: {result_value}\n\nRaw Output: {result}"
self.status = status self.status = status
return result_value return result_value