From 899275c25edb0c3814368a4d8560abe8e096d968 Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Tue, 26 Nov 2024 16:38:24 +0100 Subject: [PATCH] Rename metadata field to metadata_repr --- cognee/modules/ingestion/models/Metadata.py | 5 +++-- cognee/modules/ingestion/operations/write_metadata.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/cognee/modules/ingestion/models/Metadata.py b/cognee/modules/ingestion/models/Metadata.py index ab9fe1e01..5d6333bc3 100644 --- a/cognee/modules/ingestion/models/Metadata.py +++ b/cognee/modules/ingestion/models/Metadata.py @@ -7,10 +7,11 @@ from cognee.infrastructure.databases.relational import Base class Metadata(Base): - __tablename__ = "queries" + __tablename__ = "metadata_table" id = Column(UUID, primary_key=True, default=uuid4) - metadata = Column(String) + metadata_repr = Column(String) + metadata_source = Column(String) created_at = Column( DateTime(timezone=True), default=lambda: datetime.now(timezone.utc) diff --git a/cognee/modules/ingestion/operations/write_metadata.py b/cognee/modules/ingestion/operations/write_metadata.py index cefed21de..b1b40a3d4 100644 --- a/cognee/modules/ingestion/operations/write_metadata.py +++ b/cognee/modules/ingestion/operations/write_metadata.py @@ -15,7 +15,7 @@ async def write_metadata(data_item: Any) -> UUID: db_engine = get_relational_engine() async with db_engine.get_async_session() as session: metadata = Metadata( - metadata=json.dumps(metadata_dict), + metadata_repr=json.dumps(metadata_dict), metadata_source=parse_type(type(data_item)), ) session.add(metadata)