🔧 fix(field_typing): import BaseLanguageModel in __init__.py to resol…

…ve missing import error 🔧 fix(constants.py): import BaseLanguageModel in constants.py to resolve missing import error
Tesseract-Production · Nov 3, 2023 · 5320da2 · 5320da2
1 parent 3fe4660
commit 5320da2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/backend/langflow/field_typing/__init__.py b/src/backend/langflow/field_typing/__init__.py
@@ -31,6 +31,7 @@
  AgentExecutor,
  NestedDict,
  Data,
+ BaseLanguageModel,
 )
 
 __all__ = [
@@ -41,6 +42,7 @@
  "Chain",
  "BaseChatMemory",
  "BaseLLM",
+ "BaseLanguageModel",
  "BaseLoader",
  "BaseMemory",
  "BaseOutputParser",

diff --git a/src/backend/langflow/field_typing/constants.py b/src/backend/langflow/field_typing/constants.py
@@ -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 @@ class Data:
  "AgentExecutor": AgentExecutor,
  "Tool": Tool,
  "BaseLLM": BaseLLM,
+ "BaseLanguageModel": BaseLanguageModel,
  "PromptTemplate": PromptTemplate,
  "BaseLoader": BaseLoader,
  "Document": Document,