fix: fix legacy doc ref

Signed-off-by: Panos Vagenas <[email protected]>
DS4SD · Oct 17, 2024 · d2b7e17 · d2b7e17
1 parent 17d5c4c
commit d2b7e17
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