From b5852a2da7d58b04938523e4a6ee9c11876757b1 Mon Sep 17 00:00:00 2001 From: Daulet Amirkhanov Date: Mon, 13 Oct 2025 15:01:26 +0100 Subject: [PATCH] fix: dataset_id should be UUID --- cognee/api/v1/add/preprocessors/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cognee/api/v1/add/preprocessors/base.py b/cognee/api/v1/add/preprocessors/base.py index 97577f10b..40b501d3b 100644 --- a/cognee/api/v1/add/preprocessors/base.py +++ b/cognee/api/v1/add/preprocessors/base.py @@ -2,6 +2,7 @@ Base classes for the cognee add preprocessor system. """ +from uuid import UUID from abc import ABC, abstractmethod from typing import Any, Dict, List, Optional, Union, BinaryIO from pydantic import BaseModel @@ -23,7 +24,7 @@ class PreprocessorContext(BaseModel): node_set: Optional[List[str]] = None vector_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 incremental_loading: bool = True extra_params: Dict[str, Any] = {}