Skip to content

Commit 6ec8780

Browse files
committed
fix(test): use ONNX-compatible model for SemanticRouterIntegrationTest
The test was using sentence-transformers/all-mpnet-base-v2, which no longer has ONNX format on HuggingFace (only SafeTensors). Changed to sentence-transformers/all-MiniLM-L6-v2 which has proper ONNX support. This fix resolves CI failures where model download failed with: "Model uses SafeTensors format which is not supported. ONNX format is required."
1 parent 481d043 commit 6ec8780

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

core/src/test/java/com/redis/vl/extensions/router/SemanticRouterIntegrationTest.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,9 @@ class SemanticRouterIntegrationTest extends BaseIntegrationTest {
2323

2424
@BeforeEach
2525
void setUp() {
26-
// Use SentenceTransformersVectorizer matching Python test fixture (conftest.py:210)
27-
vectorizer = new SentenceTransformersVectorizer("sentence-transformers/all-mpnet-base-v2");
26+
// Use SentenceTransformersVectorizer with ONNX-compatible model
27+
// Note: all-mpnet-base-v2 no longer has ONNX format on HuggingFace (only SafeTensors)
28+
vectorizer = new SentenceTransformersVectorizer("sentence-transformers/all-MiniLM-L6-v2");
2829

2930
// Test routes from Python fixture (line 24-38)
3031
testRoutes =

0 commit comments

Comments
 (0)