Skip to content

Commit 6757a1f

Browse files
fix: wrap API keys with SecretStr for langchain embeddings
Co-Authored-By: David Gold <[email protected]>
1 parent 3b4c403 commit 6757a1f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

airbyte_cdk/destinations/vector_db_based/embedder.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from langchain_cohere import CohereEmbeddings
1111
from langchain_community.embeddings import FakeEmbeddings, LocalAIEmbeddings
1212
from langchain_openai import AzureOpenAIEmbeddings, OpenAIEmbeddings
13+
from pydantic import SecretStr
1314

1415
from airbyte_cdk.destinations.vector_db_based.config import (
1516
AzureOpenAIEmbeddingConfigModel,
@@ -118,7 +119,7 @@ def __init__(self, config: AzureOpenAIEmbeddingConfigModel, chunk_size: int):
118119
# Azure OpenAI API has — as of 20230927 — a limit of 16 documents per request
119120
super().__init__(
120121
AzureOpenAIEmbeddings( # type: ignore [call-arg]
121-
api_key=config.openai_key,
122+
api_key=SecretStr(config.openai_key),
122123
chunk_size=16,
123124
max_retries=15,
124125
api_version="2023-05-15",
@@ -138,7 +139,7 @@ def __init__(self, config: CohereEmbeddingConfigModel):
138139
super().__init__()
139140
# Client is set internally
140141
self.embeddings = CohereEmbeddings(
141-
cohere_api_key=config.cohere_key, model="embed-english-light-v2.0"
142+
cohere_api_key=SecretStr(config.cohere_key), model="embed-english-light-v2.0"
142143
) # type: ignore
143144

144145
def check(self) -> Optional[str]:

0 commit comments

Comments
 (0)