|
6 | 6 | from pathway.engine import ExternalIndexFactory |
7 | 7 | from pathway.tests.utils import assert_table_equality |
8 | 8 |
|
| 9 | +QDRANT_URL = "http://qdrant:6334" |
| 10 | + |
9 | 11 |
|
10 | 12 | def make_list(vector_as_str: str) -> list[float]: |
11 | 13 | return [float(x) for x in vector_as_str.split(",")] |
@@ -42,7 +44,7 @@ def test_basic_search(): |
42 | 44 | ).with_columns(data=pw.apply_with_type(make_list, list[float], pw.this.data)) |
43 | 45 |
|
44 | 46 | index_factory = ExternalIndexFactory.qdrant_factory( |
45 | | - url="http://localhost:6334", |
| 47 | + url=QDRANT_URL, |
46 | 48 | collection_name="test_basic", |
47 | 49 | vector_size=3, |
48 | 50 | ) |
@@ -91,7 +93,7 @@ def test_with_deletions(): |
91 | 93 | ).with_columns(data=pw.apply_with_type(make_list, list[float], pw.this.data)) |
92 | 94 |
|
93 | 95 | index_factory = ExternalIndexFactory.qdrant_factory( |
94 | | - url="http://localhost:6334", |
| 96 | + url=QDRANT_URL, |
95 | 97 | collection_name="test_deletions", |
96 | 98 | vector_size=3, |
97 | 99 | ) |
@@ -153,7 +155,7 @@ class QuerySchemaWithFilter(pw.Schema): |
153 | 155 | ).with_columns(data=pw.apply(make_list, pw.this.data)) |
154 | 156 |
|
155 | 157 | index_factory = ExternalIndexFactory.qdrant_factory( |
156 | | - url="http://localhost:6334", |
| 158 | + url=QDRANT_URL, |
157 | 159 | collection_name="test_filter", |
158 | 160 | vector_size=3, |
159 | 161 | ) |
|
0 commit comments