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 @@ async def get_embeddings(self, text: list[str]) -> list[list[float]]: 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 @@ async def get_embeddings(self, text: list[str]) -> list[list[float]]: 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 @@ async def test(self, event: AstrMessageEvent, tool: FunctionTool, tool_args: dic """ def decorator(awaitable): - _ = get_handler_or_create( - awaitable, EventType.OnLLMToolRespondEvent, **kwargs - ) + _ = get_handler_or_create(awaitable, EventType.OnLLMToolRespondEvent, **kwargs) return awaitable return decorator