python-arango-async 1.2.0__tar.gz → 1.2.2__tar.gz

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-arango-async
-Version: 1.2.0
+Version: 1.2.2
 Summary: Async Python Driver for ArangoDB
 Author-email: Alexandru Petenchea <alexandru.petenchea@arangodb.com>, Anthony Mahanna <anthony.mahanna@arangodb.com>
 Maintainer-email: Alexandru Petenchea <alexandru.petenchea@arangodb.com>, Anthony Mahanna <anthony.mahanna@arangodb.com>

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/arangoasync/aql.py

@@ -369,6 +369,7 @@ class AQL:
         enabled: Optional[bool] = None,
         max_slow_queries: Optional[int] = None,
         slow_query_threshold: Optional[int] = None,
+        slow_streaming_query_threshold: Optional[int] = None,
         max_query_string_length: Optional[int] = None,
         track_bind_vars: Optional[bool] = None,
         track_slow_queries: Optional[int] = None,
@@ -382,6 +383,8 @@ class AQL:
                 entries are discarded first.
             slow_query_threshold (int | None): Runtime threshold (in seconds) for treating a
                 query as slow.
+            slow_streaming_query_threshold (int | None): Runtime threshold (in seconds) for
+                treating a streaming query as slow.
             max_query_string_length (int | None): The maximum query string length (in bytes)
                 to keep in the list of queries.
             track_bind_vars (bool | None): If set to `True`, track bind variables used in
@@ -409,6 +412,8 @@ class AQL:
             data["maxQueryStringLength"] = max_query_string_length
         if slow_query_threshold is not None:
             data["slowQueryThreshold"] = slow_query_threshold
+        if slow_streaming_query_threshold is not None:
+            data["slowStreamingQueryThreshold"] = slow_streaming_query_threshold
         if track_bind_vars is not None:
             data["trackBindVars"] = track_bind_vars
         if track_slow_queries is not None:

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/arangoasync/collection.py

@@ -397,7 +397,7 @@ class Collection(Generic[T, U, V]):
 
         Args:
             type (str): Type attribute (ex. "persistent", "inverted", "ttl", "mdi",
-                "geo").
+                "geo", "vector").
             fields (dict | list): Fields to index.
             options (dict | None): Additional index options.
 

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/arangoasync/typings.py

@@ -727,6 +727,10 @@ class CollectionProperties(JsonWrapper):
     def computed_values(self) -> Optional[Json]:
         return self._data.get("computedValues")
 
+    @property
+    def supportsRBAC(self) -> Optional[bool]:
+        return self._data.get("supportsRBAC")
+
     @property
     def object_id(self) -> str:
         return self._data["objectId"]  # type: ignore[no-any-return]
@@ -808,6 +812,8 @@ class CollectionProperties(JsonWrapper):
             result["computedValues"] = data["computedValues"]
         if "internalValidatorType" in data:
             result["internal_validator_type"] = data["internalValidatorType"]
+        if "supportsRBAC" in data:
+            result["supportsRBAC"] = data["supportsRBAC"]
         return result
 
     def format(self, formatter: Optional[Formatter] = None) -> Json:
@@ -1121,6 +1127,14 @@ class IndexProperties(JsonWrapper):
     def features(self) -> Optional[List[str]]:
         return self._data.get("features")
 
+    @property
+    def error_message(self) -> Optional[str]:
+        return self._data.get("errorMessage")
+
+    @property
+    def training_state(self) -> Optional[str]:
+        return self._data.get("trainingState")
+
     @staticmethod
     def compatibility_formatter(data: Json) -> Json:
         """python-arango compatibility formatter."""
@@ -1179,6 +1193,10 @@ class IndexProperties(JsonWrapper):
             result["writebuffer_max_size"] = data["writebufferSizeMax"]
         if "optimizeTopK" in data:
             result["optimizeTopK"] = data["optimizeTopK"]
+        if "errorMessage" in data:
+            result["error_message"] = data["errorMessage"]
+        if "trainingState" in data:
+            result["training_state"] = data["trainingState"]
         return result
 
     def format(self, formatter: Optional[Formatter] = None) -> Json:

python_arango_async-1.2.2/arangoasync/version.py

