From 65c65165757406dd4c57561b59bd6c8b905b2642 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 3 Oct 2023 17:05:59 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20chore(constants.py):=20update=20?= =?UTF-8?q?import=20statements=20to=20use=20the=20correct=20module=20paths?= =?UTF-8?q?=20for=20PromptTemplate=20=F0=9F=94=80=20chore(utils.py):=20upd?= =?UTF-8?q?ate=20import=20statements=20to=20use=20the=20correct=20module?= =?UTF-8?q?=20paths=20for=20PromptTemplate=20=F0=9F=94=80=20chore(test=5Fc?= =?UTF-8?q?ustom=5Fcomponent.py):=20update=20import=20statements=20to=20us?= =?UTF-8?q?e=20the=20correct=20module=20paths=20for=20PromptTemplate?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/custom/constants.py | 4 ++-- src/backend/langflow/interface/importing/utils.py | 2 +- tests/test_custom_component.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/interface/custom/constants.py b/src/backend/langflow/interface/custom/constants.py index f5911198e..58ecef637 100644 --- a/src/backend/langflow/interface/custom/constants.py +++ b/src/backend/langflow/interface/custom/constants.py @@ -1,4 +1,4 @@ -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from langchain.chains.base import Chain from langchain.document_loaders.base import BaseLoader from langchain.schema.embeddings import Embeddings @@ -45,7 +45,7 @@ DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent from langchain.llms.base import BaseLLM from langchain.chains import LLMChain -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from langchain.schema import Document import requests diff --git a/src/backend/langflow/interface/importing/utils.py b/src/backend/langflow/interface/importing/utils.py index 000eed205..44d72df42 100644 --- a/src/backend/langflow/interface/importing/utils.py +++ b/src/backend/langflow/interface/importing/utils.py @@ -3,7 +3,7 @@ import importlib from typing import Any, Type -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from langchain.agents import Agent from langchain.base_language import BaseLanguageModel from langchain.chains.base import Chain diff --git a/tests/test_custom_component.py b/tests/test_custom_component.py index 1695cfd38..47c9cbfb2 100644 --- a/tests/test_custom_component.py +++ b/tests/test_custom_component.py @@ -21,7 +21,7 @@ from langflow.interface.custom.custom_component import CustomComponent from langchain.llms.base import BaseLLM from langchain.chains import LLMChain -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from langchain.schema import Document import requests