From d2b7e178a63031291427fc75c8c0bc8c71e380d2 Mon Sep 17 00:00:00 2001
From: Panos Vagenas <35837085+vagenas@users.noreply.github.com>
Date: Thu, 17 Oct 2024 17:11:54 +0200
Subject: [PATCH] fix: fix legacy doc ref

Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com>
---
 deepsearch/documents/core/export.py | 2 +-
 deepsearch/documents/core/utils.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/deepsearch/documents/core/export.py b/deepsearch/documents/core/export.py
index b6a73a91..77aaa869 100644
--- 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
index 1d80cb81..501bc7a6 100644
--- 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