fix: Qdrant collection exists check (#11657)

* fix: Qdrant collection exists check * fix: Handle RpcError
run-llama · Mar 6, 2024 · 0546134 · 0546134
1 parent 0f9aaaa
commit 0546134
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...s/vector_stores/llama-index-vector-stores-qdrant/llama_index/vector_stores/qdrant/base.py b/...s/vector_stores/llama-index-vector-stores-qdrant/llama_index/vector_stores/qdrant/base.py
@@ -370,7 +370,7 @@ def _create_collection(self, collection_name: str, vector_size: int) -> None:
                         distance=rest.Distance.COSINE,
                     ),
                 )
-        except (ValueError, UnexpectedResponse) as exc:
+        except (RpcError, ValueError, UnexpectedResponse) as exc:
             if "already exists" not in str(exc):
                 raise exc  # noqa: TRY201
             logger.warning(
@@ -408,7 +408,7 @@ async def _acreate_collection(self, collection_name: str, vector_size: int) -> N
                         distance=rest.Distance.COSINE,
                     ),
                 )
-        except (ValueError, UnexpectedResponse) as exc:
+        except (RpcError, ValueError, UnexpectedResponse) as exc:
             if "already exists" not in str(exc):
                 raise exc  # noqa: TRY201
             logger.warning(