github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-10-23 02:15:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: some imports for integration tests (#2612)

Browse Source 
Add more missed imports for integration tests. Bump `pytest` to the
current latest version.
Fix `tests/integration_tests/vectorstores/test_elasticsearch.py` to
update its cassette(easy fix).

Related PR: https://github.com/hwchase17/langchain/pull/2560
This commit is contained in:
sergerdn
2023-04-11 20:45:36 -07:00
committed by GitHub
parent c1521ddbdb
commit 4bdcedab54
8 changed files with 202527 additions and 1294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/integration_tests/vectorstores/test_elasticsearch.py
View File

@@ -8,9 +8,7 @@ import pytest
from elasticsearch import Elasticsearch
from langchain.docstore.document import Document
from langchain.document_loaders import TextLoader
from langchain.embeddings import OpenAIEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from langchain.vectorstores.elastic_vector_search import ElasticVectorSearch
from tests.integration_tests.vectorstores.fake_embeddings import FakeEmbeddings
@@ -45,16 +43,6 @@ class TestElasticsearch:
yield openai_api_key
@pytest.fixture(scope="class")
def documents(self) -> Generator[List[Document], None, None]:
"""Return a generator that yields a list of documents."""
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
documents = TextLoader(
os.path.join(os.path.dirname(__file__), "fixtures", "sharks.txt")
).load()
yield text_splitter.split_documents(documents)
def test_similarity_search_without_metadata(self, elasticsearch_url: str) -> None:
"""Test end to end construction and search without metadata."""
texts = ["foo", "bar", "baz"]