fix integration deps (#9722)

This commit is contained in:
Bagatur 2023-08-24 15:06:53 -07:00 committed by GitHub
parent a0800c9f15
commit 0f48e6c36e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions

View File

@ -1,3 +1,5 @@
from __future__ import annotations
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
import pytest import pytest

View File

@ -5,7 +5,6 @@ import uuid
from typing import Generator, List, Union from typing import Generator, List, Union
import pytest import pytest
from elasticsearch.helpers import BulkIndexError
from langchain.docstore.document import Document from langchain.docstore.document import Document
from langchain.vectorstores.elasticsearch import ElasticsearchStore from langchain.vectorstores.elasticsearch import ElasticsearchStore
@ -585,6 +584,7 @@ class TestElasticsearch:
caplog: pytest.LogCaptureFixture, caplog: pytest.LogCaptureFixture,
) -> None: ) -> None:
"""Test bulk exception logging is giving better hints.""" """Test bulk exception logging is giving better hints."""
from elasticsearch.helpers import BulkIndexError
docsearch = ElasticsearchStore( docsearch = ElasticsearchStore(
embedding=ConsistentFakeEmbeddings(), embedding=ConsistentFakeEmbeddings(),