From 4f249439235eaedf9736114bb06782e1700662e7 Mon Sep 17 00:00:00 2001 From: FynnCX Date: Thu, 6 Mar 2025 00:07:17 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=94=AF=E6=8C=81SiliconFlow?= =?UTF-8?q?=E7=9A=84embedding=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- embedding_adapters.py | 46 +++++++++++++++++++++++++++++++++++++++++-- ui/config_tab.py | 10 ++++++++-- 2 files changed, 52 insertions(+), 4 deletions(-) diff --git a/embedding_adapters.py b/embedding_adapters.py index 6bc6ddb..cb00673 100644 --- a/embedding_adapters.py +++ b/embedding_adapters.py @@ -1,10 +1,12 @@ # embedding_adapters.py # -*- coding: utf-8 -*- import logging -import requests import traceback from typing import List -from langchain_openai import OpenAIEmbeddings, AzureOpenAIEmbeddings + +import requests +from langchain_openai import AzureOpenAIEmbeddings, OpenAIEmbeddings + def ensure_openai_base_url_has_v1(url: str) -> str: """ @@ -187,6 +189,44 @@ class GeminiEmbeddingAdapter(BaseEmbeddingAdapter): logging.error(f"Gemini embed_content parse error: {e}\n{traceback.format_exc()}") return [] +class SiliconFlowEmbeddingAdapter(BaseEmbeddingAdapter): + """ + 基于 SiliconFlow 的 embedding 适配器 + """ + def __init__(self, api_key: str, base_url: str, model_name: str): + # 自动为 base_url 添加 scheme（如果缺失） + if not base_url.startswith("http://") and not base_url.startswith("https://"): + base_url = "https://" + base_url + self.url = base_url if base_url else "https://api.siliconflow.cn/v1/embeddings" + + self.payload = { + "model": model_name, + "input": "Silicon flow embedding online: fast, affordable, and high-quality embedding services. come try it out!", + "encoding_format": "float" + } + self.headers = { + "Authorization": "Bearer {api_key}".format(api_key=api_key), + "Content-Type": "application/json" + } + + def embed_documents(self, texts: List[str]) -> List[List[float]]: + embeddings = [] + for text in texts: + self.payload["input"] = text + response = requests.post(self.url, json=self.payload, headers=self.headers) + result = response.json() + # 从返回数据中提取第一个 embedding + emb = result.get("data", [{}])[0].get("embedding", []) + embeddings.append(emb) + return embeddings + + def embed_query(self, query: str) -> List[float]: + self.payload["input"] = query + # print('SiliconFlowEmbeddingAdapter发送',self.payload) + response = requests.post(self.url, json=self.payload, headers=self.headers) + result = response.json() + return result.get("data", [{}])[0].get("embedding", []) + def create_embedding_adapter( interface_format: str, api_key: str, @@ -207,5 +247,7 @@ def create_embedding_adapter( return MLStudioEmbeddingAdapter(api_key, base_url, model_name) elif fmt == "gemini": return GeminiEmbeddingAdapter(api_key, model_name, base_url) + elif fmt == "siliconflow": + return SiliconFlowEmbeddingAdapter(api_key, base_url, model_name) else: raise ValueError(f"Unknown embedding interface_format: {interface_format}") diff --git a/ui/config_tab.py b/ui/config_tab.py index 0ca6f69..91f05a1 100644 --- a/ui/config_tab.py +++ b/ui/config_tab.py @@ -1,10 +1,13 @@ # ui/config_tab.py # -*- coding: utf-8 -*- -import customtkinter as ctk from tkinter import messagebox + +import customtkinter as ctk + from config_manager import load_config, save_config from tooltips import tooltips + def create_label_with_help(self, parent, label_text, tooltip_key, row, column, font=None, sticky="e", padx=5, pady=5): """ @@ -177,6 +180,9 @@ def build_embeddings_config_tab(self): elif new_value == "Gemini": self.embedding_url_var.set("https://generativelanguage.googleapis.com/v1beta/") 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): self.embeddings_config_tab.grid_rowconfigure(i, weight=0) @@ -191,7 +197,7 @@ def build_embeddings_config_tab(self): # 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)) - 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.grid(row=1, column=1, padx=5, pady=5, sticky="nsew")