mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-20 06:49:37 +08:00
Compare commits
2 Commits
c01e019286
...
8f84fdee94
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8f84fdee94 | ||
|
|
1ac73e81bc |
@ -121,6 +121,8 @@ class MinimaxVideoGenerationResponse(BaseModel):
|
||||
|
||||
|
||||
class MiniMaxChatModel(str, Enum):
|
||||
M2_7 = 'MiniMax-M2.7'
|
||||
M2_7_highspeed = 'MiniMax-M2.7-highspeed'
|
||||
M2_5 = 'MiniMax-M2.5'
|
||||
M2_5_highspeed = 'MiniMax-M2.5-highspeed'
|
||||
|
||||
|
||||
@ -452,7 +452,7 @@ class MinimaxChatNode(IO.ComfyNode):
|
||||
node_id="MinimaxChatNode",
|
||||
display_name="MiniMax Chat",
|
||||
category="api node/text/MiniMax",
|
||||
description="Generate text responses using MiniMax language models (MiniMax-M2.5).",
|
||||
description="Generate text responses using MiniMax language models (MiniMax-M2.7).",
|
||||
inputs=[
|
||||
IO.String.Input(
|
||||
"prompt",
|
||||
@ -463,7 +463,7 @@ class MinimaxChatNode(IO.ComfyNode):
|
||||
IO.Combo.Input(
|
||||
"model",
|
||||
options=MiniMaxChatModel,
|
||||
default=MiniMaxChatModel.M2_5.value,
|
||||
default=MiniMaxChatModel.M2_7.value,
|
||||
tooltip="The MiniMax model to use for text generation.",
|
||||
),
|
||||
IO.String.Input(
|
||||
@ -524,7 +524,7 @@ class MinimaxChatNode(IO.ComfyNode):
|
||||
async def execute(
|
||||
cls,
|
||||
prompt: str,
|
||||
model: str = MiniMaxChatModel.M2_5.value,
|
||||
model: str = MiniMaxChatModel.M2_7.value,
|
||||
system_prompt: Optional[str] = None,
|
||||
max_tokens: int = 4096,
|
||||
temperature: float = 0.7,
|
||||
|
||||
27
tests-unit/comfy_api_test/minimax_models_test.py
Normal file
27
tests-unit/comfy_api_test/minimax_models_test.py
Normal file
@ -0,0 +1,27 @@
|
||||
import pytest
|
||||
from comfy_api_nodes.apis.minimax import MiniMaxChatModel
|
||||
|
||||
|
||||
class TestMiniMaxChatModel:
|
||||
def test_m27_in_model_list(self):
|
||||
"""MiniMax-M2.7 should be available in the chat model enum."""
|
||||
assert MiniMaxChatModel.M2_7.value == 'MiniMax-M2.7'
|
||||
|
||||
def test_m27_highspeed_in_model_list(self):
|
||||
"""MiniMax-M2.7-highspeed should be available in the chat model enum."""
|
||||
assert MiniMaxChatModel.M2_7_highspeed.value == 'MiniMax-M2.7-highspeed'
|
||||
|
||||
def test_m27_is_first_in_enum(self):
|
||||
"""M2.7 should appear before older models in the enum."""
|
||||
members = list(MiniMaxChatModel)
|
||||
assert members[0] == MiniMaxChatModel.M2_7
|
||||
assert members[1] == MiniMaxChatModel.M2_7_highspeed
|
||||
|
||||
def test_legacy_models_still_available(self):
|
||||
"""Previous M2.5 models should still be available."""
|
||||
assert MiniMaxChatModel.M2_5.value == 'MiniMax-M2.5'
|
||||
assert MiniMaxChatModel.M2_5_highspeed.value == 'MiniMax-M2.5-highspeed'
|
||||
|
||||
def test_total_model_count(self):
|
||||
"""Should have 4 chat models total (M2.7, M2.7-highspeed, M2.5, M2.5-highspeed)."""
|
||||
assert len(MiniMaxChatModel) == 4
|
||||
Loading…
Reference in New Issue
Block a user