SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 20:12:55 +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.

8
private_gpt/components/vector_store/vector_store_component.py
View file

@ -40,7 +40,9 @@ class VectorStoreComponent:
match settings.vectorstore.database:
case "pgvector":
try:
from llama_index.vector_stores.postgres import PGVectorStore
from llama_index.vector_stores.postgres import ( # type: ignore
PGVectorStore,
)
except ImportError as e:
raise ImportError(
"Postgres dependencies not found, install with `poetry install --extras postgres`"
@ -91,7 +93,9 @@ class VectorStoreComponent:
case "qdrant":
try:
from llama_index.vector_stores.qdrant import QdrantVectorStore
from llama_index.vector_stores.qdrant import ( # type: ignore
QdrantVectorStore,
)
from qdrant_client import QdrantClient
except ImportError as e:
raise ImportError(