lizhilun/AI_NovelGenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加对阿里云百炼的支持

Browse Source 
添加了对阿里云百炼的支持。接口格式里增加了“阿里云百炼”选项，并设置默认模型为“qwen-plus”。已经过测试，可以使用。
此外，还添加了DeepSeek的默认模型“deepseek-chat”和OpenAI的默认模型“gpt-4o-mini”。
This commit is contained in:
Karasukaigan
2025-02-18 00:46:31 +08:00
parent e965b3c9af
commit 0a24217eb5
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
llm_adapters.py
+2
View File
@@ -291,5 +291,7 @@ def create_llm_adapter(
elif fmt == "gemini": elif fmt == "gemini":
# base_url 对 Gemini 暂无用处，可忽略 # base_url 对 Gemini 暂无用处，可忽略
return GeminiAdapter(api_key, model_name, max_tokens, temperature, timeout) return GeminiAdapter(api_key, model_name, max_tokens, temperature, timeout)
elif fmt == "阿里云百炼":
return OpenAIAdapter(api_key, base_url, model_name, max_tokens, temperature, timeout)
else: else:
raise ValueError(f"Unknown interface_format: {interface_format}") raise ValueError(f"Unknown interface_format: {interface_format}")
ui/config_tab.py
+6 -1
View File
@@ -76,14 +76,19 @@ def build_ai_config_tab(self):
self.base_url_var.set("http://localhost:1234/v1") self.base_url_var.set("http://localhost:1234/v1")
elif new_value == "OpenAI": elif new_value == "OpenAI":
self.base_url_var.set("https://api.openai.com/v1") self.base_url_var.set("https://api.openai.com/v1")
self.model_name_var.set("gpt-4o-mini")
elif new_value == "Azure OpenAI": elif new_value == "Azure OpenAI":
self.base_url_var.set("https://[az].openai.azure.com/openai/deployments/[model]/chat/completions?api-version=2024-08-01-preview") self.base_url_var.set("https://[az].openai.azure.com/openai/deployments/[model]/chat/completions?api-version=2024-08-01-preview")
elif new_value == "DeepSeek": elif new_value == "DeepSeek":
self.base_url_var.set("https://api.deepseek.com/v1") self.base_url_var.set("https://api.deepseek.com/v1")
self.model_name_var.set("deepseek-chat")
elif new_value == "Gemini": elif new_value == "Gemini":
self.base_url_var.set("") self.base_url_var.set("")
elif new_value == "Azure AI": elif new_value == "Azure AI":
self.base_url_var.set("https://<your-endpoint>.services.ai.azure.com/models/chat/completions?api-version=2024-05-01-preview") self.base_url_var.set("https://<your-endpoint>.services.ai.azure.com/models/chat/completions?api-version=2024-05-01-preview")
elif new_value == "阿里云百炼":
self.base_url_var.set("https://dashscope.aliyuncs.com/compatible-mode/v1")
self.model_name_var.set("qwen-plus")
for i in range(7): for i in range(7):
self.ai_config_tab.grid_rowconfigure(i, weight=0) self.ai_config_tab.grid_rowconfigure(i, weight=0)
@@ -103,7 +108,7 @@ def build_ai_config_tab(self):
# 3) 接口格式 # 3) 接口格式
create_label_with_help(self, parent=self.ai_config_tab, label_text="LLM 接口格式:", tooltip_key="interface_format", row=2, column=0, font=("Microsoft YaHei", 12)) create_label_with_help(self, parent=self.ai_config_tab, label_text="LLM 接口格式:", tooltip_key="interface_format", row=2, column=0, font=("Microsoft YaHei", 12))
interface_options = ["DeepSeek", "OpenAI", "Azure OpenAI", "Azure AI", "Ollama", "ML Studio", "Gemini"] interface_options = ["DeepSeek", "阿里云百炼", "OpenAI", "Azure OpenAI", "Azure AI", "Ollama", "ML Studio", "Gemini"]
interface_dropdown = ctk.CTkOptionMenu(self.ai_config_tab, values=interface_options, variable=self.interface_format_var, command=on_interface_format_changed, font=("Microsoft YaHei", 12)) interface_dropdown = ctk.CTkOptionMenu(self.ai_config_tab, values=interface_options, variable=self.interface_format_var, command=on_interface_format_changed, font=("Microsoft YaHei", 12))
interface_dropdown.grid(row=2, column=1, padx=5, pady=5, columnspan=2, sticky="nsew") interface_dropdown.grid(row=2, column=1, padx=5, pady=5, columnspan=2, sticky="nsew")