From 2b2544bfc5c9940088c7d18eb1f0b57974c40d08 Mon Sep 17 00:00:00 2001
From: Panos Vagenas <35837085+vagenas@users.noreply.github.com>
Date: Thu, 17 Oct 2024 17:29:46 +0200
Subject: [PATCH] fix: fix legacy doc ref (#188)

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 b6a73a9..77aaa86 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 1d80cb8..501bc7a 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