Commit 85a4955

authored

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()) ```

1 parent 23f8242 commit 85a4955Copy full SHA for 85a4955

File tree

64 files changed

+5946

-48

lines changed

docs/source
- conf.py
- index.rst
- types
  - vector.rst
pyproject.toml
src/neo4j
- _async
  - io
    - __init__.py
    - _bolt6.py
  - work
    - result.py
- _codec
  - hydration
  - packstream
    - _common.py
    - v1
      - __init__.py
- _optional_deps
  - __init__.py
- _sync
  - io
    - __init__.py
    - _bolt6.py
  - work
    - result.py
- _typing.py
- time
  - __init__.py
- vector.py
testkitbackend
tests/unit

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

64 files changed

+5946

-48

lines changed

`‎docs/source/conf.py`

Lines changed: 4 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`	`# If extensions (or modules to document with autodoc) are in another directory,`
`23`	`23`	`# add these directories to sys.path here. If the directory is relative to the`
`24`	`24`	`# documentation root, use os.path.abspath to make it absolute, like shown here.`
`25`		`-sys.path.insert(0, os.path.abspath(os.path.join("..", "..")))`
	`25`	`+sys.path.insert(0, os.path.abspath(os.path.join("..", "..", "src")))`
`26`	`26`
`27`	`27`
`28`	`28`	`from neo4j import __version__ as project_version`
`@@ -345,6 +345,9 @@ def setup(app):`
`345`	`345`	`intersphinx_mapping = {`
`346`	`346`	`"python": ("https://docs.python.org/3", None),`
`347`	`347`	`"dateutil": ("https://dateutil.readthedocs.io/en/stable/", None),`
	`348`	`+ "numpy": ("https://numpy.org/doc/stable/", None),`
	`349`	`+ "pandas": ("https://pandas.pydata.org/docs/", None),`
	`350`	`+ "pyarrow": ("https://arrow.apache.org/docs/", None),`
`348`	`351`	`}`
`349`	`352`
`350`	`353`	`autodoc_default_options = {`

`‎docs/source/index.rst`

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,7 @@ Bolt protocol versions supported:`
`9`	`9`
`10`	`10`	`.. # [bolt-version-bump] search tag when changing bolt version support`
`11`	`11`
	`12`	`+* Bolt 6.0`
`12`	`13`	`* Bolt 5.0 - 5.8`
`13`	`14`	`* Bolt 4.4`
`14`	`15`
`@@ -36,6 +37,8 @@ Topics`
`36`	`37`
`37`	`38`	+ :ref:`temporal-data-types`
`38`	`39`
	`40`	++ :ref:`vector-data-types`
	`41`	`+`
`39`	`42`	+ :ref:`breaking-changes`
`40`	`43`
`41`	`44`
`@@ -47,6 +50,7 @@ Topics`
`47`	`50`	`async_api.rst`
`48`	`51`	`types/spatial.rst`
`49`	`52`	`types/temporal.rst`
	`53`	`+ types/vector.rst`
`50`	`54`	`breaking_changes.rst`
`51`	`55`
`52`	`56`

`‎docs/source/types/vector.rst`

Lines changed: 18 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,18 @@`
	`1`	`+.. _vector-data-types:`
	`2`	`+`
	`3`	`+*****************`
	`4`	`+Vector Data Types`
	`5`	`+*****************`
	`6`	`+`
	`7`	`+.. autoclass:: neo4j.vector.Vector`
	`8`	`+ :members:`
	`9`	`+`
	`10`	`+`
	`11`	`+.. autoclass:: neo4j.vector.VectorEndian`
	`12`	`+ :show-inheritance:`
	`13`	`+ :members:`
	`14`	`+`
	`15`	`+`
	`16`	`+.. autoclass:: neo4j.vector.VectorDType`
	`17`	`+ :show-inheritance:`
	`18`	`+ :members:`

`‎pyproject.toml`

Lines changed: 6 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -54,12 +54,12 @@ Forum = "https://community.neo4j.com/c/drivers-stacks/python/"`
`54`	`54`	`Discord = "https://discord.com/invite/neo4j"`
`55`	`55`
`56`	`56`	`[project.optional-dependencies]`
`57`		`-numpy = ["numpy >= 1.7.0, < 3.0.0"]`
	`57`	`+numpy = ["numpy >= 1.21.2, < 3.0.0"]`
`58`	`58`	`pandas = [`
`59`	`59`	`"pandas >= 1.1.0, < 3.0.0",`
`60`		`- "numpy >= 1.7.0, < 3.0.0",`
	`60`	`+ "numpy >= 1.21.2, < 3.0.0",`
`61`	`61`	`]`
`62`		`-pyarrow = ["pyarrow >= 1.0.0"]`
	`62`	`+pyarrow = ["pyarrow >= 6.0.0, < 21.0.0"]`
`63`	`63`
`64`	`64`
`65`	`65`	`[build-system]`
`@@ -207,8 +207,9 @@ asyncio_default_fixture_loop_scope="function"`
`207`	`207`	`[[tool.mypy.overrides]]`
`208`	`208`	`module = [`
`209`	`209`	`"pandas.*",`
`210`		`- "neo4j._codec.packstream._rust",`
`211`		`- "neo4j._codec.packstream._rust.*",`
	`210`	`+ "pyarrow.*",`
	`211`	`+ "neo4j._rust",`
	`212`	`+ "neo4j._rust.*",`
`212`	`213`	`]`
`213`	`214`	`ignore_missing_imports = true`
`214`	`215`

`‎src/neo4j/_async/io/init.py`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -37,6 +37,7 @@`
`37`	`37`	`_bolt3,`
`38`	`38`	`_bolt4,`
`39`	`39`	`_bolt5,`
	`40`	`+ _bolt6,`
`40`	`41`	`)`
`41`	`42`	`from ._bolt import AsyncBolt`
`42`	`43`	`from ._common import ConnectionErrorHandler`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 85a4955

File tree

64 files changed

Some content is hidden

64 files changed

`‎docs/source/conf.py`

`‎docs/source/index.rst`

`‎docs/source/types/vector.rst`

`‎pyproject.toml`

`‎src/neo4j/_async/io/init.py`

0 commit comments

File tree

64 files changed

Some content is hidden

64 files changed

‎docs/source/conf.py

‎docs/source/index.rst

‎docs/source/types/vector.rst

‎pyproject.toml

‎src/neo4j/_async/io/__init__.py

0 commit comments

`‎docs/source/conf.py`

`‎docs/source/index.rst`

`‎docs/source/types/vector.rst`

`‎pyproject.toml`

`‎src/neo4j/_async/io/init.py`