Bolt 6.0 support with vector types (#1214)

Introduce Bolt 6.0 support, which includes vector types.

Example:
```python
from neo4j.vector import Vector

# Create a vector (raw big-endian byte buffer by default)
v = Vector(b"\x00\x01\x02\x03", "i16")
# or (less performant, but much more user-friendly)
v = Vector.from_native([1, 513], "i16")
# `v` can now be used as query parameter

# Get a vector
v = driver.execute_query(
    "MATCH (d:Doc) RETURN d.embedding AS v LIMIT 1",
    database_="neo4j",
).records[0]["v"]
# Working with the vector
print(f"Got vector of type {v.dtype} with {len(v)} elements: {v}")
print("raw bytes:", v.raw())
print("python list:", v.to_native())
print("as numpy array:", v.to_numpy())
```

Author: robsdedude

docs/source/conf.py

@@ -22,7 +22,7 @@
 # If extensions (or modules to document with autodoc) are in another directory,
 # add these directories to sys.path here. If the directory is relative to the
 # documentation root, use os.path.abspath to make it absolute, like shown here.
-sys.path.insert(0, os.path.abspath(os.path.join("..", "..")))
+sys.path.insert(0, os.path.abspath(os.path.join("..", "..", "src")))
 
 
 from neo4j import __version__ as project_version
@@ -345,6 +345,9 @@ def setup(app):
 intersphinx_mapping = {
     "python": ("https://docs.python.org/3", None),
     "dateutil": ("https://dateutil.readthedocs.io/en/stable/", None),
+    "numpy": ("https://numpy.org/doc/stable/", None),
+    "pandas": ("https://pandas.pydata.org/docs/", None),
+    "pyarrow": ("https://arrow.apache.org/docs/", None),
 }
 
 autodoc_default_options = {

docs/source/index.rst

@@ -9,6 +9,7 @@ Bolt protocol versions supported:
 
 .. # [bolt-version-bump] search tag when changing bolt version support
 
+* Bolt 6.0
 * Bolt 5.0 - 5.8
 * Bolt 4.4
 
@@ -36,6 +37,8 @@ Topics
 
 + :ref:`temporal-data-types`
 
++ :ref:`vector-data-types`
+
 + :ref:`breaking-changes`
 
 
@@ -47,6 +50,7 @@ Topics
    async_api.rst
    types/spatial.rst
    types/temporal.rst
+   types/vector.rst
    breaking_changes.rst
 
 

docs/source/types/vector.rst

@@ -0,0 +1,18 @@
+.. _vector-data-types:
+
+*****************
+Vector Data Types
+*****************
+
+.. autoclass:: neo4j.vector.Vector
+    :members:
+
+
+.. autoclass:: neo4j.vector.VectorEndian
+    :show-inheritance:
+    :members:
+
+
+.. autoclass:: neo4j.vector.VectorDType
+    :show-inheritance:
+    :members:

pyproject.toml

@@ -54,12 +54,12 @@ Forum = "https://community.neo4j.com/c/drivers-stacks/python/"
 Discord = "https://discord.com/invite/neo4j"
 
 [project.optional-dependencies]
-numpy = ["numpy >= 1.7.0, < 3.0.0"]
+numpy = ["numpy >= 1.21.2, < 3.0.0"]
 pandas = [
     "pandas >= 1.1.0, < 3.0.0",
-    "numpy >= 1.7.0, < 3.0.0",
+    "numpy >= 1.21.2, < 3.0.0",
 ]
-pyarrow = ["pyarrow >= 1.0.0"]
+pyarrow = ["pyarrow >= 6.0.0, < 21.0.0"]
 
 
 [build-system]
@@ -207,8 +207,9 @@ asyncio_default_fixture_loop_scope="function"
 [[tool.mypy.overrides]]
 module = [
     "pandas.*",
-    "neo4j._codec.packstream._rust",
-    "neo4j._codec.packstream._rust.*",
+    "pyarrow.*",
+    "neo4j._rust",
+    "neo4j._rust.*",
 ]
 ignore_missing_imports = true
 

src/neo4j/_async/io/__init__.py

@@ -37,6 +37,7 @@
     _bolt3,
     _bolt4,
     _bolt5,
+    _bolt6,
 )
 from ._bolt import AsyncBolt
 from ._common import ConnectionErrorHandler