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

Update config_tab.py

Browse Source
This commit is contained in:
CNlaojing
2025-03-13 05:52:49 +08:00
committed by GitHub
parent 78aa58eca2
commit 1eb6ffc535
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ui/config_tab.py
+6 -2
View File
@@ -1,7 +1,7 @@
# ui/config_tab.py # ui/config_tab.py
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import customtkinter as ctk
from tkinter import messagebox from tkinter import messagebox
import customtkinter as ctk
from config_manager import load_config, save_config from config_manager import load_config, save_config
from tooltips import tooltips from tooltips import tooltips
@@ -177,6 +177,10 @@ def build_embeddings_config_tab(self):
elif new_value == "Gemini": elif new_value == "Gemini":
self.embedding_url_var.set("https://generativelanguage.googleapis.com/v1beta/") self.embedding_url_var.set("https://generativelanguage.googleapis.com/v1beta/")
self.embedding_model_name_var.set("models/text-embedding-004") self.embedding_model_name_var.set("models/text-embedding-004")
elif new_value == "SiliconFlow":
self.embedding_url_var.set("https://api.siliconflow.cn/v1/embeddings")
self.embedding_model_name_var.set("BAAI/bge-m3")
for i in range(5): for i in range(5):
self.embeddings_config_tab.grid_rowconfigure(i, weight=0) self.embeddings_config_tab.grid_rowconfigure(i, weight=0)
@@ -191,7 +195,7 @@ def build_embeddings_config_tab(self):
# 2) Embedding 接口格式 # 2) Embedding 接口格式
create_label_with_help(self, parent=self.embeddings_config_tab, label_text="Embedding 接口格式:", tooltip_key="embedding_interface_format", row=1, column=0, font=("Microsoft YaHei", 12)) create_label_with_help(self, parent=self.embeddings_config_tab, label_text="Embedding 接口格式:", tooltip_key="embedding_interface_format", row=1, column=0, font=("Microsoft YaHei", 12))
emb_interface_options = ["DeepSeek", "OpenAI", "Azure OpenAI", "Gemini", "Ollama", "ML Studio"] emb_interface_options = ["DeepSeek", "OpenAI", "Azure OpenAI", "Gemini", "Ollama", "ML Studio","SiliconFlow"]
emb_interface_dropdown = ctk.CTkOptionMenu(self.embeddings_config_tab, values=emb_interface_options, variable=self.embedding_interface_format_var, command=on_embedding_interface_changed, font=("Microsoft YaHei", 12)) emb_interface_dropdown = ctk.CTkOptionMenu(self.embeddings_config_tab, values=emb_interface_options, variable=self.embedding_interface_format_var, command=on_embedding_interface_changed, font=("Microsoft YaHei", 12))
emb_interface_dropdown.grid(row=1, column=1, padx=5, pady=5, sticky="nsew") emb_interface_dropdown.grid(row=1, column=1, padx=5, pady=5, sticky="nsew")