From 778ed3a8e7f63a3748c40f9373790e6e8183ea57 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 7 Feb 2024 18:50:04 -0300
Subject: [PATCH] Fix disallowed_special parameter in OpenAIEmbeddingsComponent

---
 .../langflow/components/embeddings/OpenAIEmbeddings.py       | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py b/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py
index 9ff91dd0e..9a16a2088 100644
--- a/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py
+++ b/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py
@@ -95,12 +95,15 @@ class OpenAIEmbeddingsComponent(CustomComponent):
         tikToken_enable: bool = True,
         tiktoken_model_name: Optional[str] = None,
     ) -> Union[OpenAIEmbeddings, Callable]:
+        # This is to avoid errors with Vector Stores (e.g Chroma)
+        if disallowed_special == ["all"]:
+            disallowed_special = "all"
         return OpenAIEmbeddings(
             tiktoken_enabled=tikToken_enable,
             default_headers=default_headers,
             default_query=default_query,
             allowed_special=set(allowed_special),
-            disallowed_special=set(disallowed_special),
+            disallowed_special="all",
             chunk_size=chunk_size,
             client=client,
             deployment=deployment,