🔧 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:
parent
3fe4660c0e
commit
5320da2f06
2 changed files with 4 additions and 1 deletions
|
|
@ -31,6 +31,7 @@ from .constants import (
|
|||
AgentExecutor,
|
||||
NestedDict,
|
||||
Data,
|
||||
BaseLanguageModel,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
|
|
@ -41,6 +42,7 @@ __all__ = [
|
|||
"Chain",
|
||||
"BaseChatMemory",
|
||||
"BaseLLM",
|
||||
"BaseLanguageModel",
|
||||
"BaseLoader",
|
||||
"BaseMemory",
|
||||
"BaseOutputParser",
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue