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

fix: dataset_id should be UUID

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-10-13 15:01:26 +01:00
parent de39d5c49a
commit b5852a2da7
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/api/v1/add/preprocessors/base.py
View file

@ -2,6 +2,7 @@
Base classes for the cognee add preprocessor system. Base classes for the cognee add preprocessor system.
""" """
from uuid import UUID
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional, Union, BinaryIO from typing import Any, Dict, List, Optional, Union, BinaryIO
from pydantic import BaseModel from pydantic import BaseModel
@ -23,7 +24,7 @@ class PreprocessorContext(BaseModel):
node_set: Optional[List[str]] = None node_set: Optional[List[str]] = None
vector_db_config: Optional[Dict] = None vector_db_config: Optional[Dict] = None
graph_db_config: Optional[Dict] = None graph_db_config: Optional[Dict] = None
dataset_id: Optional[str] = None dataset_id: Optional[UUID] = None
preferred_loaders: Optional[List[str]] = None preferred_loaders: Optional[List[str]] = None
incremental_loading: bool = True incremental_loading: bool = True
extra_params: Dict[str, Any] = {} extra_params: Dict[str, Any] = {}