SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 23:22:57 +01:00
Code Issues Projects Releases Packages Activity

Fix mypy

Browse source
This commit is contained in:
imartinez 2024-02-29 15:20:26 +01:00
parent 34d48d7b4d
commit 63d3b9f936
7 changed files with 39 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

4
private_gpt/components/vector_store/batched_chroma.py
View file

@ -3,7 +3,7 @@ from typing import Any
from llama_index.core.schema import BaseNode, MetadataMode
from llama_index.core.vector_stores.utils import node_to_metadata_dict
from llama_index.vector_stores.chroma import ChromaVectorStore
from llama_index.vector_stores.chroma import ChromaVectorStore # type: ignore
def chunk_list(
@ -22,7 +22,7 @@ def chunk_list(
yield lst[i : i + max_chunk_size]
class BatchedChromaVectorStore(ChromaVectorStore):
class BatchedChromaVectorStore(ChromaVectorStore): # type: ignore
"""Chroma vector store, batching additions to avoid reaching the max batch limit.
In this vector store, embeddings are stored within a ChromaDB collection.