github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-06 13:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Default Chroma collection name (#1198)

Browse Source 
For persistence, it's convenient to have a default collection name which
gets used everywhere.
This commit is contained in:
Anton Troynikov 2023-02-20 20:59:34 -08:00 committed by GitHub
parent 23243ae69c
commit d2ef5d6167
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
langchain/vectorstores/chroma.py
View File

@ -27,9 +27,11 @@ class Chroma(VectorStore):
vectorstore = Chroma("langchain_store", embeddings.embed_query)
"""
_LANGCHAIN_DEFAULT_COLLECTION_NAME = "langchain"
def __init__(
self,
collection_name: str,
collection_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
embedding_function: Optional[Embeddings] = None,
persist_directory: Optional[str] = None,
) -> None:
@ -155,7 +157,7 @@ class Chroma(VectorStore):
embedding: Optional[Embeddings] = None,
metadatas: Optional[List[dict]] = None,
ids: Optional[List[str]] = None,
collection_name: str = "langchain",
collection_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
persist_directory: Optional[str] = None,
**kwargs: Any,
) -> Chroma:
@ -189,7 +191,7 @@ class Chroma(VectorStore):
documents: List[Document],
embedding: Optional[Embeddings] = None,
ids: Optional[List[str]] = None,
collection_name: str = "langchain",
collection_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
persist_directory: Optional[str] = None,
**kwargs: Any,
) -> Chroma: