From ae59104b3af75ed80cee15f6c253d50cad6df383 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 7 Jul 2023 00:41:35 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20refactor(formatter/base.py):=20m?= =?UTF-8?q?ake=20FieldFormatter=20inherit=20from=20pydantic.BaseModel=20fo?= =?UTF-8?q?r=20improved=20type=20checking=20=F0=9F=94=A5=20chore(constants?= =?UTF-8?q?.py):=20remove=20unused=20DEFAULT=5FCUSTOM=5FCOMPONENT=5FCODE?= =?UTF-8?q?=20constant=20The=20FieldFormatter=20class=20now=20inherits=20f?= =?UTF-8?q?rom=20pydantic.BaseModel,=20which=20allows=20for=20improved=20t?= =?UTF-8?q?ype=20checking=20and=20validation=20of=20the=20format=20method?= =?UTF-8?q?=20arguments.=20The=20unused=20DEFAULT=5FCUSTOM=5FCOMPONENT=5FC?= =?UTF-8?q?ODE=20constant=20has=20been=20removed=20to=20clean=20up=20the?= =?UTF-8?q?=20codebase.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../template/frontend_node/formatter/base.py | 3 ++- src/backend/langflow/utils/constants.py | 20 ------------------- 2 files changed, 2 insertions(+), 21 deletions(-) diff --git a/src/backend/langflow/template/frontend_node/formatter/base.py b/src/backend/langflow/template/frontend_node/formatter/base.py index 67e906593..f582bc298 100644 --- a/src/backend/langflow/template/frontend_node/formatter/base.py +++ b/src/backend/langflow/template/frontend_node/formatter/base.py @@ -2,9 +2,10 @@ from abc import ABC, abstractmethod from typing import Optional from langflow.template.field.base import TemplateField +from pydantic import BaseModel -class FieldFormatter(ABC): +class FieldFormatter(BaseModel, ABC): @abstractmethod def format(self, field: TemplateField, name: Optional[str]) -> None: pass diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py index 7d81c5f03..e473d855b 100644 --- a/src/backend/langflow/utils/constants.py +++ b/src/backend/langflow/utils/constants.py @@ -48,24 +48,4 @@ def python_function(text: str) -> str: return text """ -DEFAULT_CUSTOM_COMPONENT_CODE = """ -from langchain.llms import OpenAI -from langchain.chains import ConversationChain -from langchain.memory import ConversationBufferMemory - - -class MyPythonClass: - def my_conversation(self, openai_api_key): - llm = OpenAI( - openai_api_key=openai_api_key, - temperature=0 - ) - return ConversationChain( - llm=llm, verbose=True, memory=ConversationBufferMemory() - ) - - def build(self, openai_api_key: str) -> ConversationChain: - return self.my_conversation(openai_api_key) -""" - DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"]