Skip to content

feat: update all database adapters to use cognee 1.6.1 - #210

Merged
ArmagedonFlamer merged 6 commits into
mainfrom
feature/sdk-857-bump-cognee-community-database-adapters-to-cognee-161
Sep 28, 2026
Merged

ArmagedonFlamer merged 6 commits into
mainfrom
feature/sdk-857-bump-cognee-community-database-adapters-to-cognee-161

Conversation

@ArmagedonFlamer

Copy link
Copy Markdown
Contributor

Resolves SDK-857. Moves every graph, vector and hybrid adapter from cognee 1.4.2 to 1.6.1. Supersedes #190: its four commits are cherry-picked here with authorship kept, so #190 can be closed.

What changed between 1.5.4 and 1.6.1 for adapters

  • The required methods of GraphDBInterface / VectorDBInterface are unchanged, and so are the factory constructor kwargs. The call shapes cognee core uses are also unchanged.
  • Removed: get/set_node_frequency_weights and get/set_edge_frequency_weights. No adapter implemented them.
  • New optional hooks, with working defaults: get_top_degree_node_ids, iter_bounded_neighborhood, get_entity_type_names (graph) and score_by_ids (vector). No community adapter overrides them yet.
  • The built-in Postgres graph adapter is now postgres_demo.PostgresDemoAdapter.

Changes

  • cognee==1.6.1 pinned in all 24 adapter packages. pggraph and typedb were still on 1.4.2. litellm stays at 1.96.2, which is what 1.6.1 locks.
  • pggraph: rebased on PostgresDemoAdapter. The pgGraph setup now runs once per adapter, because the base class calls initialize() from most of its methods. The SQL neighborhood fallback now delegates to the base get_neighborhood; the text[] cast workaround is no longer needed.
  • typedb:
    • Dropped ladybug<0.18, which can't resolve alongside cognee 1.6.1's ladybug==0.19.0 on Linux.
    • Removed the frequency-weights test and README note.
    • Ported upstream's new test_concurrent_folded_writes_and_attaches_keep_every_owner provenance contract test.
  • Contract suite: now cites the 1.6.1 call sites, and checks the call shape of the new optional hooks whenever an adapter overrides one.
  • Package versions: patch bumps. None of feat: update all database adapters to use cognee 1.5.4 #190's versions were ever published. opensearch goes to 1.0.1 because PyPI already has 1.0.0. typedb stays at 0.1.0 since it is unpublished.
  • All uv.lock and poetry.lock files regenerated (Poetry 2.4.1, matching the existing locks). READMEs and .github/workflows/README.md updated.

Verified locally

  • ruff check and ruff format --check: clean.
  • tests/unit for all 24 adapters on Python 3.11 with cognee 1.6.1: all pass.
  • pggraph tests/test_pggraph.py against postgres:16 with COGNEE_PGGRAPH_TESTS=true: 16 passed.
  • typedb tests/unit tests/integration against typedb/typedb:3.12.3: 109 passed.

Not run locally: the other adapters' integration and e2e tiers, which need their services or LLM secrets. Those are left to CI.

Follow-ups (not in this PR)

  • Native get_top_degree_node_ids / iter_bounded_neighborhood for graph backends that can bound the read server-side, and score_by_ids for vector stores that can filter by ID.
  • Publishing the new versions to PyPI.

🤖 Generated with Claude Code

siillee and others added 6 commits September 28, 2026 12:54
Builds on the 1.5.4 work (cherry-picked from #190) and moves every graph,
vector and hybrid adapter to cognee==1.6.1.

- Pins cognee 1.6.1 in all 24 adapter packages, including pggraph and
  typedb, which were still on 1.4.2; litellm stays at 1.96.2, which 1.6.1
  still locks.
- pggraph: port from the removed postgres.PostgresAdapter to
  postgres_demo.PostgresDemoAdapter; make pgGraph setup run once per
  adapter now that the base calls initialize() from most methods; fall
  back to the base get_neighborhood, which no longer needs the text[]
  cast workaround.
- typedb: drop the ladybug<0.18 pin, which conflicts with cognee 1.6.1's
  ladybug==0.19.0; drop the frequency-weight test and README note (the
  methods were removed from GraphDBInterface); port the new upstream
  concurrent folded-write provenance contract test.
- Contract suite: cite 1.6.1 call sites; check the call shape of the new
  optional hooks (get_top_degree_node_ids, iter_bounded_neighborhood,
  get_entity_type_names, score_by_ids) when an adapter overrides them.
- opensearch: version 1.0.1, since 1.0.0 is already on PyPI.
- Regenerate every uv.lock and poetry.lock.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
cognee 1.6.1 declares pydantic>=2.10.5, so re-locking kept the old 2.10.5
that networkx, azureaisearch, opensearch and redis had locked. cognee
1.6.1 fails at import on it ("Pydantic does not support mixing more than
one of TypeVar bounds, constraints and defaults"). Upgrade pydantic,
pydantic-core and typing-extensions in those locks to what cognee 1.6.1
itself locks.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
@ArmagedonFlamer
ArmagedonFlamer merged commit 8708b14 into main Sep 28, 2026
44 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants