shen/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-02-07 08:42:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2660 from cowagent/fix-zhipuai-api-base-support

Browse Source 
fix: 支持智谱AI自定义API base URL配置
This commit is contained in:
vision
2026-02-05 19:18:49 +08:00
committed by GitHub
parent 3958b6aae1 9ad3968084
commit 5b094e1097
2 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
models/zhipuai/zhipu_ai_image.py
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:

11
models/zhipuai/zhipuai_bot.py
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