Skip to content

Commit 9e11804

Browse files
authored
🐛 The embedding model occasionally has a dimension of 0
2 parents 27e251d + fb2857a commit 9e11804

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sdk/nexent/core/models/embedding_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def get_embeddings(self, inputs: Union[str, List[str]], with_metadata: bool = Fa
4040
pass
4141

4242
@abstractmethod
43-
async def dimension_check(self, timeout: float = 5.0) -> List[List[float]]:
43+
async def dimension_check(self, timeout: float = 30.0) -> List[List[float]]:
4444
"""
4545
Test the connectivity to the embedding API, supporting timeout detection.
4646
@@ -183,7 +183,7 @@ def get_multimodal_embeddings(self, inputs: List[Dict[str, str]], with_metadata:
183183
embeddings = [item["embedding"] for item in response["data"]]
184184
return embeddings
185185

186-
async def dimension_check(self, timeout: float = 5.0) -> List[List[float]]:
186+
async def dimension_check(self, timeout: float = 30.0) -> List[List[float]]:
187187
try:
188188
# Create a simple test input
189189
test_input = "Hello, nexent!"

0 commit comments

Comments
 (0)