Skip to content

Commit d749387

Browse files
committed
fix: fixing boolean env for github actions
1 parent 5eaeebd commit d749387

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,11 @@ def __init__(
3030
self.api_version = api_version
3131
self.model = model
3232
self.dimensions = dimensions
33-
self.mock = os.getenv("MOCK_EMBEDDING", False).lower() in ("true", "1", "yes")
33+
34+
enable_mocking = os.getenv("MOCK_EMBEDDING", "false")
35+
if isinstance(enable_mocking, bool):
36+
enable_mocking= str(enable_mocking).lower()
37+
self.mock = enable_mocking in ("true", "1", "yes")
3438

3539
MAX_RETRIES = 5
3640
retry_count = 0

0 commit comments

Comments
 (0)