File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed
src/flare_ai_rag/retriever Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change 1313def _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." ,
You can’t perform that action at this time.
0 commit comments