|
15 | 15 | import asyncio |
16 | 16 | from time import time_ns |
17 | 17 |
|
18 | | -import valkey |
19 | | -import valkey.asyncio |
20 | | -from valkey.commands.search.field import ( |
| 18 | +import redis |
| 19 | +import redis.asyncio |
| 20 | +from redis.commands.search.field import ( |
21 | 21 | TextField, |
22 | 22 | VectorField, |
23 | 23 | ) |
24 | | -from valkey.commands.search.indexDefinition import IndexDefinition, IndexType |
25 | | -from valkey.commands.search.query import Query |
26 | | -from valkey.exceptions import ResponseError |
| 24 | +from redis.commands.search.indexDefinition import IndexDefinition, IndexType |
| 25 | +from redis.commands.search.query import Query |
| 26 | +from redis.exceptions import ResponseError |
27 | 27 |
|
28 | 28 | from opentelemetry import trace |
29 | 29 | from opentelemetry.instrumentation.redis import RedisInstrumentor |
|
34 | 34 | class TestRedisInstrument(TestBase): |
35 | 35 | def setUp(self): |
36 | 36 | super().setUp() |
37 | | - self.redis_client = valkey.Redis(port=6379) |
| 37 | + self.redis_client = redis.Redis(port=6379) |
38 | 38 | self.redis_client.flushall() |
39 | 39 | RedisInstrumentor().instrument(tracer_provider=self.tracer_provider) |
40 | 40 |
|
@@ -209,7 +209,7 @@ def test_parent(self): |
209 | 209 | class TestRedisClusterInstrument(TestBase): |
210 | 210 | def setUp(self): |
211 | 211 | super().setUp() |
212 | | - self.redis_client = valkey.cluster.RedisCluster( |
| 212 | + self.redis_client = redis.cluster.RedisCluster( |
213 | 213 | host="localhost", port=7000 |
214 | 214 | ) |
215 | 215 | self.redis_client.flushall() |
@@ -280,7 +280,7 @@ def async_call(coro): |
280 | 280 | class TestAsyncRedisInstrument(TestBase): |
281 | 281 | def setUp(self): |
282 | 282 | super().setUp() |
283 | | - self.redis_client = valkey.asyncio.Redis(port=6379) |
| 283 | + self.redis_client = redis.asyncio.Redis(port=6379) |
284 | 284 | async_call(self.redis_client.flushall()) |
285 | 285 | RedisInstrumentor().instrument(tracer_provider=self.tracer_provider) |
286 | 286 |
|
@@ -470,7 +470,7 @@ def test_parent(self): |
470 | 470 | class TestAsyncRedisClusterInstrument(TestBase): |
471 | 471 | def setUp(self): |
472 | 472 | super().setUp() |
473 | | - self.redis_client = valkey.asyncio.cluster.RedisCluster( |
| 473 | + self.redis_client = redis.asyncio.cluster.RedisCluster( |
474 | 474 | host="localhost", port=7000 |
475 | 475 | ) |
476 | 476 | async_call(self.redis_client.flushall()) |
@@ -593,7 +593,7 @@ def test_parent(self): |
593 | 593 | class TestRedisDBIndexInstrument(TestBase): |
594 | 594 | def setUp(self): |
595 | 595 | super().setUp() |
596 | | - self.redis_client = valkey.Redis(port=6379, db=10) |
| 596 | + self.redis_client = redis.Redis(port=6379, db=10) |
597 | 597 | self.redis_client.flushall() |
598 | 598 | RedisInstrumentor().instrument(tracer_provider=self.tracer_provider) |
599 | 599 |
|
@@ -626,7 +626,7 @@ def test_get(self): |
626 | 626 | class TestRedisearchInstrument(TestBase): |
627 | 627 | def setUp(self): |
628 | 628 | super().setUp() |
629 | | - self.redis_client = valkey.Redis(port=6379) |
| 629 | + self.redis_client = redis.Redis(port=6379) |
630 | 630 | self.redis_client.flushall() |
631 | 631 | self.embedding_dim = 256 |
632 | 632 | RedisInstrumentor().instrument(tracer_provider=self.tracer_provider) |
|
0 commit comments