test(adapters): add integration tests for langchain and llama-index adapters

- Add tests/integration/adapters/test_langchain.py: CoordinodeGraph
  connect, schema, query, add_graph_documents (upsert nodes, create
  relationships, idempotency), schema invalidation
- Add tests/integration/adapters/test_llama_index.py: CoordinodePropertyGraphStore
  connect, schema, structured_query, upsert_nodes, upsert_relations,
  get_triplets, get_rel_map, delete (by id and name)
- Fix namespace package declarations in llama_index/__init__.py and
  llama_index/graph_stores/__init__.py: use pkgutil.extend_path so
  llama_index.core remains importable when package is installed editable
- Add llama-index-core and langchain-community to dev dependencies so
  adapter tests can be collected and run from the workspace root
- 19/19 integration tests pass against localhost:17080

All 19 adapter tests pass. Relates to #14.

Author: polaz

llama-index-coordinode/llama_index/__init__.py

@@ -1 +1,3 @@
-# namespace package
+from pkgutil import extend_path
+
+__path__ = extend_path(__path__, __name__)

llama-index-coordinode/llama_index/graph_stores/__init__.py

@@ -1 +1,3 @@
-# namespace package
+from pkgutil import extend_path
+
+__path__ = extend_path(__path__, __name__)

pyproject.toml

