Skip to content

Commit 58cde90

Browse files
committed
fix(qdrant): migrate Qdrant from deprecated decorator.
1 parent 15e99ee commit 58cde90

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

modules/qdrant/testcontainers/qdrant/__init__.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,11 @@
1515
from pathlib import Path
1616
from typing import Optional
1717

18-
from testcontainers.core.config import testcontainers_config as c
19-
from testcontainers.core.generic import DbContainer
20-
from testcontainers.core.waiting_utils import wait_container_is_ready, wait_for_logs
18+
from testcontainers.core.container import DockerContainer
19+
from testcontainers.core.wait_strategies import LogMessageWaitStrategy
2120

2221

23-
class QdrantContainer(DbContainer):
22+
class QdrantContainer(DockerContainer):
2423
"""
2524
Qdrant vector database container.
2625
@@ -59,9 +58,8 @@ def __init__(
5958
def _configure(self) -> None:
6059
self.with_env("QDRANT__SERVICE__API_KEY", self._api_key)
6160

62-
@wait_container_is_ready()
6361
def _connect(self) -> None:
64-
wait_for_logs(self, ".*Actix runtime found; starting in Actix runtime.*", c.timeout)
62+
LogMessageWaitStrategy(".*Actix runtime found; starting in Actix runtime.*").wait_until_ready(self)
6563

6664
def get_client(self, **kwargs):
6765
"""

0 commit comments

Comments
 (0)