🔧 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, AgentExecutor,
NestedDict, NestedDict,
Data, Data,
BaseLanguageModel,
) )
__all__ = [ __all__ = [
@ -41,6 +42,7 @@ __all__ = [
"Chain", "Chain",
"BaseChatMemory", "BaseChatMemory",
"BaseLLM", "BaseLLM",
"BaseLanguageModel",
"BaseLoader", "BaseLoader",
"BaseMemory", "BaseMemory",
"BaseOutputParser", "BaseOutputParser",

View file

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