From cc3841c9bff8a8c38d4aa7b5bf9c1384e919599b Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Mon, 1 May 2023 18:40:24 -0300 Subject: [PATCH] refactor(agents): import BaseLanguageModel from base_language module instead of schema module --- src/backend/langflow/interface/agents/custom.py | 2 +- src/backend/langflow/interface/agents/prebuilt.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index 5942e3811..67827bd9d 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -28,7 +28,7 @@ from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS as SQL_FORMAT_INSTRUCTIONS from langchain.llms.base import BaseLLM from langchain.memory.chat_memory import BaseChatMemory -from langchain.schema import BaseLanguageModel +from langchain.base_language import BaseLanguageModel from langchain.sql_database import SQLDatabase from langchain.tools.python.tool import PythonAstREPLTool from langchain.tools.sql_database.prompt import QUERY_CHECKER diff --git a/src/backend/langflow/interface/agents/prebuilt.py b/src/backend/langflow/interface/agents/prebuilt.py index e20ec3bde..957dd4f7d 100644 --- a/src/backend/langflow/interface/agents/prebuilt.py +++ b/src/backend/langflow/interface/agents/prebuilt.py @@ -3,7 +3,7 @@ from langchain.agents import AgentExecutor, ZeroShotAgent from langchain.agents.agent_toolkits.json.prompt import JSON_PREFIX, JSON_SUFFIX from langchain.agents.agent_toolkits.json.toolkit import JsonToolkit from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS -from langchain.schema import BaseLanguageModel +from langchain.base_language import BaseLanguageModel class MalfoyAgent(AgentExecutor):