Skip to content

Commit 2e685e3

Browse files
rbs333abrookins
authored andcommitted
Semantic router testing name generation (#327)
1 parent 067c83e commit 2e685e3

File tree

3 files changed

+15
-10
lines changed

3 files changed

+15
-10
lines changed

schemas/semantic_router.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: test-router-01JSHK4MJ79HH51PS6WEK6M9MF
1+
name: test-router
22
routes:
33
- name: greeting
44
references:

tests/integration/test_semantic_router.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,7 @@ def test_from_dict(semantic_router):
200200

201201
def test_to_yaml(semantic_router):
202202
yaml_file = str(get_base_path().joinpath("../../schemas/semantic_router.yaml"))
203+
semantic_router.name = "test-router"
203204
semantic_router.to_yaml(yaml_file, overwrite=True)
204205
assert pathlib.Path(yaml_file).exists()
205206

tests/integration/test_threshold_optimizer.py

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
if sys.version_info.major == 3 and sys.version_info.minor < 10:
66
pytest.skip("Test requires Python 3.10 or higher", allow_module_level=True)
77

8+
from redis import Redis
9+
810
from redisvl.extensions.cache.llm import SemanticCache
911
from redisvl.extensions.router import Route, SemanticRouter
1012
from redisvl.extensions.router.schema import RoutingConfig
@@ -35,13 +37,13 @@ def routes():
3537

3638

3739
@pytest.fixture
38-
def semantic_router(client, routes, hf_vectorizer):
40+
def semantic_router(redis_url, routes, hf_vectorizer):
3941
router = SemanticRouter(
4042
name="test-router",
4143
routes=routes,
4244
vectorizer=hf_vectorizer,
4345
routing_config=RoutingConfig(max_k=2),
44-
redis_client=client,
46+
redis_url=redis_url,
4547
overwrite=False,
4648
)
4749
yield router
@@ -87,9 +89,10 @@ def test_data_optimization():
8789

8890

8991
def test_routes_different_distance_thresholds_optimizer_default(
90-
semantic_router, routes, redis_url, test_data_optimization, hf_vectorizer
92+
routes, redis_url, test_data_optimization, hf_vectorizer
9193
):
92-
redis_version = semantic_router._index.client.info()["redis_version"]
94+
redis = Redis.from_url(redis_url)
95+
redis_version = redis.info()["redis_version"]
9396
if not compare_versions(redis_version, "7.0.0"):
9497
pytest.skip("Not using a late enough version of Redis")
9598

@@ -121,10 +124,10 @@ def test_routes_different_distance_thresholds_optimizer_default(
121124

122125

123126
def test_routes_different_distance_thresholds_optimizer_precision(
124-
semantic_router, routes, redis_url, test_data_optimization, hf_vectorizer
127+
routes, redis_url, test_data_optimization, hf_vectorizer
125128
):
126-
127-
redis_version = semantic_router._index.client.info()["redis_version"]
129+
redis = Redis.from_url(redis_url)
130+
redis_version = redis.info()["redis_version"]
128131
if not compare_versions(redis_version, "7.0.0"):
129132
pytest.skip("Not using a late enough version of Redis")
130133

@@ -158,9 +161,10 @@ def test_routes_different_distance_thresholds_optimizer_precision(
158161

159162

160163
def test_routes_different_distance_thresholds_optimizer_recall(
161-
semantic_router, routes, redis_url, test_data_optimization, hf_vectorizer
164+
routes, redis_url, test_data_optimization, hf_vectorizer, client
162165
):
163-
redis_version = semantic_router._index.client.info()["redis_version"]
166+
redis = Redis.from_url(redis_url)
167+
redis_version = redis.info()["redis_version"]
164168
if not compare_versions(redis_version, "7.0.0"):
165169
pytest.skip("Not using a late enough version of Redis")
166170

0 commit comments

Comments
 (0)