Underground-Digital/Workflow-Engine

1import os2 3import pytest4 5from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult6from core.model_runtime.errors.validate import CredentialsValidateFailedError7from core.model_runtime.model_providers.minimax.text_embedding.text_embedding import MinimaxTextEmbeddingModel8 9 10def test_validate_credentials():11    model = MinimaxTextEmbeddingModel()12 13    with pytest.raises(CredentialsValidateFailedError):14        model.validate_credentials(15            model="embo-01",16            credentials={"minimax_api_key": "invalid_key", "minimax_group_id": os.environ.get("MINIMAX_GROUP_ID")},17        )18 19    model.validate_credentials(20        model="embo-01",21        credentials={22            "minimax_api_key": os.environ.get("MINIMAX_API_KEY"),23            "minimax_group_id": os.environ.get("MINIMAX_GROUP_ID"),24        },25    )26 27 28def test_invoke_model():29    model = MinimaxTextEmbeddingModel()30 31    result = model.invoke(32        model="embo-01",33        credentials={34            "minimax_api_key": os.environ.get("MINIMAX_API_KEY"),35            "minimax_group_id": os.environ.get("MINIMAX_GROUP_ID"),36        },37        texts=["hello", "world"],38        user="abc-123",39    )40 41    assert isinstance(result, TextEmbeddingResult)42    assert len(result.embeddings) == 243    assert result.usage.total_tokens == 1644 45 46def test_get_num_tokens():47    model = MinimaxTextEmbeddingModel()48 49    num_tokens = model.get_num_tokens(50        model="embo-01",51        credentials={52            "minimax_api_key": os.environ.get("MINIMAX_API_KEY"),53            "minimax_group_id": os.environ.get("MINIMAX_GROUP_ID"),54        },55        texts=["hello", "world"],56    )57 58    assert num_tokens == 259