github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-07 13:40:46 +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) vectorstore = Chroma("langchain_store", embeddings.embed_query)
""" """
_LANGCHAIN_DEFAULT_COLLECTION_NAME = "langchain"
def __init__( def __init__(
self, self,
collection_name: str, collection_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
embedding_function: Optional[Embeddings] = None, embedding_function: Optional[Embeddings] = None,
persist_directory: Optional[str] = None, persist_directory: Optional[str] = None,
) -> None: ) -> None:
@ -155,7 +157,7 @@ class Chroma(VectorStore):
embedding: Optional[Embeddings] = None, embedding: Optional[Embeddings] = None,
metadatas: Optional[List[dict]] = None, metadatas: Optional[List[dict]] = None,
ids: Optional[List[str]] = None, ids: Optional[List[str]] = None,
collection_name: str = "langchain", collection_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
persist_directory: Optional[str] = None, persist_directory: Optional[str] = None,
**kwargs: Any, **kwargs: Any,
) -> Chroma: ) -> Chroma:
@ -189,7 +191,7 @@ class Chroma(VectorStore):
documents: List[Document], documents: List[Document],
embedding: Optional[Embeddings] = None, embedding: Optional[Embeddings] = None,
ids: Optional[List[str]] = None, ids: Optional[List[str]] = None,
collection_name: str = "langchain", collection_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
persist_directory: Optional[str] = None, persist_directory: Optional[str] = None,
**kwargs: Any, **kwargs: Any,
) -> Chroma: ) -> Chroma: