Skip to content

Commit 61e2ca1

Browse files
hotfix adding finetuned bgm (#516)
1 parent aba42f1 commit 61e2ca1

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

aixplain/enums/embedding_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ class EmbeddingModel(Enum):
2626
SNOWFLAKE_ARCTIC_EMBED_L_V2_0 = "678a4f8547f687504744960a"
2727
JINA_CLIP_V2_MULTIMODAL = "67c5f705d8f6a65d6f74d732"
2828
MULTILINGUAL_E5_LARGE = "67efd0772a0a850afa045af3"
29-
BGE_M3 = "67f401032a0a850afa045b19"
29+
BGE_M3 = "67efd4f92a0a850afa045af7"
30+
AIXPLAIN_LEGAL_EMBEDDINGS = "681254b668e47e7844c1f15a"
3031

3132

3233

tests/functional/model/run_model_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ def test_run_async():
6565
pytest.param(EmbeddingModel.SNOWFLAKE_ARCTIC_EMBED_L_V2_0, id="Snowflake Arctic Embed L v2.0"),
6666
pytest.param(EmbeddingModel.MULTILINGUAL_E5_LARGE, id="Multilingual E5 Large"),
6767
pytest.param(EmbeddingModel.BGE_M3, id="BGE M3"),
68+
pytest.param(EmbeddingModel.AIXPLAIN_LEGAL_EMBEDDINGS, id="aiXplain Legal Embeddings"),
6869
],
6970
)
7071
def test_index_model(embedding_model):
@@ -112,6 +113,7 @@ def test_index_model(embedding_model):
112113
pytest.param(EmbeddingModel.JINA_CLIP_V2_MULTIMODAL, id="Jina Clip v2 Multimodal"),
113114
pytest.param(EmbeddingModel.MULTILINGUAL_E5_LARGE, id="Multilingual E5 Large"),
114115
pytest.param(EmbeddingModel.BGE_M3, id="BGE M3"),
116+
pytest.param(EmbeddingModel.AIXPLAIN_LEGAL_EMBEDDINGS, id="aiXplain Legal Embeddings"),
115117
],
116118
)
117119
def test_index_model_with_filter(embedding_model):

0 commit comments

Comments
 (0)