From 8d8fae78c878ee01e4c8157eca9cb9ea7cc47bcf Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 27 Jul 2023 07:02:14 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(=5F=5Finit=5F=5F.py):=20upda?= =?UTF-8?q?te=20import=20statement=20for=20CustomComponent=20to=20reflect?= =?UTF-8?q?=20new=20location=20=F0=9F=94=A7=20fix(constants.py):=20update?= =?UTF-8?q?=20import=20statement=20for=20CustomComponent=20to=20reflect=20?= =?UTF-8?q?new=20location=20=F0=9F=94=A7=20fix(constants.py):=20update=20f?= =?UTF-8?q?ield=5Fconfig=20to=20build=5Fconfig=20to=20match=20changes=20in?= =?UTF-8?q?=20CustomComponent=20class=20=F0=9F=94=A7=20fix(constants.py):?= =?UTF-8?q?=20update=20build=20method=20signature=20to=20match=20changes?= =?UTF-8?q?=20in=20CustomComponent=20class?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/__init__.py | 4 ++-- src/backend/langflow/interface/custom/constants.py | 10 +++++----- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/backend/langflow/__init__.py b/src/backend/langflow/__init__.py index d6c645486..5920369e2 100644 --- a/src/backend/langflow/__init__.py +++ b/src/backend/langflow/__init__.py @@ -1,7 +1,7 @@ from importlib import metadata from langflow.cache import cache_manager from langflow.processing.process import load_flow_from_json -from langflow.utils.types import Prompt +from langflow.interface.custom.custom_component import CustomComponent try: __version__ = metadata.version(__package__) @@ -10,4 +10,4 @@ except metadata.PackageNotFoundError: __version__ = "" del metadata # optional, avoids polluting the results of dir(__package__) -__all__ = ["load_flow_from_json", "cache_manager", "Prompt"] +__all__ = ["load_flow_from_json", "cache_manager", "CustomComponent"] diff --git a/src/backend/langflow/interface/custom/constants.py b/src/backend/langflow/interface/custom/constants.py index 00aa5314d..8e5db39b8 100644 --- a/src/backend/langflow/interface/custom/constants.py +++ b/src/backend/langflow/interface/custom/constants.py @@ -35,8 +35,7 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = { DEFAULT_CUSTOM_COMPONENT_CODE = """ -from langflow import Prompt -from langflow.interface.custom.custom_component import CustomComponent +from langflow import CustomComponent from langchain.llms.base import BaseLLM from langchain.chains import LLMChain @@ -48,11 +47,12 @@ import requests class YourComponent(CustomComponent): display_name: str = "Your Component" description: str = "Your description" - field_config = { "url": { "multiline": True, "required": True } } - def build(self, url: str, llm: BaseLLM, template: Prompt) -> Document: + def build_config(self): + return { "url": { "multiline": True, "required": True } } + + def build(self, url: str, llm: BaseLLM, prompt: PromptTemplate) -> Document: response = requests.get(url) - prompt = PromptTemplate.from_template(template) chain = LLMChain(llm=llm, prompt=prompt) result = chain.run(response.text[:300]) return Document(page_content=str(result))