From d7493879af1cae93e84190e31140f5e135e33796 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Wed, 18 Dec 2024 14:39:54 +0100
Subject: [PATCH] fix: fixing boolean env for github actions

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

diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
index 31e49a676..dce12b318 100644
--- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
@@ -30,7 +30,11 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine):
         self.api_version = api_version
         self.model = model
         self.dimensions = dimensions
-        self.mock = os.getenv("MOCK_EMBEDDING", False).lower() in ("true", "1", "yes")
+
+        enable_mocking = os.getenv("MOCK_EMBEDDING", "false")
+        if isinstance(enable_mocking, bool):
+            enable_mocking= str(enable_mocking).lower()
+        self.mock = enable_mocking in ("true", "1", "yes")
 
     MAX_RETRIES = 5
     retry_count = 0