From 44c8c638994282c1821ec1a86df58a6ba5892af4 Mon Sep 17 00:00:00 2001 From: jiangnan <1394485448@qq.com> Date: Mon, 16 Mar 2026 17:42:35 +0800 Subject: [PATCH] fix: only pass dimensions param when explicitly configured Models like bge-m3 don't support the dimensions parameter in the embedding API, causing HTTP 400 errors. Previously dimensions was always sent with a default value of 1024, even when the user never configured it. Now dimensions is only included in the request when embedding_dimensions is explicitly set in provider config. Closes #6421 Signed-off-by: JiangNan <1394485448@qq.com> --- .../provider/sources/openai_embedding_source.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/astrbot/core/provider/sources/openai_embedding_source.py b/astrbot/core/provider/sources/openai_embedding_source.py index 8bf92ef4d..57b409fed 100644 --- a/astrbot/core/provider/sources/openai_embedding_source.py +++ b/astrbot/core/provider/sources/openai_embedding_source.py @@ -40,22 +40,31 @@ class OpenAIEmbeddingProvider(EmbeddingProvider): async def get_embedding(self, text: str) -> list[float]: """获取文本的嵌入""" + kwargs = self._embedding_kwargs() embedding = await self.client.embeddings.create( input=text, model=self.model, - dimensions=self.get_dim(), + **kwargs, ) return embedding.data[0].embedding async def get_embeddings(self, text: list[str]) -> list[list[float]]: """批量获取文本的嵌入""" + kwargs = self._embedding_kwargs() embeddings = await self.client.embeddings.create( input=text, model=self.model, - dimensions=self.get_dim(), + **kwargs, ) return [item.embedding for item in embeddings.data] + def _embedding_kwargs(self) -> dict: + """构建嵌入请求的可选参数""" + kwargs = {} + if "embedding_dimensions" in self.provider_config: + kwargs["dimensions"] = int(self.provider_config["embedding_dimensions"]) + return kwargs + def get_dim(self) -> int: """获取向量的维度""" return int(self.provider_config.get("embedding_dimensions", 1024))