langchain[patch], docs: update agent toolkit imports (#15434)

cookbook/autogpt/marathon_times.ipynb

@@ -39,8 +39,8 @@
     "\n",
     "import nest_asyncio\n",
     "import pandas as pd\n",
-    "from langchain.agents.agent_toolkits.pandas.base import create_pandas_dataframe_agent\n",
     "from langchain.docstore.document import Document\n",
+    "from langchain_community.agent_toolkits.pandas.base import create_pandas_dataframe_agent\n",
     "from langchain_community.chat_models import ChatOpenAI\n",
     "from langchain_experimental.autonomous_agents import AutoGPT\n",
     "\n",

cookbook/custom_agent_with_plugin_retrieval.ipynb

@@ -40,10 +40,10 @@
     "    AgentOutputParser,\n",
     "    LLMSingleActionAgent,\n",
     ")\n",
-    "from langchain.agents.agent_toolkits import NLAToolkit\n",
     "from langchain.chains import LLMChain\n",
     "from langchain.prompts import StringPromptTemplate\n",
     "from langchain.schema import AgentAction, AgentFinish\n",
+    "from langchain_community.agent_toolkits import NLAToolkit\n",
     "from langchain_community.llms import OpenAI\n",
     "from langchain_community.tools.plugin import AIPlugin"
    ]

cookbook/custom_agent_with_plugin_retrieval_using_plugnplai.ipynb

@@ -65,10 +65,10 @@
     "    AgentOutputParser,\n",
     "    LLMSingleActionAgent,\n",
     ")\n",
-    "from langchain.agents.agent_toolkits import NLAToolkit\n",
     "from langchain.chains import LLMChain\n",
     "from langchain.prompts import StringPromptTemplate\n",
     "from langchain.schema import AgentAction, AgentFinish\n",
+    "from langchain_community.agent_toolkits import NLAToolkit\n",
     "from langchain_community.llms import OpenAI\n",
     "from langchain_community.tools.plugin import AIPlugin"
    ]

cookbook/databricks_sql_db.ipynb

