From cb3825bb00827dd764f6cd8f8048130446c31eb6 Mon Sep 17 00:00:00 2001
From: Soulter <37870767+Soulter@users.noreply.github.com>
Date: Sun, 18 Jan 2026 17:09:25 +0800
Subject: [PATCH] fix: ensure embedding dimensions are returned as integers in
 providers (#4547)

* fix: ensure embedding dimensions are returned as integers in providers

* chore: ruff format
---
 astrbot/core/provider/sources/gemini_embedding_source.py | 2 +-
 astrbot/core/provider/sources/openai_embedding_source.py | 2 +-
 astrbot/core/star/register/star_handler.py               | 4 +---
 3 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/astrbot/core/provider/sources/gemini_embedding_source.py b/astrbot/core/provider/sources/gemini_embedding_source.py
index 146b50a4e..01046bebb 100644
--- a/astrbot/core/provider/sources/gemini_embedding_source.py
+++ b/astrbot/core/provider/sources/gemini_embedding_source.py
@@ -68,4 +68,4 @@ class GeminiEmbeddingProvider(EmbeddingProvider):
 
     def get_dim(self) -> int:
         """获取向量的维度"""
-        return self.provider_config.get("embedding_dimensions", 768)
+        return int(self.provider_config.get("embedding_dimensions", 768))
diff --git a/astrbot/core/provider/sources/openai_embedding_source.py b/astrbot/core/provider/sources/openai_embedding_source.py
index c9e03d7af..ad20dd3df 100644
--- a/astrbot/core/provider/sources/openai_embedding_source.py
+++ b/astrbot/core/provider/sources/openai_embedding_source.py
@@ -37,4 +37,4 @@ class OpenAIEmbeddingProvider(EmbeddingProvider):
 
     def get_dim(self) -> int:
         """获取向量的维度"""
-        return self.provider_config.get("embedding_dimensions", 1024)
+        return int(self.provider_config.get("embedding_dimensions", 1024))
diff --git a/astrbot/core/star/register/star_handler.py b/astrbot/core/star/register/star_handler.py
index 779144b40..eefbcedb7 100644
--- a/astrbot/core/star/register/star_handler.py
+++ b/astrbot/core/star/register/star_handler.py
@@ -452,9 +452,7 @@ def register_on_llm_tool_respond(**kwargs):
     """
 
     def decorator(awaitable):
-        _ = get_handler_or_create(
-            awaitable, EventType.OnLLMToolRespondEvent, **kwargs
-        )
+        _ = get_handler_or_create(awaitable, EventType.OnLLMToolRespondEvent, **kwargs)
         return awaitable
 
     return decorator