|
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