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
import pytest

View File

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