85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
|
"""
|
||
|
|
MiniMax适配器
|
||
|
|
"""
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
from ..base_adapter import BaseLLMAdapter
|
||
|
|
from ..types import LLMConfig, LLMRequest, LLMResponse, LLMError
|
||
|
|
|
||
|
|
|
||
|
|
class MinimaxAdapter(BaseLLMAdapter):
|
||
|
|
"""MiniMax API适配器"""
|
||
|
|
|
||
|
|
def __init__(self, config: LLMConfig):
|
||
|
|
super().__init__(config)
|
||
|
|
self._base_url = config.base_url or "https://api.minimax.chat/v1"
|
||
|
|
|
||
|
|
async def _do_complete(self, request: LLMRequest) -> LLMResponse:
|
||
|
|
"""执行实际的API调用"""
|
||
|
|
url = f"{self._base_url}/text/chatcompletion_v2"
|
||
|
|
|
||
|
|
messages = [{"role": m.role, "content": m.content} for m in request.messages]
|
||
|
|
|
||
|
|
payload = {
|
||
|
|
"model": self.config.model or "abab6.5-chat",
|
||
|
|
"messages": messages,
|
||
|
|
"temperature": request.temperature or self.config.temperature,
|
||
|
|
"top_p": request.top_p or self.config.top_p,
|
||
|
|
}
|
||
|
|
|
||
|
|
if request.max_tokens or self.config.max_tokens:
|
||
|
|
payload["max_tokens"] = request.max_tokens or self.config.max_tokens
|
||
|
|
|
||
|
|
headers = {
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
"Authorization": f"Bearer {self.config.api_key}",
|
||
|
|
}
|
||
|
|
|
||
|
|
async with httpx.AsyncClient(timeout=self.config.timeout) as client:
|
||
|
|
response = await client.post(url, json=payload, headers=headers)
|
||
|
|
|
||
|
|
if response.status_code != 200:
|
||
|
|
raise LLMError(
|
||
|
|
f"MiniMax API错误: {response.text}",
|
||
|
|
provider="minimax",
|
||
|
|
status_code=response.status_code
|
||
|
|
)
|
||
|
|
|
||
|
|
data = response.json()
|
||
|
|
|
||
|
|
if data.get("base_resp", {}).get("status_code") != 0:
|
||
|
|
raise LLMError(
|
||
|
|
f"MiniMax API错误: {data.get('base_resp', {}).get('status_msg', '未知错误')}",
|
||
|
|
provider="minimax"
|
||
|
|
)
|
||
|
|
|
||
|
|
choices = data.get("choices", [])
|
||
|
|
if not choices:
|
||
|
|
raise LLMError("MiniMax API返回空响应", provider="minimax")
|
||
|
|
|
||
|
|
return LLMResponse(
|
||
|
|
content=choices[0].get("message", {}).get("content", ""),
|
||
|
|
model=self.config.model or "abab6.5-chat",
|
||
|
|
usage=data.get("usage"),
|
||
|
|
finish_reason=choices[0].get("finish_reason")
|
||
|
|
)
|
||
|
|
|
||
|
|
async def validate_config(self) -> bool:
|
||
|
|
"""验证配置是否有效"""
|
||
|
|
try:
|
||
|
|
test_request = LLMRequest(
|
||
|
|
messages=[{"role": "user", "content": "Hi"}],
|
||
|
|
max_tokens=10
|
||
|
|
)
|
||
|
|
await self._do_complete(test_request)
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def get_provider(self) -> str:
|
||
|
|
return "minimax"
|
||
|
|
|
||
|
|
def get_model(self) -> str:
|
||
|
|
return self.config.model or "abab6.5-chat"
|
||
|
|
|