Skip to content

Commit e464469

Browse files
committed
fix(errors): downgrade file size errors to warnings
1 parent e91f610 commit e464469

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/flare_ai_rag/retriever/qdrant_collection.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,11 @@
1313
def _create_collection(
1414
client: QdrantClient, collection_name: str, vector_size: int
1515
) -> None:
16+
"""
17+
Creates a Qdrant collection with the given parameters.
18+
:param collection_name: Name of the collection.
19+
:param vector_size: Dimension of the vectors.
20+
"""
1621
client.recreate_collection(
1722
collection_name=collection_name,
1823
vectors_config=VectorParams(size=vector_size, distance=Distance.COSINE),
@@ -25,6 +30,7 @@ def generate_collection(
2530
retriever_config: RetrieverConfig,
2631
embedding_client: GeminiEmbedding,
2732
) -> None:
33+
"""Routine for generating a Qdrant collection for a specific CSV file type."""
2834
_create_collection(
2935
qdrant_client, retriever_config.collection_name, retriever_config.vector_size
3036
)
@@ -54,6 +60,7 @@ def generate_collection(
5460
)
5561
except google.api_core.exceptions.InvalidArgument as e:
5662
# Check if it's the known "Request payload size exceeds the limit" error
63+
# If so, downgrade it to a warning
5764
if "400 Request payload size exceeds the limit" in str(e):
5865
logger.warning(
5966
"Skipping document due to size limit.",

0 commit comments

Comments
 (0)