Test connection to Qdrant (apache#39273)

RodrigoGanancia · May 10, 2024 · caddf84 · caddf84
1 parent 5153f2b
commit caddf84
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)
  )