mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-23 16:29:16 +08:00
fix: 支持智谱AI自定义API base URL配置
- 修复 ZhipuAiClient 初始化时未传入 base_url 参数的问题 - 使配置文件中的 zhipu_ai_api_base 配置项生效 - 支持智谱国际版(z.ai)等自定义API端点 - 同时修复对话和图片生成功能 - 添加日志输出便于确认使用的API地址 Fixes #2659
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user