pull/21191/head
Eugene Yurtsev 2 months ago
parent 690d7314af
commit 7f8101e30f

@ -1,11 +1,12 @@
"""Integration test for embedding-based redundant doc filtering."""
from langchain_core.documents import Document
from langchain_community.document_transformers.embeddings_redundant_filter import (
EmbeddingsClusteringFilter,
EmbeddingsRedundantFilter,
_DocumentWithState,
)
from langchain_community.embeddings import OpenAIEmbeddings
from langchain_core.documents import Document
def test_embeddings_redundant_filter() -> None:

@ -2,9 +2,10 @@ import pathlib
from typing import Any, Dict, List
import pandas as pd
from langchain.chains.graph_qa.prompts import CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT
from langchain.memory import ConversationBufferMemory, ReadOnlySharedMemory
from langchain_core.prompts import PromptTemplate
from langchain_community.chains.graph_qa.cypher import (
GraphCypherQAChain,
construct_schema,
@ -16,7 +17,6 @@ from langchain_community.chains.graph_qa.cypher_utils import (
)
from langchain_community.graphs.graph_document import GraphDocument
from langchain_community.graphs.graph_store import GraphStore
from langchain_core.prompts import PromptTemplate
from tests.unit_tests.llms.fake_llm import FakeLLM

Loading…
Cancel
Save