MongoDB agent fixes

.github/workflows/_compile_integration_test.yml

@@ -36,6 +36,13 @@ jobs:
           working-directory: ${{ inputs.working-directory }}
           cache-key: compile-integration
 
+      - name: MongoDB in GitHub Actions
+        uses: supercharge/[email protected]
+        with:
+          mongodb-version: 7.0
+          mongodb-replica-set: test
+          mongodb-port: 27017
+
       - name: Install integration dependencies
         shell: bash
         run: poetry install --with=test_integration

docs/docs/integrations/toolkits/mongo_database.ipynb

@@ -6,7 +6,7 @@
    "source": [
     "# MongoDB Database\n",
     "\n",
-    "This notebook showcases an agent designed to interact with and query a `MongoDB` database using PyMongo. \n",
+    "This notebook showcases an experimental agent designed to interact with and query a `MongoDB` database using PyMongo. \n",
     "The agent is similar to [SQL Database](https://python.langchain.com/docs/integrations/toolkits/sql_database).\n",
     "\n",
     "As this agent is in development, currently it only supports one database per instance `MongoDatabase` class using URI. Additionally, all answers may not be correct, and it is not guaranteed that the agent won't perform destructive commands on your database (or in general) given certain questions."
@@ -25,11 +25,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.agents import create_mongo_agent\n",
-    "from langchain.agents.agent_toolkits import MongoDatabaseToolkit\n",
     "from langchain.agents.agent_types import AgentType\n",
     "from langchain.llms.openai import OpenAI\n",
-    "from langchain.utilities.mongo_database import MongoDatabase\n",
+    "from langchain_experimental.agents.agent_toolkits import (\n",
+    "    MongoDatabaseToolkit,\n",
+    "    create_mongo_agent,\n",
+    ")\n",
+    "from langchain_experimental.utilities import MongoDatabase\n",
     "\n",
     "db = MongoDatabase.from_uri(\"mongodb://localhost:27017/my_db\")\n",
     "db._client[\"my_db\"][\"my_collection\"].insert_many(\n",

libs/experimental/langchain_experimental/agents/agent_toolkits/__init__.py

@@ -1,4 +1,8 @@
 from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_agent
+from langchain_experimental.agents.agent_toolkits.mongo.base import (
+    MongoDatabaseToolkit,
+    create_mongo_agent,
+)
 from langchain_experimental.agents.agent_toolkits.pandas.base import (
     create_pandas_dataframe_agent,
 )
@@ -16,4 +20,6 @@
     "create_spark_dataframe_agent",
     "create_python_agent",
     "create_csv_agent",
+    "create_mongo_agent",
+    "MongoDatabaseToolkit",
 ]

libs/langchain/langchain/agents/agent_toolkits/mongo/base.py → libs/experimental/langchain_experimental/agents/agent_toolkits/mongo/base.py

@@ -1,6 +1,14 @@
 """MongoDB agent."""
 from typing import Any, Dict, List, Optional, Sequence
 
+from langchain.agents.agent import AgentExecutor, BaseSingleActionAgent
+from langchain.agents.agent_types import AgentType
+from langchain.agents.mrkl.base import ZeroShotAgent
+from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS
+from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent
+from langchain.callbacks.base import BaseCallbackManager
+from langchain.chains.llm import LLMChain
+from langchain.tools import BaseTool
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import AIMessage, SystemMessage
 from langchain_core.prompts.chat import (
@@ -9,20 +17,14 @@
     MessagesPlaceholder,
 )
 
-from langchain.agents.agent import AgentExecutor, BaseSingleActionAgent
-from langchain.agents.agent_toolkits.mongo.prompt import (
+from langchain_experimental.agents.agent_toolkits.mongo.prompt import (
     MONGO_FUNCTIONS_SUFFIX,
     MONGO_PREFIX,
     MONGO_SUFFIX,
 )
-from langchain.agents.agent_toolkits.mongo.toolkit import MongoDatabaseToolkit
-from langchain.agents.agent_types import AgentType
-from langchain.agents.mrkl.base import ZeroShotAgent
-from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS
-from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent
-from langchain.callbacks.base import BaseCallbackManager
-from langchain.chains.llm import LLMChain
-from langchain.tools import BaseTool
+from langchain_experimental.agents.agent_toolkits.mongo.toolkit import (
+    MongoDatabaseToolkit,
+)
 
 
 def create_mongo_agent(

libs/langchain/langchain/agents/agent_toolkits/mongo/toolkit.py → libs/experimental/langchain_experimental/agents/agent_toolkits/mongo/toolkit.py

@@ -1,18 +1,18 @@
 """Toolkit for interacting with a Mongo database."""
 from typing import List
 
+from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.tools import BaseTool
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.pydantic_v1 import Field
 
-from langchain.agents.agent_toolkits.base import BaseToolkit
-from langchain.tools import BaseTool
-from langchain.tools.mongo_database.tool import (
+from langchain_experimental.tools.mongo_database.tool import (
     InfoMongoDBTool,
     ListMongoDBTool,
     QueryMongoDBCheckerTool,
     QueryMongoDBTool,
 )
-from langchain.utilities.mongo_database import MongoDatabase
+from langchain_experimental.utilities.mongo_database import MongoDatabase
 
 
 class MongoDatabaseToolkit(BaseToolkit):

libs/experimental/langchain_experimental/tools/__init__.py

@@ -1,3 +1,16 @@
+from langchain_experimental.tools.mongo_database.tool import (
+    InfoMongoDBTool,
+    ListMongoDBTool,
+    QueryMongoDBCheckerTool,
+    QueryMongoDBTool,
+)
 from langchain_experimental.tools.python.tool import PythonAstREPLTool, PythonREPLTool
 
-__all__ = ["PythonREPLTool", "PythonAstREPLTool"]
+__all__ = [
+    "PythonREPLTool",
+    "PythonAstREPLTool",
+    "InfoMongoDBTool",
+    "ListMongoDBTool",
+    "QueryMongoDBCheckerTool",
+    "QueryMongoDBTool",
+]

libs/langchain/langchain/tools/mongo_database/tool.py → libs/experimental/langchain_experimental/tools/mongo_database/tool.py

@@ -11,9 +11,9 @@
 )
 from langchain.chains.llm import LLMChain
 from langchain.prompts import PromptTemplate
-from langchain.utilities.mongo_database import MongoDatabase
+from langchain_experimental.utilities.mongo_database import MongoDatabase
 from langchain.tools.base import BaseTool
-from langchain.tools.mongo_database.prompt import QUERY_CHECKER
+from langchain_experimental.tools.mongo_database.prompt import QUERY_CHECKER
 
 
 class BaseMongoDBTool(BaseModel):

libs/experimental/langchain_experimental/utilities/__init__.py

@@ -1,3 +1,4 @@
+from langchain_experimental.utilities.mongo_database import MongoDatabase
 from langchain_experimental.utilities.python import PythonREPL
 
-__all__ = ["PythonREPL"]
+__all__ = ["PythonREPL", "MongoDatabase"]