From d4b0706580210da03bec0d5e57b0242fa764433c Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Tue, 16 Jan 2024 11:40:35 -0300
Subject: [PATCH] Update CohereEmbeddingsComponent and ChatOpenAIComponent
 configurations

---
 .../langflow/components/embeddings/CohereEmbeddings.py        | 4 +++-
 src/backend/langflow/components/llms/ChatOpenAI.py            | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/embeddings/CohereEmbeddings.py b/src/backend/langflow/components/embeddings/CohereEmbeddings.py
index f6b4ee1c1..cac7c1d75 100644
--- a/src/backend/langflow/components/embeddings/CohereEmbeddings.py
+++ b/src/backend/langflow/components/embeddings/CohereEmbeddings.py
@@ -20,12 +20,14 @@ class CohereEmbeddingsComponent(CustomComponent):
     def build(
         self,
         request_timeout: Optional[float] = None,
-        cohere_api_key: Optional[str] = None,
+        cohere_api_key: str = None,
+        max_retries: Optional[int] = None,
         model: str = "embed-english-v2.0",
         truncate: Optional[str] = None,
         user_agent: Optional[str] = "langchain",
     ) -> CohereEmbeddings:
         return CohereEmbeddings(
+            max_retries=max_retries,
             user_agent=user_agent,
             request_timeout=request_timeout,
             cohere_api_key=cohere_api_key,
diff --git a/src/backend/langflow/components/llms/ChatOpenAI.py b/src/backend/langflow/components/llms/ChatOpenAI.py
index 8663747bf..44fd8dd40 100644
--- a/src/backend/langflow/components/llms/ChatOpenAI.py
+++ b/src/backend/langflow/components/llms/ChatOpenAI.py
@@ -51,6 +51,7 @@ class ChatOpenAIComponent(CustomComponent):
                 "field_type": "str",
                 "advanced": False,
                 "required": False,
+                "password": True,
             },
             "temperature": {
                 "display_name": "Temperature",