chore: switch to docling-core-provided MD export

Signed-off-by: Panos Vagenas <[email protected]>
DS4SD · Jul 18, 2024 · cbf92a6 · cbf92a6
1 parent f09ffcc
commit cbf92a6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 10 deletions.
diff --git a/docling/datamodel/document.py b/docling/datamodel/document.py
@@ -3,7 +3,6 @@
 from pathlib import Path, PurePath
 from typing import ClassVar, Dict, Iterable, List, Optional, Type, Union
 
-from deepsearch.documents.core.export import export_to_markdown
 from docling_core.types import BaseCell, BaseText
 from docling_core.types import BoundingBox as DsBoundingBox
 from docling_core.types import Document as DsDocument
@@ -299,9 +298,7 @@ def render_as_dict(self):
 
     def render_as_markdown(self):
         if self.output:
-            return export_to_markdown(
-                self.output.model_dump(by_alias=True, exclude_none=True)
-            )
+            return self.output.export_to_markdown()
         else:
             return ""
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,10 +23,9 @@ packages = [{include = "docling"}]
 [tool.poetry.dependencies]
 python = "^3.11"
 pydantic = "^2.0.0"
-docling-core = "^0.2.0"
+docling-core = "^1.1.0"
 docling-ibm-models = "^0.2.0"
 deepsearch-glm = ">=0.19.0,<1"
-deepsearch-toolkit = ">=0.47.0,<1"
 filetype = "^1.2.0"
 pypdfium2 = "^4.30.0"
 pydantic-settings = "^2.3.0"