@@ -0,0 +1 @@
+__version__ = "1.2.2"

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/python_arango_async.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-arango-async
-Version: 1.2.0
+Version: 1.2.2
 Summary: Async Python Driver for ArangoDB
 Author-email: Alexandru Petenchea <alexandru.petenchea@arangodb.com>, Anthony Mahanna <anthony.mahanna@arangodb.com>
 Maintainer-email: Alexandru Petenchea <alexandru.petenchea@arangodb.com>, Anthony Mahanna <anthony.mahanna@arangodb.com>

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/tests/static/cluster-3.12.conf

@@ -9,7 +9,12 @@ jwt-secret = /tests/static/keyfile
 
 [args]
 all.database.password = passwd
+all.vector-index = true
 all.database.extended-names = true
 all.log.api-enabled = true
 all.javascript.allow-admin-execute = true
 all.server.options-api = admin
+all.javascript.files-allowlist = ".*";
+all.javascript.environment-variables-allowlist = ".*";
+all.javascript.endpoints-allowlist = ".*";
+all.javascript.startup-options-allowlist = ".*";

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/tests/static/single-3.12.conf

@@ -8,6 +8,12 @@ jwt-secret = /tests/static/keyfile
 
 [args]
 all.database.password = passwd
+all.vector-index = true
 all.database.extended-names = true
+all.log.api-enabled = true
 all.javascript.allow-admin-execute = true
 all.server.options-api = admin
+all.javascript.files-allowlist = ".*";
+all.javascript.environment-variables-allowlist = ".*";
+all.javascript.endpoints-allowlist = ".*";
+all.javascript.startup-options-allowlist = ".*";

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/tests/test_aql.py

@@ -1,5 +1,6 @@
 import asyncio
 import time
+import warnings
 
 import pytest
 from packaging import version
@@ -64,9 +65,16 @@ async def test_query_tracking(db, bad_db):
     assert tracking.enabled is False
 
     # Re-enable.
-    tracking = await aql.set_tracking(enabled=True, max_slow_queries=5)
+    tracking = await aql.set_tracking(
+        enabled=True,
+        max_slow_queries=5,
+        slow_streaming_query_threshold=20,
+        slow_query_threshold=15,
+    )
     assert tracking.enabled is True
     assert tracking.max_slow_queries == 5
+    assert tracking.slow_streaming_query_threshold == 20
+    assert tracking.slow_query_threshold == 15
 
     # Exceptions with bad database
     with pytest.raises(AQLQueryTrackingGetError):
@@ -131,10 +139,15 @@ async def test_kill_query(db, bad_db, superuser):
         queries = await aql.queries()
         if len(queries) > 0:
             break
-
-    # Kill the query
-    query_id = queries[0]["id"]
-    assert await aql.kill(query_id) is True
+        time.sleep(1)
+
+    if len(queries) > 0:
+        # Kill the query
+        query_id = queries[0]["id"]
+        assert await aql.kill(query_id) is True
+    else:
+        # This test is prone to races
+        warnings.warn("Query not found in the list, skipping!")
 
     # Ignore missing
     assert await aql.kill("fakeid", ignore_missing=True) is False

{python_arango_async-1.2.0 → python_arango_async-1.2.2}/tests/test_collection.py

@@ -228,15 +228,36 @@ async def test_collection_index(doc_col, bad_col, cluster):
         await bad_col.load_indexes()
     assert err.value.error_code == DATA_SOURCE_NOT_FOUND
 
+    # Create a vector index
+    docs = []
+    for key in range(100):
+        docs.append({"_key": f"key_{key}", "embedding": [1] * 128})
+    await doc_col.insert_many(docs)
+    idx4 = await doc_col.add_index(
+        "vector",
+        ["embedding"],
+        {
+            "name": "vector_index",
+            "params": {
+                "metric": "cosine",
+                "dimension": 128,
+                "nLists": 2,
+            },
+        },
+    )
+    assert idx4.name == "vector_index"
+
     # Delete indexes
-    del1, del2, del3 = await asyncio.gather(
+    del1, del2, del3, del4 = await asyncio.gather(
         doc_col.delete_index(idx1.id),
         doc_col.delete_index(idx2.numeric_id),
         doc_col.delete_index(str(idx3.numeric_id)),
+        doc_col.delete_index(idx4.id),
     )
     assert del1 is True
     assert del2 is True
     assert del3 is True
+    assert del4 is True
 
     # Now, the indexes should be gone
     with pytest.raises(IndexDeleteError) as err:

python_arango_async-1.2.0/arangoasync/version.py

@@ -1 +0,0 @@
-__version__ = "1.2.0"