Skip to content

[Feature]: Custom label names #1769

@Vasilije1990

Description

@Vasilije1990

Problem Statement

As Cognee data can be anything a way to provide a custom name for data in Cognee is needed, with this users can have a way to differentiate between different data objects.

If a text is provided directly to Cognee a custom name will enable users to easily differentiate text by data name (default text name is just the content hash).

This can be achieved by expanding the Data SQL model to also support a label column and filling the column value with the provided label value during Cognee add.

To send a label value along with data in Cognee add create and add support for a python Dataclass you can call DataItem.
This DataItem class should contain the data value along with the label value as attributes. When a Dataclass is forwarded like this extract the data during ingest_data and forward the data value as it's currently forwarded and add the label value to the Data model SQL instance (in ingest_data)

In the end expand the get_dataset_data function to also return the new label information, make sure the get_dataset_data router in the Cognee backend also works with this change

Proposed Solution

TBD

Alternatives Considered

No response

Use Case

Add custom label names that are more human friendly and that will help user use the product

Implementation Ideas

No response

Additional Context

No response

Pre-submission Checklist

  • I have searched existing issues to ensure this feature hasn't been requested already
  • I have provided a clear problem statement and proposed solution
  • I have described my specific use case

Metadata

Metadata

Labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions