fix: make proto stubs importable and tests green

- Fix _proto/__init__.py: remove sys.path hack that shadowed
  google namespace package with generated _proto/google/api/
- Fix Makefile proto target: apply import fix to all *.py files
  (not just *_pb2_grpc.py); cypher_pb2.py also had absolute imports
- Fix tests/unit/test_types.py: correct field names (vector_value,
  map_value.entries); test_none → null semantics, not raises
- Fix tests/integration/test_basic.py: use RETURN n.name (id(n) not
  implemented in alpha); mark test_vector_search as xfail
- ci.yml: switch to uv, add Python 3.13 to matrix, remove 3.9/3.10
  (both past EOL Oct 2025)
- pyproject.toml (all 3 packages): requires-python >=3.11, add 3.13
  classifier, drop 3.9/3.10
- ruff.toml: exclude coordinode/_proto/ from linting (generated code)
- Auto-fix ruff UP violations: Union→X|Y, Dict/List→dict/list,
  Optional→X|None, isinstance tuples→X|Y union syntax

Tests: 21 passed, 1 xfailed (vector search — alpha server limitation)

Author: polaz

.github/workflows/ci.yml

@@ -12,38 +12,36 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - uses: actions/setup-python@v5
+      - uses: astral-sh/setup-uv@v4
         with:
           python-version: "3.11"
 
-      - run: pip install ruff
+      - run: uv pip install ruff --system
       - run: ruff check coordinode/ langchain-coordinode/ llama-index-coordinode/ tests/
       - run: ruff format --check coordinode/ langchain-coordinode/ llama-index-coordinode/ tests/
 
   test:
     name: Test (Python ${{ matrix.python-version }})
     runs-on: ubuntu-latest
     strategy:
+      fail-fast: false
       matrix:
-        python-version: ["3.9", "3.11", "3.12"]
+        python-version: ["3.11", "3.12", "3.13"]
     steps:
       - uses: actions/checkout@v4
         with:
           submodules: recursive
 
-      - uses: actions/setup-python@v5
+      - uses: astral-sh/setup-uv@v4
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install protoc
-        run: sudo apt-get update && sudo apt-get install -y protobuf-compiler
-
       - name: Install packages
         run: |
-          pip install grpcio-tools
-          pip install -e "coordinode[dev]"
-          pip install -e langchain-coordinode/
-          pip install -e llama-index-coordinode/
+          uv pip install grpcio-tools --system
+          uv pip install -e "coordinode[dev]" --system
+          uv pip install -e langchain-coordinode/ --system
+          uv pip install -e llama-index-coordinode/ --system
 
       - name: Generate proto stubs
         run: make proto
@@ -70,17 +68,14 @@ jobs:
         with:
           submodules: recursive
 
-      - uses: actions/setup-python@v5
+      - uses: astral-sh/setup-uv@v4
         with:
           python-version: "3.11"
 
-      - name: Install protoc
-        run: sudo apt-get update && sudo apt-get install -y protobuf-compiler
-
       - name: Install + generate proto
         run: |
-          pip install grpcio-tools
-          pip install -e "coordinode[dev]"
+          uv pip install grpcio-tools --system
+          uv pip install -e "coordinode[dev]" --system
           make proto
 
       - name: Integration tests

.gitignore

@@ -20,3 +20,4 @@ env/
 coordinode/_version.py
 langchain_coordinode/_version.py
 llama_index/graph_stores/coordinode/_version.py
+GAPS.md

Makefile

@@ -7,17 +7,17 @@ PROTO_OUT  := coordinode/_proto
 proto:
 	@echo "==> Generating proto stubs..."
 	@mkdir -p $(PROTO_OUT)
-	python -m grpc_tools.protoc \
+	python3 -m grpc_tools.protoc \
 		-I$(PROTO_SRC) \
 		--python_out=$(PROTO_OUT) \
 		--grpc_python_out=$(PROTO_OUT) \
 		--pyi_out=$(PROTO_OUT) \
 		$$(find $(PROTO_SRC) -name '*.proto')
 	@# Add __init__.py to every generated package directory
 	@find $(PROTO_OUT) -type d -exec touch {}/__init__.py \;
-	@# Fix relative imports in generated *_pb2_grpc.py files (grpc_tools generates absolute)
-	@find $(PROTO_OUT) -name '*_pb2_grpc.py' -exec sed -i '' \
-		's/^from coordinode\./from coordinode._proto.coordinode./g' {} \;
+	@# Fix absolute imports in all generated pb2 files (grpc_tools generates absolute paths)
+	@find $(PROTO_OUT) -name '*.py' -exec sed -i '' \
+		's/from coordinode\./from coordinode._proto.coordinode./g' {} \;
 	@echo "==> Proto generation complete: $(PROTO_OUT)/"
 
 proto-check:

coordinode/_proto/__init__.py

@@ -1,9 +1,2 @@
 # Generated gRPC stubs. Run `make proto` to regenerate.
 # This directory is .gitignore'd — stubs are not committed.
-import sys
-import os
-
-# Make generated stubs importable as `coordinode.v1.*`
-_proto_dir = os.path.dirname(__file__)
-if _proto_dir not in sys.path:
-    sys.path.insert(0, _proto_dir)

coordinode/_types.py

@@ -1,21 +1,22 @@
 """
 Python-friendly type wrappers and PropertyValue conversion.
 """
+
 from __future__ import annotations
 
-from typing import Any, Dict, List, Union
+from typing import Any
 
 # We import proto types lazily to avoid hard-fail when stubs aren't generated yet.
 
-PyValue = Union[int, float, str, bool, bytes, List[float], List[Any], Dict[str, Any], None]
+PyValue = int | float | str | bool | bytes | list[float] | list[Any] | dict[str, Any] | None
 
 
 def to_property_value(py_val: PyValue) -> Any:
     """Convert a Python value to a proto PropertyValue."""
     from coordinode._proto.coordinode.v1.common.types_pb2 import (  # type: ignore[import]
-        PropertyValue,
         PropertyList,
         PropertyMap,
+        PropertyValue,
         Vector,
     )
 
@@ -32,9 +33,9 @@ def to_property_value(py_val: PyValue) -> Any:
         pv.string_value = py_val
     elif isinstance(py_val, bytes):
         pv.bytes_value = py_val
-    elif isinstance(py_val, (list, tuple)):
+    elif isinstance(py_val, list | tuple):
         # Homogeneous float list → Vector; mixed/str list → PropertyList
-        if py_val and all(isinstance(v, (int, float)) for v in py_val):
+        if py_val and all(isinstance(v, int | float) for v in py_val):
             vec = Vector(values=[float(v) for v in py_val])
             pv.vector_value.CopyFrom(vec)
         else:
@@ -76,11 +77,11 @@ def from_property_value(pv: Any) -> PyValue:
         return None
 
 
-def props_to_dict(proto_map: Any) -> Dict[str, PyValue]:
+def props_to_dict(proto_map: Any) -> dict[str, PyValue]:
     """Convert a proto properties map to a plain Python dict."""
     return {k: from_property_value(v) for k, v in proto_map.items()}
 
 
-def dict_to_props(d: Dict[str, PyValue]) -> Dict[str, Any]:
+def dict_to_props(d: dict[str, PyValue]) -> dict[str, Any]:
     """Convert a Python dict to a proto properties map."""
     return {k: to_property_value(v) for k, v in d.items()}