fix: deeplake pin

ComposioHQ · Feb 17, 2025 · dd1e618 · dd1e618
1 parent c4d962f
commit dd1e618
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/python/composio/tools/local/codeanalysis/embedder.py b/python/composio/tools/local/codeanalysis/embedder.py
@@ -148,6 +148,10 @@ def get_topn_chunks_from_query(
         ValueError: If the input parameters are invalid.
         RuntimeError: If there's an error during the search process.
     """
+    from deeplake.core.vectorstore.deeplake_vectorstore import (  # pylint: disable=import-outside-toplevel
+        DeepLakeVectorStore,
+    )
+
     if not isinstance(vector_store, DeepLakeVectorStore):
         raise ValueError("vector_store must be an instance of DeepLakeVectorStore")
     try:

diff --git a/python/composio/tools/local/codeanalysis/tool.py b/python/composio/tools/local/codeanalysis/tool.py
@@ -16,9 +16,10 @@ class CodeAnalysisTool(LocalTool, autoload=True):
 
     requires = [
         "tree_sitter>=0.22.0",
-        "deeplake>3.9,<4",
+        "deeplake>3.9,<3.9.39",
         "sentence-transformers",
         "tree_sitter_python>=0.22.0",
+        "tokenizers>=0.19,<0.20",
         "git+https://github.com/DataDog/jedi.git@92d0c807b0dcd115b1ffd0a4ed21e44db127c2fb#egg=jedi",
         "PyJWT",  # deeplake/client/client.py:41
     ]

diff --git a/python/tox.ini b/python/tox.ini
@@ -96,7 +96,7 @@ deps =
     ; composio_langgraph==0.5.13
     langgraph==0.2.16
     langchain-aws==0.1.17
-    deeplake>3.9,<4 # codeanalysis
+    deeplake>3.9,<3.9.39 # codeanalysis
     git+https://github.com/DataDog/jedi.git@92d0c807b0dcd115b1ffd0a4ed21e44db127c2fb#egg=jedi # codeanalysis
     libcst # codeanalysis
     sentence_transformers # codeanalysis