diff --git a/README.md b/README.md index 38a7961..67151f1 100644 --- a/README.md +++ b/README.md @@ -281,7 +281,7 @@ MiniOneRec supports multiple LLM providers for text enrichment tasks (e.g., user |----------|-----------------|------------------|----------------| | OpenAI | `"openai"` | — | `text-davinci-003` | | DeepSeek | `"deepseek"` | `https://api.deepseek.com` | `deepseek-chat` | -| [MiniMax](https://www.minimaxi.com) | `"minimax"` | `https://api.minimax.io/v1` | `MiniMax-M2.7`, `MiniMax-M2.5` | +| [MiniMax](https://www.minimaxi.com) | `"minimax"` | `https://api.minimax.io/v1` | `MiniMax-M3`, `MiniMax-M2.7` | **Example — using MiniMax:** @@ -291,7 +291,7 @@ api_info = { "api_key_list": ["your-minimax-api-key"], "base_url": "https://api.minimax.io/v1", # optional, this is the default } -get_res_batch("MiniMax-M2.7", prompt_list, max_tokens=512, api_info=api_info) +get_res_batch("MiniMax-M3", prompt_list, max_tokens=512, api_info=api_info) ``` --- diff --git a/tests/test_minimax_provider.py b/tests/test_minimax_provider.py index 17986ba..bed925d 100644 --- a/tests/test_minimax_provider.py +++ b/tests/test_minimax_provider.py @@ -349,7 +349,7 @@ def test_single_completion(self): "api_key_list": [self.api_key], } results = text2emb_utils.get_res_batch( - "MiniMax-M2.5", ["Say 'hello' in one word."], 10, api_info + "MiniMax-M3", ["Say 'hello' in one word."], 10, api_info ) self.assertEqual(len(results), 1) self.assertIn("hello", results[0].lower()) @@ -364,7 +364,7 @@ def test_batch_completion(self): "What is 2+2? Answer with just the number.", ] results = text2emb_utils.get_res_batch( - "MiniMax-M2.5", prompts, 64, api_info + "MiniMax-M3", prompts, 64, api_info ) self.assertEqual(len(results), 2) self.assertTrue(len(results[0]) > 0) @@ -382,7 +382,7 @@ def test_end_to_end_preference_prompt(self): review="Great sound quality and comfortable fit. Battery lasts all day." ) results = text2emb_utils.get_res_batch( - "MiniMax-M2.5", [prompt], 256, api_info + "MiniMax-M3", [prompt], 256, api_info ) self.assertEqual(len(results), 1) self.assertTrue(len(results[0]) > 0)