Test connection to Qdrant (#39273)

apache · Apr 26, 2024 · 0c96b06 · 0c96b06
1 parent 8188da7
commit 0c96b06
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 4 deletions.
diff --git a/airflow/providers/qdrant/hooks/qdrant.py b/airflow/providers/qdrant/hooks/qdrant.py
@@ -126,3 +126,7 @@ def verify_connection(self) -> tuple[bool, str]:
             return True, "Connection established!"
         except (UnexpectedResponse, RpcError, ValueError) as e:
             return False, str(e)
+
+    def test_connection(self) -> tuple[bool, str]:
+        """Test the connection to the Qdrant instance."""
+        return self.verify_connection()
diff --git a/airflow/providers/qdrant/provider.yaml b/airflow/providers/qdrant/provider.yaml
@@ -38,7 +38,7 @@ integrations:
     tags: [software]
 
 dependencies:
-  - qdrant_client>=1.7.0
+  - qdrant_client>=1.9.0
   - apache-airflow>=2.7.0
 
 hooks:

diff --git a/generated/provider_dependencies.json b/generated/provider_dependencies.json
@@ -950,7 +950,7 @@
   "qdrant": {
     "deps": [
       "apache-airflow>=2.7.0",
-      "qdrant_client>=1.7.0"
+      "qdrant_client>=1.9.0"
     ],
     "devel-deps": [],
     "cross-providers-deps": [],

diff --git a/scripts/ci/docker-compose/integration-qdrant.yml b/scripts/ci/docker-compose/integration-qdrant.yml
@@ -18,7 +18,7 @@
 version: "3.8"
 services:
   qdrant:
-    image: qdrant/qdrant:latest
+    image: qdrant/qdrant:v1.9.0
     labels:
       breeze.description: "Integration required for Qdrant tests."
     ports:

diff --git a/tests/integration/providers/qdrant/operators/test_qdrant_ingest.py b/tests/integration/providers/qdrant/operators/test_qdrant_ingest.py
@@ -60,7 +60,7 @@ def test_execute_hello(self):
 
         hook = operator.hook
 
-        hook.conn.recreate_collection(
+        hook.conn.create_collection(
             collection_name, vectors_config=VectorParams(size=dimensions, distance=Distance.COSINE)
         )