@@ -177,7 +177,7 @@
    "outputs": [],
    "source": [
     "from langchain.agents import create_sql_agent\n",
-    "from langchain.agents.agent_toolkits import SQLDatabaseToolkit\n",
+    "from langchain_community.agent_toolkits import SQLDatabaseToolkit\n",
     "\n",
     "toolkit = SQLDatabaseToolkit(db=db, llm=llm)\n",
     "agent = create_sql_agent(llm=llm, toolkit=toolkit, verbose=True)"

docs/docs/integrations/platforms/google.mdx

@@ -504,7 +504,7 @@ pip install google-api-python-client google-auth-oauthlib google-auth-httplib2
 See a [usage example and authorization instructions](/docs/integrations/toolkits/gmail).
 
 ```python
-from langchain.agents.agent_toolkits import GmailToolkit
+from langchain_community.agent_toolkits import GmailToolkit
 ```
 
 

docs/docs/integrations/platforms/microsoft.mdx

@@ -288,7 +288,7 @@ pip install azure-ai-formrecognizer azure-cognitiveservices-speech azure-ai-visi
 See a [usage example](/docs/integrations/toolkits/azure_cognitive_services).
 
 ```python
-from langchain.agents.agent_toolkits import O365Toolkit
+from langchain_community.agent_toolkits import O365Toolkit
 ```
 ### Microsoft Office 365 email and calendar
 
@@ -302,7 +302,7 @@ pip install O365
 See a [usage example](/docs/integrations/toolkits/office365).
 
 ```python
-from langchain.agents.agent_toolkits import O365Toolkit
+from langchain_community.agent_toolkits import O365Toolkit
 ```
 
 ### Microsoft Azure PowerBI
@@ -316,7 +316,7 @@ pip install azure-identity
 See a [usage example](/docs/integrations/toolkits/powerbi).
 
 ```python
-from langchain.agents.agent_toolkits import PowerBIToolkit
+from langchain_community.agent_toolkits import PowerBIToolkit
 from langchain.utilities.powerbi import PowerBIDataset
 ```
 

docs/docs/integrations/providers/ainetwork.mdx

@@ -18,6 +18,6 @@ You need to set the `AIN_BLOCKCHAIN_ACCOUNT_PRIVATE_KEY` environmental variable
 See a [usage example](/docs/integrations/toolkits/ainetwork).
 
 ```python
-from langchain.agents.agent_toolkits.ainetwork.toolkit import AINetworkToolkit
+from langchain_community.agent_toolkits.ainetwork.toolkit import AINetworkToolkit
 ```
 

docs/docs/integrations/providers/cnosdb.mdx

@@ -65,7 +65,7 @@ Answer:The average temperature of air at station XiaoMaiDao between October 19,
 This example demonstrates the use of the SQL Database Agent for answering questions over a CnosDB.
 ```python
 from langchain.agents import create_sql_agent
-from langchain.agents.agent_toolkits import SQLDatabaseToolkit
+from langchain_community.agent_toolkits import SQLDatabaseToolkit
 
 toolkit = SQLDatabaseToolkit(db=db, llm=llm)
 agent = create_sql_agent(llm=llm, toolkit=toolkit, verbose=True)

docs/docs/integrations/toolkits/ainetwork.ipynb

@@ -108,7 +108,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits.ainetwork.toolkit import AINetworkToolkit\n",
+    "from langchain_community.agent_toolkits.ainetwork.toolkit import AINetworkToolkit\n",
     "\n",
     "toolkit = AINetworkToolkit()\n",
     "tools = toolkit.get_tools()\n",

docs/docs/integrations/toolkits/amadeus.ipynb

@@ -65,7 +65,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits.amadeus.toolkit import AmadeusToolkit\n",
+    "from langchain_community.agent_toolkits.amadeus.toolkit import AmadeusToolkit\n",
     "\n",
     "toolkit = AmadeusToolkit()\n",
     "tools = toolkit.get_tools()"

docs/docs/integrations/toolkits/azure_cognitive_services.ipynb

@@ -66,7 +66,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits import AzureCognitiveServicesToolkit\n",
+    "from langchain_community.agent_toolkits import AzureCognitiveServicesToolkit\n",
     "\n",
     "toolkit = AzureCognitiveServicesToolkit()"
    ]

docs/docs/integrations/toolkits/clickup.ipynb

@@ -22,8 +22,8 @@
     "from datetime import datetime\n",
     "\n",
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits.clickup.toolkit import ClickupToolkit\n",
     "from langchain.utilities.clickup import ClickupAPIWrapper\n",
+    "from langchain_community.agent_toolkits.clickup.toolkit import ClickupToolkit\n",
     "from langchain_community.llms import OpenAI"
    ]
   },

docs/docs/integrations/toolkits/github.ipynb

@@ -107,8 +107,8 @@
     "import os\n",
     "\n",
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits.github.toolkit import GitHubToolkit\n",
     "from langchain.utilities.github import GitHubAPIWrapper\n",
+    "from langchain_community.agent_toolkits.github.toolkit import GitHubToolkit\n",
     "from langchain_community.chat_models import ChatOpenAI"
    ]
   },

docs/docs/integrations/toolkits/gitlab.ipynb

@@ -101,8 +101,8 @@
     "import os\n",
     "\n",
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits.gitlab.toolkit import GitLabToolkit\n",
     "from langchain.utilities.gitlab import GitLabAPIWrapper\n",
+    "from langchain_community.agent_toolkits.gitlab.toolkit import GitLabToolkit\n",
     "from langchain_community.llms import OpenAI"
    ]
   },

docs/docs/integrations/toolkits/gmail.ipynb

@@ -40,7 +40,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits import GmailToolkit\n",
+    "from langchain_community.agent_toolkits import GmailToolkit\n",
     "\n",
     "toolkit = GmailToolkit()"
    ]

docs/docs/integrations/toolkits/jira.ipynb

@@ -50,8 +50,8 @@
     "import os\n",
     "\n",
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits.jira.toolkit import JiraToolkit\n",
     "from langchain.utilities.jira import JiraAPIWrapper\n",
+    "from langchain_community.agent_toolkits.jira.toolkit import JiraToolkit\n",
     "from langchain_community.llms import OpenAI"
    ]
   },

docs/docs/integrations/toolkits/json.ipynb

@@ -34,7 +34,7 @@
    "source": [
     "import yaml\n",
     "from langchain.agents import create_json_agent\n",
-    "from langchain.agents.agent_toolkits import JsonToolkit\n",
+    "from langchain_community.agent_toolkits import JsonToolkit\n",
     "from langchain_community.llms.openai import OpenAI\n",
     "from langchain_community.tools.json.tool import JsonSpec"
    ]

docs/docs/integrations/toolkits/multion.ipynb

@@ -26,7 +26,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits import MultionToolkit\n",
+    "from langchain_community.agent_toolkits import MultionToolkit\n",
     "\n",
     "toolkit = MultionToolkit()\n",
     "\n",
@@ -83,7 +83,7 @@
     "from langchain_community.llms import OpenAI\n",
     "\n",
     "llm = OpenAI(temperature=0)\n",
-    "from langchain.agents.agent_toolkits import MultionToolkit\n",
+    "from langchain_community.agent_toolkits import MultionToolkit\n",
     "\n",
     "toolkit = MultionToolkit()\n",
     "tools = toolkit.get_tools()\n",

docs/docs/integrations/toolkits/nasa.ipynb

@@ -30,8 +30,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits.nasa.toolkit import NasaToolkit\n",
     "from langchain.utilities.nasa import NasaAPIWrapper\n",
+    "from langchain_community.agent_toolkits.nasa.toolkit import NasaToolkit\n",
     "from langchain_community.llms import OpenAI\n",
     "\n",
     "llm = OpenAI(temperature=0, openai_api_key=\"\")\n",

docs/docs/integrations/toolkits/office365.ipynb

@@ -77,7 +77,7 @@
     }
    ],
    "source": [
-    "from langchain.agents.agent_toolkits import O365Toolkit\n",
+    "from langchain_community.agent_toolkits import O365Toolkit\n",
     "\n",
     "toolkit = O365Toolkit()\n",
     "tools = toolkit.get_tools()\n",

docs/docs/integrations/toolkits/openapi.ipynb

@@ -105,7 +105,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits.openapi.spec import reduce_openapi_spec"
+    "from langchain_community.agent_toolkits.openapi.spec import reduce_openapi_spec"
    ]
   },
   {
@@ -260,7 +260,7 @@
     }
    ],
    "source": [
-    "from langchain.agents.agent_toolkits.openapi import planner\n",
+    "from langchain_community.agent_toolkits.openapi import planner\n",
     "from langchain_community.llms.openai import OpenAI\n",
     "\n",
     "llm = OpenAI(model_name=\"gpt-4\", temperature=0.0)"
@@ -583,7 +583,7 @@
    "outputs": [],
    "source": [
     "from langchain.agents import create_openapi_agent\n",
-    "from langchain.agents.agent_toolkits import OpenAPIToolkit\n",
+    "from langchain_community.agent_toolkits import OpenAPIToolkit\n",
     "from langchain_community.llms.openai import OpenAI\n",
     "from langchain_community.tools.json.tool import JsonSpec"
    ]

docs/docs/integrations/toolkits/openapi_nla.ipynb

@@ -26,8 +26,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits import NLAToolkit\n",
     "from langchain.requests import Requests\n",
+    "from langchain_community.agent_toolkits import NLAToolkit\n",
     "from langchain_community.llms import OpenAI"
    ]
   },

docs/docs/integrations/toolkits/powerbi.ipynb

@@ -38,8 +38,8 @@
    "outputs": [],
    "source": [
     "from azure.identity import DefaultAzureCredential\n",
-    "from langchain.agents.agent_toolkits import PowerBIToolkit, create_pbi_agent\n",
     "from langchain.utilities.powerbi import PowerBIDataset\n",
+    "from langchain_community.agent_toolkits import PowerBIToolkit, create_pbi_agent\n",
     "from langchain_community.chat_models import ChatOpenAI"
    ]
   },

docs/docs/integrations/toolkits/slack.ipynb

@@ -54,7 +54,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits import SlackToolkit\n",
+    "from langchain_community.agent_toolkits import SlackToolkit\n",
     "\n",
     "toolkit = SlackToolkit()\n",
     "tools = toolkit.get_tools()\n",

docs/docs/integrations/toolkits/spark_sql.ipynb

@@ -25,8 +25,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import create_spark_sql_agent\n",
-    "from langchain.agents.agent_toolkits import SparkSQLToolkit\n",
     "from langchain.utilities.spark_sql import SparkSQL\n",
+    "from langchain_community.agent_toolkits import SparkSQLToolkit\n",
     "from langchain_community.chat_models import ChatOpenAI"
    ]
   },

docs/docs/integrations/toolkits/sql_database.ipynb

@@ -35,9 +35,9 @@
    "outputs": [],
    "source": [
     "from langchain.agents import create_sql_agent\n",
-    "from langchain.agents.agent_toolkits import SQLDatabaseToolkit\n",
     "from langchain.agents.agent_types import AgentType\n",
     "from langchain.sql_database import SQLDatabase\n",
+    "from langchain_community.agent_toolkits import SQLDatabaseToolkit\n",
     "from langchain_community.llms.openai import OpenAI"
    ]
   },

docs/docs/integrations/toolkits/steam.ipynb

@@ -75,8 +75,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits.steam.toolkit import SteamToolkit\n",
     "from langchain.utilities.steam import SteamWebAPIWrapper\n",
+    "from langchain_community.agent_toolkits.steam.toolkit import SteamToolkit\n",
     "from langchain_community.llms import OpenAI"
    ]
   },

docs/docs/integrations/tools/filesystem.ipynb

@@ -28,7 +28,7 @@
    "source": [
     "from tempfile import TemporaryDirectory\n",
     "\n",
-    "from langchain.agents.agent_toolkits import FileManagementToolkit\n",
+    "from langchain_community.agent_toolkits import FileManagementToolkit\n",
     "\n",
     "# We'll make a temporary directory to avoid clutter\n",
     "working_directory = TemporaryDirectory()"

docs/docs/integrations/tools/zapier.ipynb

@@ -61,8 +61,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import AgentType, initialize_agent\n",
-    "from langchain.agents.agent_toolkits import ZapierToolkit\n",
     "from langchain.utilities.zapier import ZapierNLAWrapper\n",
+    "from langchain_community.agent_toolkits import ZapierToolkit\n",
     "from langchain_community.llms import OpenAI"
    ]
   },

docs/docs/use_cases/qa_structured/sql.ipynb

@@ -518,10 +518,10 @@
    "outputs": [],
    "source": [
     "from langchain.agents import create_sql_agent\n",
-    "from langchain.agents.agent_toolkits import SQLDatabaseToolkit\n",
     "\n",
     "# from langchain.agents import AgentExecutor\n",
     "from langchain.agents.agent_types import AgentType\n",
+    "from langchain_community.agent_toolkits import SQLDatabaseToolkit\n",
     "\n",
     "db = SQLDatabase.from_uri(\"sqlite:///Chinook.db\")\n",
     "\n",
@@ -809,7 +809,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits import create_retriever_tool\n",
+    "from langchain_community.agent_toolkits import create_retriever_tool\n",
     "\n",
     "tool_description = \"\"\"\n",
     "This tool will help you understand similar examples to adapt them to the user question.\n",
@@ -836,8 +836,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import AgentType, create_sql_agent\n",
-    "from langchain.agents.agent_toolkits import SQLDatabaseToolkit\n",
     "from langchain.utilities import SQLDatabase\n",
+    "from langchain_community.agent_toolkits import SQLDatabaseToolkit\n",
     "from langchain_community.chat_models import ChatOpenAI\n",
     "\n",
     "db = SQLDatabase.from_uri(\"sqlite:///Chinook.db\")\n",
@@ -969,7 +969,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents.agent_toolkits import create_retriever_tool\n",
+    "from langchain_community.agent_toolkits import create_retriever_tool\n",
     "from langchain_community.embeddings.openai import OpenAIEmbeddings\n",
     "from langchain_community.vectorstores import FAISS\n",
     "\n",
@@ -995,8 +995,8 @@
    "outputs": [],
    "source": [
     "from langchain.agents import AgentType, create_sql_agent\n",
-    "from langchain.agents.agent_toolkits import SQLDatabaseToolkit\n",
     "from langchain.utilities import SQLDatabase\n",
+    "from langchain_community.agent_toolkits import SQLDatabaseToolkit\n",
     "from langchain_community.chat_models import ChatOpenAI\n",
     "\n",
     "# db = SQLDatabase.from_uri(\"sqlite:///Chinook.db\")\n",

libs/langchain/langchain/agents/__init__.py

@@ -31,6 +31,14 @@ Agents select and use **Tools** and **Toolkits** for actions.
 from pathlib import Path
 from typing import Any
 
+from langchain_community.agent_toolkits import (
+    create_json_agent,
+    create_openapi_agent,
+    create_pbi_agent,
+    create_pbi_chat_agent,
+    create_spark_sql_agent,
+    create_sql_agent,
+)
 from langchain_core._api.path import as_import_path
 
 from langchain.agents.agent import (
@@ -42,13 +50,7 @@ from langchain.agents.agent import (
     LLMSingleActionAgent,
 )
 from langchain.agents.agent_iterator import AgentExecutorIterator
-from langchain.agents.agent_toolkits import (
-    create_json_agent,
-    create_openapi_agent,
-    create_pbi_agent,
-    create_pbi_chat_agent,
-    create_spark_sql_agent,
-    create_sql_agent,
+from langchain.agents.agent_toolkits.vectorstore.base import (
     create_vectorstore_agent,
     create_vectorstore_router_agent,
 )

libs/langchain/langchain/agents/agent_toolkits/__init__.py

@@ -16,39 +16,42 @@ See [Security](https://python.langchain.com/docs/security) for more information.
 from pathlib import Path
 from typing import Any
 
-from langchain_core._api.path import as_import_path
-
-from langchain.agents.agent_toolkits.ainetwork.toolkit import AINetworkToolkit
-from langchain.agents.agent_toolkits.amadeus.toolkit import AmadeusToolkit
-from langchain.agents.agent_toolkits.azure_cognitive_services import (
+from langchain_community.agent_toolkits.ainetwork.toolkit import AINetworkToolkit
+from langchain_community.agent_toolkits.amadeus.toolkit import AmadeusToolkit
+from langchain_community.agent_toolkits.azure_cognitive_services import (
     AzureCognitiveServicesToolkit,
 )
+from langchain_community.agent_toolkits.file_management.toolkit import (
+    FileManagementToolkit,
+)
+from langchain_community.agent_toolkits.gmail.toolkit import GmailToolkit
+from langchain_community.agent_toolkits.jira.toolkit import JiraToolkit
+from langchain_community.agent_toolkits.json.base import create_json_agent
+from langchain_community.agent_toolkits.json.toolkit import JsonToolkit
+from langchain_community.agent_toolkits.multion.toolkit import MultionToolkit
+from langchain_community.agent_toolkits.nasa.toolkit import NasaToolkit
+from langchain_community.agent_toolkits.nla.toolkit import NLAToolkit
+from langchain_community.agent_toolkits.office365.toolkit import O365Toolkit
+from langchain_community.agent_toolkits.openapi.base import create_openapi_agent
+from langchain_community.agent_toolkits.openapi.toolkit import OpenAPIToolkit
+from langchain_community.agent_toolkits.playwright.toolkit import (
+    PlayWrightBrowserToolkit,
+)
+from langchain_community.agent_toolkits.powerbi.base import create_pbi_agent
+from langchain_community.agent_toolkits.powerbi.chat_base import create_pbi_chat_agent
+from langchain_community.agent_toolkits.powerbi.toolkit import PowerBIToolkit
+from langchain_community.agent_toolkits.slack.toolkit import SlackToolkit
+from langchain_community.agent_toolkits.spark_sql.base import create_spark_sql_agent
+from langchain_community.agent_toolkits.spark_sql.toolkit import SparkSQLToolkit
+from langchain_community.agent_toolkits.sql.base import create_sql_agent
+from langchain_community.agent_toolkits.sql.toolkit import SQLDatabaseToolkit
+from langchain_community.agent_toolkits.steam.toolkit import SteamToolkit
+from langchain_community.agent_toolkits.zapier.toolkit import ZapierToolkit
+from langchain_core._api.path import as_import_path
+
 from langchain.agents.agent_toolkits.conversational_retrieval.openai_functions import (
     create_conversational_retrieval_agent,
 )
-from langchain.agents.agent_toolkits.file_management.toolkit import (
-    FileManagementToolkit,
-)
-from langchain.agents.agent_toolkits.gmail.toolkit import GmailToolkit
-from langchain.agents.agent_toolkits.jira.toolkit import JiraToolkit
-from langchain.agents.agent_toolkits.json.base import create_json_agent
-from langchain.agents.agent_toolkits.json.toolkit import JsonToolkit
-from langchain.agents.agent_toolkits.multion.toolkit import MultionToolkit
-from langchain.agents.agent_toolkits.nasa.toolkit import NasaToolkit
-from langchain.agents.agent_toolkits.nla.toolkit import NLAToolkit
-from langchain.agents.agent_toolkits.office365.toolkit import O365Toolkit
-from langchain.agents.agent_toolkits.openapi.base import create_openapi_agent
-from langchain.agents.agent_toolkits.openapi.toolkit import OpenAPIToolkit
-from langchain.agents.agent_toolkits.playwright.toolkit import PlayWrightBrowserToolkit
-from langchain.agents.agent_toolkits.powerbi.base import create_pbi_agent
-from langchain.agents.agent_toolkits.powerbi.chat_base import create_pbi_chat_agent
-from langchain.agents.agent_toolkits.powerbi.toolkit import PowerBIToolkit
-from langchain.agents.agent_toolkits.slack.toolkit import SlackToolkit
-from langchain.agents.agent_toolkits.spark_sql.base import create_spark_sql_agent
-from langchain.agents.agent_toolkits.spark_sql.toolkit import SparkSQLToolkit
-from langchain.agents.agent_toolkits.sql.base import create_sql_agent
-from langchain.agents.agent_toolkits.sql.toolkit import SQLDatabaseToolkit
-from langchain.agents.agent_toolkits.steam.toolkit import SteamToolkit
 from langchain.agents.agent_toolkits.vectorstore.base import (
     create_vectorstore_agent,
     create_vectorstore_router_agent,
@@ -58,7 +61,6 @@ from langchain.agents.agent_toolkits.vectorstore.toolkit import (
     VectorStoreRouterToolkit,
     VectorStoreToolkit,
 )
-from langchain.agents.agent_toolkits.zapier.toolkit import ZapierToolkit
 from langchain.tools.retriever import create_retriever_tool
 
 DEPRECATED_AGENTS = [

libs/langchain/langchain/agents/agent_toolkits/file_management/__init__.py

@@ -1,6 +1,6 @@
 """Local file management toolkit."""
 
-from langchain.agents.agent_toolkits.file_management.toolkit import (
+from langchain_community.agent_toolkits.file_management.toolkit import (
     FileManagementToolkit,
 )
 

libs/langchain/langchain/agents/agent_toolkits/playwright/__init__.py

@@ -1,4 +1,6 @@
 """Playwright browser toolkit."""
-from langchain.agents.agent_toolkits.playwright.toolkit import PlayWrightBrowserToolkit
+from langchain_community.agent_toolkits.playwright.toolkit import (
+    PlayWrightBrowserToolkit,
+)
 
 __all__ = ["PlayWrightBrowserToolkit"]

libs/langchain/langchain/agents/agent_toolkits/vectorstore/toolkit.py

@@ -1,6 +1,7 @@
 """Toolkit for interacting with a vector store."""
 from typing import List
 
+from langchain_community.agent_toolkits.base import BaseToolkit
 from langchain_community.llms.openai import OpenAI
 from langchain_community.tools.vectorstore.tool import (
     VectorStoreQATool,
@@ -10,7 +11,6 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.vectorstores import VectorStore
 
-from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.tools import BaseTool
 
 

libs/langchain/tests/integration_tests/agent/test_ainetwork_agent.py

@@ -8,11 +8,11 @@ from typing import Any
 from urllib.error import HTTPError
 
 import pytest
+from langchain_community.agent_toolkits.ainetwork.toolkit import AINetworkToolkit
 from langchain_community.chat_models import ChatOpenAI
 from langchain_community.tools.ainetwork.utils import authenticate
 
 from langchain.agents import AgentType, initialize_agent
-from langchain.agents.agent_toolkits.ainetwork.toolkit import AINetworkToolkit
 
 
 class Match(Enum):

libs/langchain/tests/integration_tests/agent/test_powerbi_agent.py

@@ -1,7 +1,7 @@
 import pytest
+from langchain_community.agent_toolkits import PowerBIToolkit, create_pbi_agent
 from langchain_community.chat_models import ChatOpenAI
 
-from langchain.agents.agent_toolkits import PowerBIToolkit, create_pbi_agent
 from langchain.utilities.powerbi import PowerBIDataset
 from langchain.utils import get_from_env
 

libs/langchain/tests/unit_tests/agents/test_sql.py

@@ -1,5 +1,6 @@
+from langchain_community.agent_toolkits import SQLDatabaseToolkit
+
 from langchain.agents import create_sql_agent
-from langchain.agents.agent_toolkits import SQLDatabaseToolkit
 from langchain.utilities.sql_database import SQLDatabase
 from tests.unit_tests.llms.fake_llm import FakeLLM