From 7074d17aefc64ffeda909aaf64419fb1f85e840a Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Mon, 18 Mar 2024 20:17:07 -0700 Subject: [PATCH] langchain[patch]: update base imports to core (#19248) still deprecated, but was misleading before --- libs/langchain/langchain/__init__.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/libs/langchain/langchain/__init__.py b/libs/langchain/langchain/__init__.py index dad447585e4f0..78afad3a0aaf8 100644 --- a/libs/langchain/langchain/__init__.py +++ b/libs/langchain/langchain/__init__.py @@ -233,27 +233,29 @@ def __getattr__(name: str) -> Any: elif name == "FewShotPromptTemplate": from langchain_core.prompts import FewShotPromptTemplate - _warn_on_import(name, replacement="langchain.prompts.FewShotPromptTemplate") + _warn_on_import( + name, replacement="langchain_core.prompts.FewShotPromptTemplate" + ) return FewShotPromptTemplate elif name == "Prompt": - from langchain.prompts import Prompt + from langchain_core.prompts import PromptTemplate - _warn_on_import(name, replacement="langchain.prompts.Prompt") + _warn_on_import(name, replacement="langchain_core.prompts.PromptTemplate") - return Prompt + # it's renamed as prompt template anyways + # this is just for backwards compat + return PromptTemplate elif name == "PromptTemplate": from langchain_core.prompts import PromptTemplate - _warn_on_import(name, replacement="langchain.prompts.PromptTemplate") + _warn_on_import(name, replacement="langchain_core.prompts.PromptTemplate") return PromptTemplate elif name == "BasePromptTemplate": from langchain_core.prompts import BasePromptTemplate - _warn_on_import( - name, replacement="langchain.schema.prompt_template.BasePromptTemplate" - ) + _warn_on_import(name, replacement="langchain_core.prompts.BasePromptTemplate") return BasePromptTemplate elif name == "ArxivAPIWrapper":