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

still deprecated, but was misleading before
This commit is contained in:
Erick Friis 2024-03-18 20:17:07 -07:00 committed by GitHub
parent 21c45475c5
commit 95904fe443
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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":