@@ -28,6 +28,9 @@ coordinode = { workspace = true }
 dev = [
     "build>=1.2",
     "grpcio-tools>=1.60",
+    "langchain-community>=0.3",
+    "langchain-core>=0.3",
+    "llama-index-core>=0.12",
     "pytest>=8",
     "pytest-asyncio>=0.23",
     "pytest-timeout>=2",

tests/integration/adapters/__init__.py

@@ -0,0 +1,130 @@
+"""Integration tests for CoordinodeGraph (LangChain adapter).
+
+Requires a running CoordiNode instance. Set COORDINODE_ADDR env var
+(default: localhost:7080).
+
+Run via:
+    COORDINODE_ADDR=localhost:17080 pytest tests/integration/adapters/test_langchain.py -v
+"""
+
+import os
+import uuid
+
+import pytest
+from langchain_community.graphs.graph_document import GraphDocument, Node, Relationship
+from langchain_core.documents import Document
+
+from langchain_coordinode import CoordinodeGraph
+
+ADDR = os.environ.get("COORDINODE_ADDR", "localhost:7080")
+
+
+@pytest.fixture(scope="module")
+def graph():
+    with CoordinodeGraph(ADDR) as g:
+        yield g
+
+
+# ── Basic connectivity ────────────────────────────────────────────────────────
+
+def test_connect(graph):
+    assert graph is not None
+
+
+def test_schema_returns_string(graph):
+    schema = graph.schema
+    assert isinstance(schema, str)
+
+
+def test_refresh_schema_does_not_raise(graph):
+    graph.refresh_schema()
+    assert isinstance(graph.schema, str)
+    assert isinstance(graph.structured_schema, dict)
+    assert "node_props" in graph.structured_schema
+    assert "rel_props" in graph.structured_schema
+    assert "relationships" in graph.structured_schema
+
+
+# ── Cypher query ──────────────────────────────────────────────────────────────
+
+def test_query_returns_list(graph):
+    result = graph.query("RETURN 1 AS n")
+    assert isinstance(result, list)
+    assert result[0]["n"] == 1
+
+
+def test_query_count(graph):
+    result = graph.query("MATCH (n) RETURN count(n) AS total")
+    assert isinstance(result, list)
+    assert isinstance(result[0]["total"], int)
+
+
+# ── add_graph_documents ───────────────────────────────────────────────────────
+
+@pytest.fixture
+def unique_tag():
+    return uuid.uuid4().hex[:8]
+
+
+def test_add_graph_documents_upserts_nodes(graph, unique_tag):
+    node_a = Node(id=f"Alice-{unique_tag}", type="LCPerson", properties={"role": "researcher"})
+    node_b = Node(id=f"Bob-{unique_tag}", type="LCPerson", properties={"role": "engineer"})
+    doc = GraphDocument(nodes=[node_a, node_b], relationships=[], source=Document(page_content="test"))
+
+    graph.add_graph_documents([doc])
+
+    result = graph.query(
+        "MATCH (n:LCPerson {name: $name}) RETURN n.name AS name",
+        params={"name": f"Alice-{unique_tag}"},
+    )
+    assert len(result) >= 1
+    assert result[0]["name"] == f"Alice-{unique_tag}"
+
+
+def test_add_graph_documents_creates_relationship(graph, unique_tag):
+    node_a = Node(id=f"Charlie-{unique_tag}", type="LCPerson2")
+    node_b = Node(id=f"GraphRAG-{unique_tag}", type="LCConcept")
+    rel = Relationship(source=node_a, target=node_b, type="LC_RESEARCHES")
+    doc = GraphDocument(
+        nodes=[node_a, node_b],
+        relationships=[rel],
+        source=Document(page_content="test"),
+    )
+
+    graph.add_graph_documents([doc])
+
+    # Verify both nodes exist
+    result = graph.query(
+        "MATCH (n:LCPerson2 {name: $name}) RETURN n.name AS name",
+        params={"name": f"Charlie-{unique_tag}"},
+    )
+    assert len(result) >= 1, f"source node not found: {result}"
+
+
+def test_add_graph_documents_idempotent(graph, unique_tag):
+    """Calling add_graph_documents twice must not raise."""
+    node = Node(id=f"Idempotent-{unique_tag}", type="LCIdempotent")
+    doc = GraphDocument(nodes=[node], relationships=[], source=Document(page_content="test"))
+
+    graph.add_graph_documents([doc])
+    graph.add_graph_documents([doc])  # second call must not raise
+
+    result = graph.query(
+        "MATCH (n:LCIdempotent {name: $name}) RETURN count(n) AS cnt",
+        params={"name": f"Idempotent-{unique_tag}"},
+    )
+    assert result[0]["cnt"] >= 1
+
+
+def test_schema_refreshes_after_add(graph, unique_tag):
+    """structured_schema is invalidated and re-fetched after add_graph_documents."""
+    graph._schema = None  # force refresh
+    schema_before = graph.schema
+
+    node = Node(id=f"SchemaNode-{unique_tag}", type="LCSchemaTest")
+    doc = GraphDocument(nodes=[node], relationships=[], source=Document(page_content="test"))
+    graph.add_graph_documents([doc])
+
+    graph.refresh_schema()
+    # schema must still be a string after refresh (content depends on server)
+    assert isinstance(graph.schema, str)

tests/integration/adapters/test_langchain.py

@@ -0,0 +1,141 @@
+"""Integration tests for CoordinodePropertyGraphStore (LlamaIndex adapter).
+
+Requires a running CoordiNode instance. Set COORDINODE_ADDR env var
+(default: localhost:7080).
+
+Run via:
+    COORDINODE_ADDR=localhost:17080 pytest tests/integration/adapters/test_llama_index.py -v
+"""
+
+import os
+import uuid
+
+import pytest
+from llama_index.core.graph_stores.types import EntityNode, Relation
+
+from llama_index.graph_stores.coordinode import CoordinodePropertyGraphStore
+
+ADDR = os.environ.get("COORDINODE_ADDR", "localhost:7080")
+
+
+@pytest.fixture(scope="module")
+def store():
+    with CoordinodePropertyGraphStore(ADDR) as s:
+        yield s
+
+
+@pytest.fixture
+def tag():
+    return uuid.uuid4().hex[:8]
+
+
+# ── Basic connectivity ────────────────────────────────────────────────────────
+
+def test_connect(store):
+    assert store is not None
+
+
+def test_get_schema(store):
+    schema = store.get_schema()
+    assert isinstance(schema, str)
+
+
+def test_structured_query_literal(store):
+    result = store.structured_query("RETURN 1 AS n")
+    assert isinstance(result, list)
+    assert result[0]["n"] == 1
+
+
+# ── Node operations ───────────────────────────────────────────────────────────
+
+def test_upsert_and_get_nodes(store, tag):
+    nodes = [
+        EntityNode(label="LITestPerson", name=f"Alice-{tag}", properties={"role": "researcher"}),
+        EntityNode(label="LITestConcept", name=f"GraphRAG-{tag}", properties={"field": "AI"}),
+    ]
+    store.upsert_nodes(nodes)
+
+    found = store.get(properties={"name": f"Alice-{tag}"})
+    assert len(found) >= 1
+    assert any(getattr(n, "name", None) == f"Alice-{tag}" for n in found)
+
+
+def test_upsert_nodes_idempotent(store, tag):
+    """Upserting the same node twice must not raise and must not duplicate."""
+    node = EntityNode(label="LIIdempotent", name=f"Idem-{tag}")
+    store.upsert_nodes([node])
+    store.upsert_nodes([node])  # second call must not raise
+
+    found = store.get(properties={"name": f"Idem-{tag}"})
+    assert len(found) >= 1
+
+
+def test_get_by_id(store, tag):
+    node = EntityNode(label="LIGetById", name=f"ById-{tag}")
+    node_id = node.id
+    store.upsert_nodes([node])
+
+    found = store.get(ids=[node_id])
+    assert len(found) >= 1
+
+
+# ── Relation operations ───────────────────────────────────────────────────────
+
+def test_upsert_and_get_triplets(store, tag):
+    src = EntityNode(label="LIRelPerson", name=f"Src-{tag}")
+    dst = EntityNode(label="LIRelConcept", name=f"Dst-{tag}")
+    store.upsert_nodes([src, dst])
+
+    rel = Relation(
+        label="LI_RESEARCHES",
+        source_id=src.id,
+        target_id=dst.id,
+        properties={"since": 2024},
+    )
+    store.upsert_relations([rel])
+
+    # CoordiNode does not support wildcard [r] patterns yet — must pass relation_names.
+    # See: get_triplets() implementation note.
+    triplets = store.get_triplets(
+        entity_names=[f"Src-{tag}"],
+        relation_names=["LI_RESEARCHES"],
+    )
+    assert isinstance(triplets, list)
+    assert len(triplets) >= 1
+
+    labels = [t[1].label for t in triplets]
+    assert "LI_RESEARCHES" in labels
+
+
+def test_get_rel_map(store, tag):
+    src = EntityNode(label="LIRelMap", name=f"RMapSrc-{tag}")
+    dst = EntityNode(label="LIRelMap", name=f"RMapDst-{tag}")
+    store.upsert_nodes([src, dst])
+
+    rel = Relation(label="LI_RELATED", source_id=src.id, target_id=dst.id)
+    store.upsert_relations([rel])
+
+    result = store.get_rel_map([src], depth=1, limit=10)
+    assert isinstance(result, list)
+
+
+# ── Delete ────────────────────────────────────────────────────────────────────
+
+def test_delete_by_id(store, tag):
+    node = EntityNode(label="LIDelete", name=f"Del-{tag}")
+    store.upsert_nodes([node])
+
+    store.delete(ids=[node.id])
+
+    found = store.get(ids=[node.id])
+    assert len(found) == 0
+
+
+def test_delete_by_entity_name(store, tag):
+    node = EntityNode(label="LIDeleteByName", name=f"DelNamed-{tag}")
+    store.upsert_nodes([node])
+
+    store.delete(entity_names=[f"DelNamed-{tag}"])
+
+    found = store.get(properties={"name": f"DelNamed-{tag}"})
+    assert len(found) == 0