jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Fix disallowed_special parameter in OpenAIEmbeddingsComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-07 18:50:04 -03:00
commit 778ed3a8e7
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/components/embeddings/OpenAIEmbeddings.py
View file

@ -95,12 +95,15 @@ class OpenAIEmbeddingsComponent(CustomComponent):
tikToken_enable: bool = True, tikToken_enable: bool = True,
tiktoken_model_name: Optional[str] = None, tiktoken_model_name: Optional[str] = None,
) -> Union[OpenAIEmbeddings, Callable]: ) -> Union[OpenAIEmbeddings, Callable]:
# This is to avoid errors with Vector Stores (e.g Chroma)
if disallowed_special == ["all"]:
disallowed_special = "all"
return OpenAIEmbeddings( return OpenAIEmbeddings(
tiktoken_enabled=tikToken_enable, tiktoken_enabled=tikToken_enable,
default_headers=default_headers, default_headers=default_headers,
default_query=default_query, default_query=default_query,
allowed_special=set(allowed_special), allowed_special=set(allowed_special),
disallowed_special=set(disallowed_special), disallowed_special="all",
chunk_size=chunk_size, chunk_size=chunk_size,
client=client, client=client,
deployment=deployment, deployment=deployment,