fix: address CI failures with types-pytz and CohereEmbedder updates

Co-Authored-By: natik@airbyte.io <natik@airbyte.io>
airbytehq · Mar 1, 2025 · 6350683 · 6350683
1 parent 42d831a
commit 6350683
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/airbyte_cdk/destinations/vector_db_based/embedder.py b/airbyte_cdk/destinations/vector_db_based/embedder.py
@@ -145,9 +145,13 @@ class CohereEmbedder(Embedder):
     def __init__(self, config: CohereEmbeddingConfigModel):
         super().__init__()
         # Client is set internally
+        from pydantic import SecretStr
+
         self.embeddings = CohereEmbeddings(
-            cohere_api_key=config.cohere_key, model="embed-english-light-v2.0"
-        )  # type: ignore
+            api_key=SecretStr(config.cohere_key), 
+            model="embed-english-light-v2.0",
+            client_kwargs={"user_agent": "langchain"}
+        )
 
     def check(self) -> Optional[str]:
         try:

diff --git a/pyproject.toml b/pyproject.toml
@@ -105,6 +105,7 @@ types-requests = "^2.32.0.20241016"
 types-python-dateutil = "^2.9.0.20241003"
 types-pyyaml = "^6.0.12.20240917"
 types-cachetools = "^5.5.0.20240820"
+types-pytz = "^2024.1.0.20240417"
 
 [tool.poetry.extras]
 file-based = ["avro", "fastavro", "pyarrow", "unstructured", "pdf2image", "pdfminer.six", "unstructured.pytesseract", "pytesseract", "markdown", "python-calamine", "python-snappy", "pi_heif"]