mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
|
"""
|
|||
|
|
智谱 AI API 适配器
|
|||
|
|
|
|||
|
|
支持智谱 AI 的 GLM 系列模型，使用 OpenAI 兼容的接口格式。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from typing import TYPE_CHECKING
|
|||
|
|
|
|||
|
|
from .base import OpenAICompatAdapter, RequestData
|
|||
|
|
|
|||
|
|
if TYPE_CHECKING:
|
|||
|
|
from ..service import LLMModel
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ZhipuAdapter(OpenAICompatAdapter):
|
|||
|
|
"""智谱AI适配器 - 使用智谱AI专用的OpenAI兼容接口"""
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def api_type(self) -> str:
|
|||
|
|
return "zhipu"
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def supported_api_types(self) -> list[str]:
|
|||
|
|
return ["zhipu"]
|
|||
|
|
|
|||
|
|
def get_chat_endpoint(self) -> str:
|
|||
|
|
"""返回智谱AI聊天完成端点"""
|
|||
|
|
return "/api/paas/v4/chat/completions"
|
|||
|
|
|
|||
|
|
def get_embedding_endpoint(self) -> str:
|
|||
|
|
"""返回智谱AI嵌入端点"""
|
|||
|
|
return "/v4/embeddings"
|
|||
|
|
|
|||
|
|
def prepare_simple_request(
|
|||
|
|
self,
|
|||
|
|
model: "LLMModel",
|
|||
|
|
api_key: str,
|
|||
|
|
prompt: str,
|
|||
|
|
history: list[dict[str, str]] | None = None,
|
|||
|
|
) -> RequestData:
|
|||
|
|
"""准备简单文本生成请求 - 智谱AI优化实现"""
|
|||
|
|
url = self.get_api_url(model, self.get_chat_endpoint())
|
|||
|
|
headers = self.get_base_headers(api_key)
|
|||
|
|
|
|||
|
|
messages = []
|
|||
|
|
if history:
|
|||
|
|
messages.extend(history)
|
|||
|
|
messages.append({"role": "user", "content": prompt})
|
|||
|
|
|
|||
|
|
body = {
|
|||
|
|
"model": model.model_name,
|
|||
|
|
"messages": messages,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
body = self.apply_config_override(model, body)
|
|||
|
|
|
|||
|
|
return RequestData(url=url, headers=headers, body=body)
|