🔧 fix(field_typing): import BaseLanguageModel in __init__.py to resolve missing import error

🔧 fix(constants.py): import BaseLanguageModel in constants.py to resolve missing import error
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-02 22:19:19 -03:00
commit 5320da2f06
2 changed files with 4 additions and 1 deletions

View file

@ -31,6 +31,7 @@ from .constants import (
AgentExecutor,
NestedDict,
Data,
BaseLanguageModel,
)
__all__ = [
@ -41,6 +42,7 @@ __all__ = [
"Chain",
"BaseChatMemory",
"BaseLLM",
"BaseLanguageModel",
"BaseLoader",
"BaseMemory",
"BaseOutputParser",

View file

@ -1,7 +1,7 @@
from langchain.agents.agent import AgentExecutor
from langchain.chains.base import Chain
from langchain.document_loaders.base import BaseLoader
from langchain.llms.base import BaseLLM
from langchain.llms.base import BaseLLM, BaseLanguageModel
from langchain.memory.chat_memory import BaseChatMemory
from langchain.prompts import PromptTemplate
from langchain.schema import BaseOutputParser, BaseRetriever, Document
@ -25,6 +25,7 @@ LANGCHAIN_BASE_TYPES = {
"AgentExecutor": AgentExecutor,
"Tool": Tool,
"BaseLLM": BaseLLM,
"BaseLanguageModel": BaseLanguageModel,
"PromptTemplate": PromptTemplate,
"BaseLoader": BaseLoader,
"Document": Document,