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

Fix method argument type in CustomComponent class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-02 00:47:15 -03:00
commit 801c5462fc
1 changed files with 2 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/backend/langflow/interface/custom/custom_component/custom_component.py
View file

@ -14,7 +14,6 @@ from uuid import UUID
import yaml import yaml
from cachetools import TTLCache, cachedmethod from cachetools import TTLCache, cachedmethod
from fastapi import HTTPException
from langchain_core.documents import Document from langchain_core.documents import Document
from sqlmodel import select from sqlmodel import select
@ -85,7 +84,7 @@ class CustomComponent(Component):
def append_state(self, name: str, value: Any): def append_state(self, name: str, value: Any):
try: try:
self.vertex.graph.append_state( self.vertex.graph.append_state(
key=name, record=value, caller=self.vertex.id name=name, record=value, caller=self.vertex.id
) )
except Exception as e: except Exception as e:
raise ValueError(f"Error appending state: {e}") raise ValueError(f"Error appending state: {e}")
@ -201,18 +200,7 @@ class CustomComponent(Component):
args = build_method["args"] args = build_method["args"]
for arg in args: for arg in args:
if arg.get("type") == "prompt": if not arg.get("type") and arg.get("name") != "self":
raise HTTPException(
status_code=400,
detail={
"error": "Type hint Error",
"traceback": (
"Prompt type is not supported in the build method."
" Try using PromptTemplate instead."
),
},
)
elif not arg.get("type") and arg.get("name") != "self":
# Set the type to Data # Set the type to Data
arg["type"] = "Data" arg["type"] = "Data"
return args return args