gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: add attributes to edges

Browse source
This commit is contained in:
Boris Arzentar 2025-11-22 13:09:31 +01:00
parent 663dd16647
commit cb380e51e9
No known key found for this signature in database
GPG key ID: D5CC274C784807B7
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
alembic/versions/84e5d08260d6_replace_graph_ledger_table_with_nodes_.py
View file

@ -36,6 +36,7 @@ def upgrade() -> None:
sa.Column("dataset_id", sa.UUID, index=True),
sa.Column("label", sa.String()),
sa.Column("type", sa.String(), nullable=False),
sa.Column("attributes", sa.JSON()),
sa.Column("indexed_fields", sa.JSON(), nullable=False),
sa.Column(
"created_at", sa.DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
@ -54,7 +55,7 @@ def upgrade() -> None:
sa.Column("destination_node_id", sa.UUID, nullable=False),
sa.Column("label", sa.Text()),
sa.Column("relationship_name", sa.Text(), nullable=False),
sa.Column("props", sa.JSON()),
sa.Column("attributes", sa.JSON()),
sa.Column(
"created_at", sa.DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
),

2
cognee/modules/graph/methods/upsert_nodes.py
View file

@ -1,5 +1,6 @@
from typing import List
from uuid import NAMESPACE_OID, UUID, uuid5
from fastapi.encoders import jsonable_encoder
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.dialects.postgresql import insert
@ -44,6 +45,7 @@ async def upsert_nodes(
"type": node.type,
"indexed_fields": DataPoint.get_embeddable_property_names(node),
"label": getattr(node, "label", getattr(node, "name", str(node.id))),
"attributes": jsonable_encoder(node),
}
for node in nodes
]