From 8fd96c7275c828ba056445f68d66f581ff99f382 Mon Sep 17 00:00:00 2001
From: shijianglong <stonyme@vip.qq.com>
Date: Mon, 5 Jan 2026 16:00:36 +0800
Subject: [PATCH] fix: handle unsupported dimensions and ensure mock mode
 safety

---
 .../databases/vector/embeddings/LiteLLMEmbeddingEngine.py     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
index 3a08b7eaf..7c458f352 100644
--- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
@@ -28,6 +28,7 @@ from cognee.infrastructure.llm.tokenizer.TikToken import (
 from cognee.shared.rate_limiting import embedding_rate_limiter_context_manager
 
 litellm.set_verbose = False
+litellm.drop_params = True
 logger = get_logger("LiteLLMEmbeddingEngine")
 
 
@@ -111,7 +112,8 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine):
         """
         try:
             if self.mock:
-                response = {"data": [{"embedding": [0.0] * self.dimensions} for _ in text]}
+                dim = self.dimensions if self.dimensions is not None else 3072 
+                response = {"data": [{"embedding": [0.0] * dim} for _ in text]}
                 return [data["embedding"] for data in response["data"]]
             else:
                 async with embedding_rate_limiter_context_manager():