langchain[patch]: update base imports to core (langchain-ai#19248)

still deprecated, but was misleading before
FalkorDB · Mar 30, 2024 · 7074d17 · 7074d17
1 parent 65c7f4d
commit 7074d17
Showing 1 changed file with 10 additions and 8 deletions.
diff --git 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":