fix blanklineSplitter (#129)

nuclia · Dec 9, 2024 · 423b905 · 423b905
1 parent 9fb7070
commit 423b905
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,7 +3,7 @@
 ## 4.3.8 (unreleased)
 
 
-- Nothing changed yet.
+- Support blanklineSplitter
 
 
 ## 4.3.7 (2024-12-02)

diff --git a/nuclia/sdk/upload.py b/nuclia/sdk/upload.py
@@ -62,6 +62,7 @@ def file(
         rid: Optional[str] = None,
         field: Optional[str] = None,
         interpretTables: Optional[bool] = False,
+        blanklineSplitter: Optional[bool] = False,
         mimetype: Optional[str] = None,
         **kwargs,
     ) -> Optional[str]:
@@ -74,6 +75,8 @@ def file(
             mimetype = "application/octet-stream"
         if interpretTables:
             mimetype += "+aitable"
+        if blanklineSplitter:
+            mimetype += "+blankline"
         is_new_resource = False
         if rid is not None:
             rid, is_new_resource = self._get_or_create_resource(
@@ -244,6 +247,7 @@ def remote(
         rid: Optional[str] = None,
         field: Optional[str] = "file",
         interpretTables: Optional[bool] = False,
+        blanklineSplitter: Optional[bool] = False,
         **kwargs,
     ) -> str:
         """Upload a remote url to a Nuclia KnowledgeBox"""
@@ -263,6 +267,8 @@ def remote(
             mimetype = r.headers.get("Content-Type", "application/octet-stream")
             if interpretTables:
                 mimetype += "+aitable"
+            if blanklineSplitter:
+                mimetype += "+blankline"
             rid, is_new_resource = self._get_or_create_resource(
                 rid=rid, icon=mimetype, **kwargs
             )
@@ -368,6 +374,7 @@ async def file(
         field: Optional[str] = None,
         mimetype: Optional[str] = None,
         interpretTables: Optional[bool] = False,
+        blanklineSplitter: Optional[bool] = False,
         **kwargs,
     ) -> str:
         """Upload a file from filesystem to a Nuclia KnowledgeBox"""
@@ -379,6 +386,8 @@ async def file(
             mimetype = "application/octet-stream"
         if interpretTables:
             mimetype += "+aitable"
+        if blanklineSplitter:
+            mimetype += "+blankline"
         rid, is_new_resource = await self._get_or_create_resource(
             rid=rid, icon=mimetype, **kwargs
         )
@@ -542,6 +551,7 @@ async def remote(
         rid: Optional[str] = None,
         field: Optional[str] = "file",
         interpretTables: Optional[bool] = False,
+        blanklineSplitter: Optional[bool] = False,
         **kwargs,
     ) -> str:
         """Upload a remote url to a Nuclia KnowledgeBox"""
@@ -556,6 +566,8 @@ async def remote(
             mimetype = r.headers.get("Content-Type", "application/octet-stream")
             if interpretTables:
                 mimetype += "+aitable"
+            if blanklineSplitter:
+                mimetype += "+blankline"
             rid, is_new_resource = await self._get_or_create_resource(
                 rid=rid, icon=mimetype, **kwargs
             )