Skip to content

Commit d39c44f

Browse files
committed
test(oci): skip embed-light when region lacks model
1 parent c539611 commit d39c44f

1 file changed

Lines changed: 11 additions & 5 deletions

File tree

tests/test_oci_client.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from unittest.mock import MagicMock, mock_open, patch
2020

2121
import cohere
22+
from cohere.errors import NotFoundError
2223

2324
if "tokenizers" not in sys.modules:
2425
tokenizers_stub = types.ModuleType("tokenizers")
@@ -472,11 +473,16 @@ def test_embed_english_v3(self):
472473

473474
def test_embed_light_v3(self):
474475
"""Test embed-english-light-v3.0 model."""
475-
response = self.client.embed(
476-
model="embed-english-light-v3.0",
477-
texts=["Test"],
478-
input_type="search_document",
479-
)
476+
try:
477+
response = self.client.embed(
478+
model="embed-english-light-v3.0",
479+
texts=["Test"],
480+
input_type="search_document",
481+
)
482+
except NotFoundError as exc:
483+
if "embed-english-light-v3.0" in str(exc):
484+
self.skipTest("embed-english-light-v3.0 is not available in the current OCI region/profile")
485+
raise
480486
self.assertIsNotNone(response.embeddings)
481487
self.assertEqual(len(response.embeddings[0]), 384)
482488

0 commit comments

Comments
 (0)