move default test to rerank and fuyu model names

langchain-ai · May 17, 2024 · 6c5fcd8 · 6c5fcd8
1 parent 598a427
commit 6c5fcd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/ai-endpoints/tests/integration_tests/conftest.py b/libs/ai-endpoints/tests/integration_tests/conftest.py
@@ -56,7 +56,7 @@ def get_all_models() -> List[Model]:
         metafunc.parametrize("chat_model", models, ids=models)
 
     if "rerank_model" in metafunc.fixturenames:
-        models = ["ai-rerank-qa-mistral-4b"]
+        models = ["nv-rerank-qa-mistral-4b:1"]
         if model := metafunc.config.getoption("rerank_model_id"):
             models = [model]
         # nim-mode reranking does not support model listing via /v1/models endpoint
@@ -74,7 +74,7 @@ def get_all_models() -> List[Model]:
         metafunc.parametrize("rerank_model", models, ids=models)
 
     if "image_in_model" in metafunc.fixturenames:
-        models = ["ai-fuyu-8b"]
+        models = ["adept/fuyu-8b"]
         if metafunc.config.getoption("all_models"):
             models = [
                 model.id for model in get_all_models() if model.model_type == "image_in"