Unverified Commit 76c52300 authored by takatost's avatar takatost Committed by GitHub

feat: abab6-chat supported (#2184)

parent dda32c68
...@@ -23,7 +23,7 @@ parameter_rules: ...@@ -23,7 +23,7 @@ parameter_rules:
- name: frequency_penalty - name: frequency_penalty
use_template: frequency_penalty use_template: frequency_penalty
pricing: pricing:
input: '0.00' input: '0.015'
output: '0.015' output: '0.015'
unit: '0.001' unit: '0.001'
currency: RMB currency: RMB
...@@ -36,7 +36,7 @@ parameter_rules: ...@@ -36,7 +36,7 @@ parameter_rules:
en_US: Enable Web Search en_US: Enable Web Search
zh_Hans: 开启网页搜索 zh_Hans: 开启网页搜索
pricing: pricing:
input: '0.00' input: '0.015'
output: '0.015' output: '0.015'
unit: '0.001' unit: '0.001'
currency: RMB currency: RMB
...@@ -29,7 +29,7 @@ parameter_rules: ...@@ -29,7 +29,7 @@ parameter_rules:
- name: frequency_penalty - name: frequency_penalty
use_template: frequency_penalty use_template: frequency_penalty
pricing: pricing:
input: '0.00' input: '0.005'
output: '0.005' output: '0.005'
unit: '0.001' unit: '0.001'
currency: RMB currency: RMB
model: abab6-chat
label:
en_US: Abab6-Chat
model_type: llm
features:
- agent-thought
model_properties:
mode: chat
context_size: 32768
parameter_rules:
- name: temperature
use_template: temperature
min: 0.01
max: 1
default: 0.1
- name: top_p
use_template: top_p
min: 0.01
max: 1
default: 0.9
- name: max_tokens
use_template: max_tokens
required: true
default: 2048
min: 1
max: 32768
- name: presence_penalty
use_template: presence_penalty
- name: frequency_penalty
use_template: frequency_penalty
pricing:
input: '0.1'
output: '0.1'
unit: '0.001'
currency: RMB
...@@ -22,9 +22,6 @@ class MinimaxChatCompletionPro(object): ...@@ -22,9 +22,6 @@ class MinimaxChatCompletionPro(object):
""" """
generate chat completion generate chat completion
""" """
if model not in ['abab5.5-chat', 'abab5.5s-chat']:
raise BadRequestError(f'Invalid model: {model}')
if not api_key or not group_id: if not api_key or not group_id:
raise InvalidAPIKeyError('Invalid API key or group ID') raise InvalidAPIKeyError('Invalid API key or group ID')
......
from typing import Generator, List, Optional, Union from typing import Generator, List
from core.model_runtime.entities.llm_entities import LLMMode, LLMResult, LLMResultChunk, LLMResultChunkDelta, LLMUsage from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import (AssistantPromptMessage, PromptMessage, PromptMessageTool, from core.model_runtime.entities.message_entities import (AssistantPromptMessage, PromptMessage, PromptMessageTool,
SystemPromptMessage, UserPromptMessage) SystemPromptMessage, UserPromptMessage)
from core.model_runtime.entities.model_entities import AIModelEntity, FetchFrom, ModelType, ParameterRule, ParameterType
from core.model_runtime.errors.invoke import (InvokeAuthorizationError, InvokeBadRequestError, InvokeConnectionError, from core.model_runtime.errors.invoke import (InvokeAuthorizationError, InvokeBadRequestError, InvokeConnectionError,
InvokeError, InvokeRateLimitError, InvokeServerUnavailableError) InvokeError, InvokeRateLimitError, InvokeServerUnavailableError)
from core.model_runtime.errors.validate import CredentialsValidateFailedError from core.model_runtime.errors.validate import CredentialsValidateFailedError
...@@ -18,6 +17,7 @@ from core.model_runtime.model_providers.minimax.llm.types import MinimaxMessage ...@@ -18,6 +17,7 @@ from core.model_runtime.model_providers.minimax.llm.types import MinimaxMessage
class MinimaxLargeLanguageModel(LargeLanguageModel): class MinimaxLargeLanguageModel(LargeLanguageModel):
model_apis = { model_apis = {
'abab6-chat': MinimaxChatCompletionPro,
'abab5.5s-chat': MinimaxChatCompletionPro, 'abab5.5s-chat': MinimaxChatCompletionPro,
'abab5.5-chat': MinimaxChatCompletionPro, 'abab5.5-chat': MinimaxChatCompletionPro,
'abab5-chat': MinimaxChatCompletion 'abab5-chat': MinimaxChatCompletion
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment