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

Update OpenAI API Key handling in OpenAIEmbeddingsComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-31 21:30:49 -03:00
commit bd96a863a7
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/components/embeddings/OpenAIEmbeddings.py
View file

@ -57,7 +57,7 @@ class OpenAIEmbeddingsComponent(CustomComponent):
"password": True, "password": True,
"advanced": True, "advanced": True,
}, },
"openai_api_key": {"display_name": "OpenAI API Key", "password": True, "required": True}, "openai_api_key": {"display_name": "OpenAI API Key", "password": True},
"openai_api_type": { "openai_api_type": {
"display_name": "OpenAI API Type", "display_name": "OpenAI API Type",
"advanced": True, "advanced": True,
@ -87,6 +87,7 @@ class OpenAIEmbeddingsComponent(CustomComponent):
def build( def build(
self, self,
openai_api_key: str,
default_headers: Optional[Dict[str, str]] = None, default_headers: Optional[Dict[str, str]] = None,
default_query: Optional[NestedDict] = {}, default_query: Optional[NestedDict] = {},
allowed_special: List[str] = [], allowed_special: List[str] = [],
@ -99,7 +100,6 @@ class OpenAIEmbeddingsComponent(CustomComponent):
model: str = "text-embedding-3-small", model: str = "text-embedding-3-small",
model_kwargs: NestedDict = {}, model_kwargs: NestedDict = {},
openai_api_base: Optional[str] = None, openai_api_base: Optional[str] = None,
openai_api_key: Optional[str] = "",
openai_api_type: Optional[str] = None, openai_api_type: Optional[str] = None,
openai_api_version: Optional[str] = None, openai_api_version: Optional[str] = None,
openai_organization: Optional[str] = None, openai_organization: Optional[str] = None,