fix: Change imports for DuplicateDocumentError and DuplicatePolicy as…

… per latest changes in haystack
prosto · Jan 19, 2024 · 7a1f053 · 7a1f053
1 parent 0131059
commit 7a1f053
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/neo4j_haystack/document_stores/neo4j_store.py b/src/neo4j_haystack/document_stores/neo4j_store.py
@@ -3,7 +3,8 @@
 
 import numpy as np
 from haystack import Document, default_from_dict, default_to_dict
-from haystack.document_stores import DuplicateDocumentError, DuplicatePolicy
+from haystack.document_stores.errors import DuplicateDocumentError
+from haystack.document_stores.types import DuplicatePolicy
 from neo4j.exceptions import DatabaseError
 from tqdm import tqdm
 

diff --git a/src/neo4j_haystack/errors.py b/src/neo4j_haystack/errors.py
@@ -1,6 +1,6 @@
 from typing import Optional
 
-from haystack.document_stores import DocumentStoreError
+from haystack.document_stores.errors import DocumentStoreError
 
 
 class Neo4jDocumentStoreError(DocumentStoreError):

diff --git a/tests/neo4j_haystack/document_stores/test_ds_protocol.py b/tests/neo4j_haystack/document_stores/test_ds_protocol.py
@@ -3,7 +3,8 @@
 
 import pytest
 from haystack import Document
-from haystack.document_stores import DuplicateDocumentError, DuplicatePolicy
+from haystack.document_stores.errors import DuplicateDocumentError
+from haystack.document_stores.types import DuplicatePolicy
 
 from neo4j_haystack.client import Neo4jClient, Neo4jClientConfig
 from neo4j_haystack.document_stores.neo4j_store import Neo4jDocumentStore