feat: update all database adapters to use cognee 1.6.1 - #210
Merged
ArmagedonFlamer merged 6 commits intoSep 28, 2026
Merged
ArmagedonFlamer merged 6 commits into
ArmagedonFlamer merged 6 commits into
Conversation
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>
siillee
approved these changes
Sep 28, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
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
GraphDBInterface/VectorDBInterfaceare unchanged, and so are the factory constructor kwargs. The call shapes cognee core uses are also unchanged.get/set_node_frequency_weightsandget/set_edge_frequency_weights. No adapter implemented them.get_top_degree_node_ids,iter_bounded_neighborhood,get_entity_type_names(graph) andscore_by_ids(vector). No community adapter overrides them yet.postgres_demo.PostgresDemoAdapter.Changes
cognee==1.6.1pinned 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.PostgresDemoAdapter. The pgGraph setup now runs once per adapter, because the base class callsinitialize()from most of its methods. The SQL neighborhood fallback now delegates to the baseget_neighborhood; the text[] cast workaround is no longer needed.ladybug<0.18, which can't resolve alongside cognee 1.6.1'sladybug==0.19.0on Linux.test_concurrent_folded_writes_and_attaches_keep_every_ownerprovenance contract test.uv.lockandpoetry.lockfiles regenerated (Poetry 2.4.1, matching the existing locks). READMEs and.github/workflows/README.mdupdated.Verified locally
ruff checkandruff format --check: clean.tests/unitfor all 24 adapters on Python 3.11 with cognee 1.6.1: all pass.tests/test_pggraph.pyagainstpostgres:16withCOGNEE_PGGRAPH_TESTS=true: 16 passed.tests/unit tests/integrationagainsttypedb/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)
get_top_degree_node_ids/iter_bounded_neighborhoodfor graph backends that can bound the read server-side, andscore_by_idsfor vector stores that can filter by ID.🤖 Generated with Claude Code