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

Replace import statement in custom component class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-27 15:37:22 -03:00
commit 9abad780b2
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/interface/custom/custom_component/component.py
View file

@ -34,6 +34,10 @@ class Component:
if key == "user_id": if key == "user_id":
setattr(self, "_user_id", value) setattr(self, "_user_id", value)
else: else:
if key == "code" and "from langflow import CustomComponent" in value:
value = value.replace(
"from langflow import CustomComponent", "from langflow.custom import CustomComponent"
)
setattr(self, key, value) setattr(self, key, value)
def __setattr__(self, key, value): def __setattr__(self, key, value):