github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-16 15:04:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: deprecate langchain_community Chroma in favor of langchain_chroma (#24474)

Browse Source
This commit is contained in:
ccurme
2024-07-22 11:00:13 -04:00
committed by GitHub
parent 0f7569ddbc
commit dcba7df2fe
67 changed files with 108 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/community/tests/unit_tests/chains/test_pebblo_retrieval.py
View File

@@ -11,7 +11,11 @@ from langchain_core.callbacks import (
CallbackManagerForRetrieverRun,
)
from langchain_core.documents import Document
from langchain_core.vectorstores import VectorStore, VectorStoreRetriever
from langchain_core.vectorstores import (
InMemoryVectorStore,
VectorStore,
VectorStoreRetriever,
)
from langchain_community.chains import PebbloRetrievalQA
from langchain_community.chains.pebblo_retrieval.models import (
@@ -19,7 +23,6 @@ from langchain_community.chains.pebblo_retrieval.models import (
ChainInput,
SemanticContext,
)
from langchain_community.vectorstores.chroma import Chroma
from langchain_community.vectorstores.pinecone import Pinecone
from tests.unit_tests.llms.fake_llm import FakeLLM
@@ -49,8 +52,8 @@ def unsupported_retriever() -> FakeRetriever:
"""
retriever = FakeRetriever()
retriever.search_kwargs = {}
# Set the class of vectorstore to Chroma
retriever.vectorstore.__class__ = Chroma
# Set the class of vectorstore
retriever.vectorstore.__class__ = InMemoryVectorStore
return retriever