Limit cohere to <5 (#38348)

apache · Mar 20, 2024 · 3400dfe · 3400dfe
1 parent 8e7b1ad
commit 3400dfe
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/airflow/providers/cohere/hooks/cohere.py b/airflow/providers/cohere/hooks/cohere.py
@@ -53,7 +53,7 @@ def __init__(
         self.max_retries = max_retries
 
     @cached_property
-    def get_conn(self) -> cohere.Client:
+    def get_conn(self) -> cohere.Client:  # type: ignore[override]
         conn = self.get_connection(self.conn_id)
         return cohere.Client(
             api_key=conn.password, timeout=self.timeout, max_retries=self.max_retries, api_url=conn.host

diff --git a/airflow/providers/cohere/provider.yaml b/airflow/providers/cohere/provider.yaml
@@ -42,7 +42,7 @@ integrations:
 
 dependencies:
   - apache-airflow>=2.6.0
-  - cohere>=4.37
+  - cohere>=4.37,<5
 
 hooks:
   - integration-name: Cohere

diff --git a/generated/provider_dependencies.json b/generated/provider_dependencies.json
@@ -330,7 +330,7 @@
   "cohere": {
     "deps": [
       "apache-airflow>=2.6.0",
-      "cohere>=4.37"
+      "cohere>=4.37,<5"
     ],
     "devel-deps": [],
     "cross-providers-deps": [],

diff --git a/pyproject.toml b/pyproject.toml
@@ -663,7 +663,7 @@ cncf-kubernetes = [ # source: airflow/providers/cncf/kubernetes/provider.yaml
   "kubernetes_asyncio>=28.1.0,<=29.0.0",
 ]
 cohere = [ # source: airflow/providers/cohere/provider.yaml
-  "cohere>=4.37",
+  "cohere>=4.37,<5",
 ]
 common-io = [] # source: airflow/providers/common/io/provider.yaml
 common-sql = [ # source: airflow/providers/common/sql/provider.yaml