Bump astrapy to 2.x (#91)

* tentative => astrapy2 port

* astrapy 2 rc2

* actual astrapy2 dependency (lockfile todo)

* regenerate poetry.lock

* bump to astrapy 2.0.1

Author: hemidactylus

client/astra_assistants/tools/astra_data_api.py

@@ -12,17 +12,12 @@
 
 class AstraDataAPITool(ToolInterface):
     def __init__(self, db_url: str, collection_name, vectorize: bool = False, openai_client: OpenAI = None, embedding_model:str = None, namespace: str = None):
-        client = DataAPIClient(token=os.environ["ASTRA_DB_APPLICATION_TOKEN"])
-        db = None
-        if namespace:
-            db = client.get_database_by_api_endpoint(
-                db_url,
-                namespace=namespace,
-            )
-        else:
-            db = client.get_database_by_api_endpoint(
-                db_url,
-            )
+        client = DataAPIClient()
+        db = client.get_database(
+            db_url,
+            token=os.environ["ASTRA_DB_APPLICATION_TOKEN"],
+            keyspace=namespace,  # passing None is on (same as omitting the parameter)
+        )
         self.collection = db.get_collection(collection_name)
         self.vectorize = vectorize
         self.openai_client = openai_client

client/poetry.lock

@@ -6,7 +6,7 @@ authors = ["phact <estevezsebastian@gmail.com>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = ">=3.10.0"
+python = ">=3.10.0,<4.0"
 httpx = "^0.27.0"
 openai = "^1.21.0"
 litellm = "^1.36.0"
@@ -18,7 +18,7 @@ lsprotocol = "^2023.0.1"
 python-lsp-jsonrpc = "^1.1.2"
 ruff = ">=0.6.2,<9.0.0"
 tree-sitter = "^0.23.0"
-astrapy = { version = "^1.2.1", optional = true }
+astrapy = { version = ">=2.0.1,<3.0.0", optional = true }
 e2b-code-interpreter = { version = "^1.0.1", optional = true }
 e2b = { version = "^1.0.1", optional = true }
 mcp = ">=0.9.1"