Skip to content

Commit 7dae97e

Browse files
committed
fix unit test
1 parent 19dbf7e commit 7dae97e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

airbyte_cdk/destinations/vector_db_based/embedder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,9 @@ def __init__(self, config: CohereEmbeddingConfigModel):
140140
super().__init__()
141141
# Client is set internally
142142
self.embeddings = CohereEmbeddings(
143-
cohere_api_key=config.cohere_key, model="embed-english-light-v2.0"
143+
cohere_api_key=config.cohere_key,
144+
model="embed-english-light-v2.0",
145+
user_agent="airbyte-cdk",
144146
) # type: ignore
145147

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

0 commit comments

Comments
 (0)