fix: ensure embedding dimensions are returned as integers in providers (#4547)
* fix: ensure embedding dimensions are returned as integers in providers * chore: ruff format
This commit is contained in:
@@ -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))
|
||||
|
||||
@@ -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))
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user