Amy mongo integration (#3)

Fix formatting/linting and poetry.lock

Author: amiaxys

docs/docs/integrations/toolkits/mongo_database.ipynb

@@ -32,18 +32,20 @@
     "from langchain.utilities.mongo_database import MongoDatabase\n",
     "\n",
     "db = MongoDatabase.from_uri(\"mongodb://localhost:27017/my_db\")\n",
-    "db._client[\"my_db\"][\"my_collection\"].insert_many([\n",
-    "    {\"text\": \"Hello, world!\", \"language\": \"en\"},\n",
-    "    {\"text\": \"Bonjour, monde!\", \"language\": \"fr\"},\n",
-    "    {\"text\": \"Hola, mundo!\", \"language\": \"es\"},\n",
-    "    {\"text\": \"Hallo, Welt!\", \"language\": \"de\"},\n",
-    "    {\"text\": \"Ciao, mondo!\", \"language\": \"it\"},\n",
-    "    {\"text\": \"Olá, mundo!\", \"language\": \"pt\"},\n",
-    "    {\"text\": \"Привет, мир!\", \"language\": \"ru\"},\n",
-    "    {\"text\": \"你好，世界！\", \"language\": \"zh\"},\n",
-    "    {\"text\": \"こんにちは世界！\", \"language\": \"ja\"},\n",
-    "    {\"text\": \"안녕, 세상아!\", \"language\": \"ko\"},\n",
-    "])\n",
+    "db._client[\"my_db\"][\"my_collection\"].insert_many(\n",
+    "    [\n",
+    "        {\"text\": \"Hello, world!\", \"language\": \"en\"},\n",
+    "        {\"text\": \"Bonjour, monde!\", \"language\": \"fr\"},\n",
+    "        {\"text\": \"Hola, mundo!\", \"language\": \"es\"},\n",
+    "        {\"text\": \"Hallo, Welt!\", \"language\": \"de\"},\n",
+    "        {\"text\": \"Ciao, mondo!\", \"language\": \"it\"},\n",
+    "        {\"text\": \"Olá, mundo!\", \"language\": \"pt\"},\n",
+    "        {\"text\": \"Привет, мир!\", \"language\": \"ru\"},\n",
+    "        {\"text\": \"你好，世界！\", \"language\": \"zh\"},\n",
+    "        {\"text\": \"こんにちは世界！\", \"language\": \"ja\"},\n",
+    "        {\"text\": \"안녕, 세상아!\", \"language\": \"ko\"},\n",
+    "    ]\n",
+    ")\n",
     "# insert more documents if you would like\n",
     "toolkit = MongoDatabaseToolkit(db=db, llm=OpenAI(temperature=0))"
    ]
@@ -67,7 +69,7 @@
     "    llm=OpenAI(temperature=0),\n",
     "    toolkit=toolkit,\n",
     "    verbose=True,\n",
-    "    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION\n",
+    "    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,\n",
     ")"
    ]
   },

docs/docs/integrations/toolkits/sql_database.ipynb

@@ -3,7 +3,7 @@
 
 import re
 from ast import literal_eval
-from pprint import pformat, pprint
+from pprint import pformat
 from typing import Any, Iterable, List, Optional
 
 from pymongo import MongoClient
@@ -176,7 +176,7 @@ def _execute(self, command: str) -> dict[str, Any]:
 
         # checks if command is a find query
         if not result and re.match(r"^db\.\w+\.find\w*\(\{.*\}\)", command):
-            cursor = eval(command) # dangerous, might need to find a better solution
+            cursor = eval(command)  # dangerous, might need to find a better solution
             result_list = []
             for doc in cursor:
                 result_list.append(doc)

libs/langchain/langchain/utilities/mongo_database.py

@@ -5,7 +5,9 @@
 
 
 def test_create_mongo_agent() -> None:
-    db = MongoDatabase.from_uri("mongodb://%2Ftmp%2Fmongodb-27017.sock/test_db")
+    db = MongoDatabase.from_uri(
+        "mongodb://%2Ftmp%2Fmongodb-27017.sock/test_db?inMemory=true"
+    )
     queries = {"foo": "Final Answer: baz"}
     llm = FakeLLM(queries=queries, sequential_responses=True)
     toolkit = MongoDatabaseToolkit(db=db, llm=llm)

libs/langchain/poetry.lock

@@ -6,7 +6,7 @@
 
 from langchain.utilities.mongo_database import MongoDatabase
 
-uri = "mongodb://%2Ftmp%2Fmongodb-27017.sock/test_db"
+uri = "mongodb://%2Ftmp%2Fmongodb-27017.sock/test_db?inMemory=true"
 
 
 def test_collection_info() -> None: