From 2ae46f2b080ee9dd751d5d56902aef2648c42e59 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 11 Jun 2024 22:56:43 -0300 Subject: [PATCH] refactor: Update LCModelComponent inheritance and imports Update the LCModelComponent class in model.py to inherit from the Component class instead of the deprecated CustomComponent class. Also, update the import statement for the Component class. This change ensures compatibility with the latest version of the langflow library and improves code readability. --- src/backend/base/langflow/base/models/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/base/models/model.py b/src/backend/base/langflow/base/models/model.py index 74d350446..a8069dcf9 100644 --- a/src/backend/base/langflow/base/models/model.py +++ b/src/backend/base/langflow/base/models/model.py @@ -5,11 +5,11 @@ from langchain_core.language_models.chat_models import BaseChatModel from langchain_core.language_models.llms import LLM from langchain_core.messages import AIMessage, HumanMessage, SystemMessage -from langflow.custom import CustomComponent +from langflow.custom import Component from langflow.field_typing.prompt import Prompt -class LCModelComponent(CustomComponent): +class LCModelComponent(Component): display_name: str = "Model Name" description: str = "Model Description"