set astrapy version to ^1 (#21)

* Set minimum version of astrapy to v1.0.0
langchain-ai · Apr 17, 2024 · 6a24829 · 6a24829
1 parent 18caf27
commit 6a24829
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 6 deletions.
diff --git a/libs/astradb/poetry.lock b/libs/astradb/poetry.lock
diff --git a/libs/astradb/pyproject.toml b/libs/astradb/pyproject.toml
@@ -13,7 +13,7 @@ license = "MIT"
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
 langchain-core = "^0.1.31"
-astrapy = ">=0.7.7, <2"
+astrapy = "^1"
 numpy = "^1"
 
 [tool.poetry.group.test]

diff --git a/libs/astradb/tests/integration_tests/test_storage.py b/libs/astradb/tests/integration_tests/test_storage.py
@@ -21,8 +21,6 @@ def _has_env_vars() -> bool:
 
 @pytest.fixture
 def astra_db() -> AstraDB:
-    from astrapy.db import AstraDB
-
     return AstraDB(
         token=os.environ["ASTRA_DB_APPLICATION_TOKEN"],
         api_endpoint=os.environ["ASTRA_DB_API_ENDPOINT"],
@@ -32,8 +30,6 @@ def astra_db() -> AstraDB:
 
 @pytest.fixture
 def async_astra_db() -> AsyncAstraDB:
-    from astrapy.db import AsyncAstraDB
-
     return AsyncAstraDB(
         token=os.environ["ASTRA_DB_APPLICATION_TOKEN"],
         api_endpoint=os.environ["ASTRA_DB_API_ENDPOINT"],