removing docker dependency from pyproject.toml

deepsense-ai · Sep 29, 2023 · 1f8f9f4 · 1f8f9f4
1 parent 07d5b8c
commit 1f8f9f4
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 9 deletions.
diff --git a/libs/langchain/langchain/tools/python/tool.py b/libs/langchain/langchain/tools/python/tool.py
@@ -6,7 +6,7 @@
 import sys
 from contextlib import redirect_stdout
 from io import StringIO
-from typing import TYPE_CHECKING, Any, Dict, Optional, Type, Union
+from typing import Any, Dict, Optional, Type, Union
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForToolRun,
@@ -15,9 +15,7 @@
 from langchain.pydantic_v1 import BaseModel, Field, root_validator
 from langchain.tools.base import BaseTool
 from langchain.utilities import PythonREPL
-
-if TYPE_CHECKING:
-    from langchain.utilities import PythonContainerREPL
+from langchain.utilities.python_docker_repl import PythonContainerREPL
 
 
 def _get_default_python_repl() -> PythonREPL:

diff --git a/libs/langchain/langchain/utilities/__init__.py b/libs/langchain/langchain/utilities/__init__.py
@@ -25,7 +25,6 @@
 from langchain.utilities.powerbi import PowerBIDataset
 from langchain.utilities.pubmed import PubMedAPIWrapper
 from langchain.utilities.python import PythonREPL
-from langchain.utilities.python_docker_repl import PythonContainerREPL
 from langchain.utilities.requests import Requests, RequestsWrapper, TextRequestsWrapper
 from langchain.utilities.scenexplain import SceneXplainAPIWrapper
 from langchain.utilities.searx_search import SearxSearchWrapper
@@ -61,7 +60,6 @@
     "PowerBIDataset",
     "PubMedAPIWrapper",
     "PythonREPL",
-    "PythonContainerREPL",
     "Requests",
     "RequestsWrapper",
     "SQLDatabase",

diff --git a/libs/langchain/pyproject.toml b/libs/langchain/pyproject.toml
@@ -132,7 +132,6 @@ sqlite-vss = {version = "^0.1.2", optional = true}
 anyio = "<4.0"
 jsonpatch = "^1.33"
 timescale-vector = {version = "^0.0.1", optional = true}
-docker = {version = "^6.1.3", optional = true}
 
 [tool.poetry.group.test.dependencies]
 # The only dependencies that should be added are
@@ -213,7 +212,6 @@ cohere = ["cohere"]
 docarray = ["docarray"]
 embeddings = ["sentence-transformers"]
 javascript = ["esprima"]
-docker = ["docker"]
 azure = [
     "azure-identity",
     "azure-cosmos",
@@ -298,7 +296,6 @@ all = [
     "amadeus",
     "librosa",
     "python-arango",
-    "docker",
 ]
 
 # An extra used to be able to add extended testing.