Skip to content

Commit 4f58cec

Browse files
chore: remove large JSON test files from tracking
1 parent 7e96f02 commit 4f58cec

File tree

3 files changed

+24
-0
lines changed

3 files changed

+24
-0
lines changed
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
from dataclasses import dataclass
2+
from typing import Optional
3+
from cognee.modules.pipelines.models.DataItemStatus import DataItemStatus
4+
5+
@dataclass
6+
class DataItem:
7+
id: str
8+
name: str
9+
source: str
10+
status: DataItemStatus
11+
label: Optional[str] = None # new field to add a label

cognee/modules/pipelines/models/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,4 @@
77
PipelineRunErrored,
88
)
99
from .DataItemStatus import DataItemStatus
10+
from .DataItem import DataItem
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from cognee.modules.pipelines.models.DataItem import DataItem
2+
from cognee.modules.pipelines.models.DataItemStatus import DataItemStatus
3+
4+
def test_data_item_label_field():
5+
item = DataItem(
6+
id="123",
7+
name="Sample Item",
8+
source="mock_source",
9+
status=DataItemStatus.DATA_ITEM_PROCESSING_COMPLETED,
10+
label="Important"
11+
)
12+
assert item.label == "Important"

0 commit comments

Comments
 (0)