fix: 支持智谱AI自定义API base URL配置

- 修复 ZhipuAiClient 初始化时未传入 base_url 参数的问题
- 使配置文件中的 zhipu_ai_api_base 配置项生效
- 支持智谱国际版(z.ai)等自定义API端点
- 同时修复对话和图片生成功能
- 添加日志输出便于确认使用的API地址

Fixes #2659
This commit is contained in:
cowagent
2026-02-05 19:06:46 +08:00
parent 3958b6aae1
commit 9ad3968084
2 changed files with 20 additions and 2 deletions

View File

@@ -7,7 +7,16 @@ from config import conf
class ZhipuAIImage(object):
def __init__(self):
from zai import ZhipuAiClient
self.client = ZhipuAiClient(api_key=conf().get("zhipu_ai_api_key"))
# 初始化客户端,支持自定义 API base URL例如智谱国际版 z.ai
api_key = conf().get("zhipu_ai_api_key")
api_base = conf().get("zhipu_ai_api_base")
if api_base:
self.client = ZhipuAiClient(api_key=api_key, base_url=api_base)
logger.info(f"[ZHIPU_AI_IMAGE] 使用自定义 API Base URL: {api_base}")
else:
self.client = ZhipuAiClient(api_key=api_key)
logger.info("[ZHIPU_AI_IMAGE] 使用默认 API Base URL")
def create_img(self, query, retry_count=0, api_key=None, api_base=None):
try:

View File

@@ -24,7 +24,16 @@ class ZHIPUAIBot(Bot, ZhipuAIImage):
"temperature": conf().get("temperature", 0.9), # 值在(0,1)之间(智谱AI 的温度不能取 0 或者 1)
"top_p": conf().get("top_p", 0.7), # 值在(0,1)之间(智谱AI 的 top_p 不能取 0 或者 1)
}
self.client = ZhipuAiClient(api_key=conf().get("zhipu_ai_api_key"))
# 初始化客户端,支持自定义 API base URL例如智谱国际版 z.ai
api_key = conf().get("zhipu_ai_api_key")
api_base = conf().get("zhipu_ai_api_base")
if api_base:
self.client = ZhipuAiClient(api_key=api_key, base_url=api_base)
logger.info(f"[ZHIPU_AI] 使用自定义 API Base URL: {api_base}")
else:
self.client = ZhipuAiClient(api_key=api_key)
logger.info("[ZHIPU_AI] 使用默认 API Base URL")
def reply(self, query, context=None):
# acquire reply content