Add PythonFunctionComponent to langflow utilities

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 16:38:01 -03:00
commit 63c709ee1d

View file

@ -0,0 +1,24 @@
from typing import Callable
from langflow import CustomComponent
from langflow.field_typing import Code
from langflow.interface.custom.utils import get_function
class PythonFunctionComponent(CustomComponent):
display_name = "Python Function"
description = "Define a Python function."
def build_config(self):
return {
"function_code": {
"display_name": "Code",
"info": "The code for the function.",
"show": True,
},
}
def build(self, function_code: Code) -> Callable:
self.status = function_code
func = get_function(function_code)
return func