fix: fix legacy doc ref (#188)

Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com>
DS4SD · Oct 17, 2024 · 2b2544b · 2b2544b
1 parent 17d5c4c
commit 2b2544b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepsearch/documents/core/export.py b/deepsearch/documents/core/export.py
@@ -3,7 +3,7 @@
 import warnings
 from typing import Any, Dict
 
-from docling_core.types import Document as DsDocument
+from docling_core.types.legacy_doc.document import ExportedCCSDocument as DsDocument
 
 _log = logging.getLogger(__name__)
 

diff --git a/deepsearch/documents/core/utils.py b/deepsearch/documents/core/utils.py
@@ -9,7 +9,7 @@
 from typing import Any, Dict, Iterator, List, Optional
 
 import requests
-from docling_core.types import Document as DsDocument
+from docling_core.types.legacy_doc.document import ExportedCCSDocument as DsDocument
 from pydantic import BaseModel
 from tqdm import tqdm