Merge remote-tracking branch 'upstream/main'

cognee/api/v1/add/routers/get_add_router.py

@@ -21,7 +21,8 @@ def get_add_router() -> APIRouter:
     async def add(
         file: UploadFile = UploadFile(None),
         url: str = Form(None),
-        datasetId: str = Form(...),
+        datasetId: Optional[UUID] = Form(default=None),
+        datasetName: Optional[str] = Form(default=None),
         user: User = Depends(get_authenticated_user),
     ):
         """
@@ -31,6 +32,17 @@ async def add(
         - url: a URL to fetch data from (supports GitHub clone or direct file download)
         """
         from cognee.api.v1.add import add as cognee_add
+
+        if not datasetId and not datasetName:
+            raise ValueError("Either datasetId or datasetName must be provided.")
+
+        if datasetId and not datasetName:
+            dataset = await get_dataset(user_id=user.id, dataset_id=datasetId)
+            try:
+                datasetName = dataset.name
+            except IndexError:
+                raise ValueError("No dataset found with the provided datasetName.")
+
         try:
             logger.info(f"Received data for datasetId={datasetId}")
             if file and file.filename:
@@ -40,15 +52,15 @@ async def add(
                     logger.info(f"Passing uploaded file as text to cognee_add")
                     return await cognee_add(
                         text,
-                        datasetId,
+                        datasetName,
                         user=user
                     )
                 except Exception as e:
                     logger.info(f"Could not decode file as text, falling back to binary. Error: {e}")
                     file.file.seek(0)
                     return await cognee_add(
                         file.file,
-                        datasetId,
+                        datasetName,
                         user=user
                     )
             elif url:
@@ -72,7 +84,7 @@ async def add(
                         logger.info(f"Fetched content from URL: {response.text}")
                         return await cognee_add(
                             response.text,
-                            datasetId,
+                            datasetName,
                             user=user
                         )
                 else: