utkarsharma2 commented on code in PR #38465:
URL: https://github.com/apache/airflow/pull/38465#discussion_r1607935037
##########
airflow/providers/cohere/operators/embedding.py:
##########
@@ -39,6 +41,17 @@ class CohereEmbeddingOperator(BaseOperator):
information for Cohere. Defaults to "cohere_default".
:param timeout: Timeout in seconds for Cohere API.
:param max_retries: Number of times to retry before failing.
+ :param request_options: Request-specific configuration.
+ Fields:
+ - timeout_in_seconds: int. The number of seconds to await an API call
before timing out.
+
+ - max_retries: int. The max number of retries to attempt if the API
call fails.
+
+ - additional_headers: typing.Dict[str, typing.Any]. A dictionary
containing additional parameters to spread into the request's header dict
+
+ - additional_query_parameters: typing.Dict[str, typing.Any]. A
dictionary containing additional parameters to spread into the request's query
parameters dict
+
+ - additional_body_parameters: typing.Dict[str, typing.Any]. A
dictionary containing additional parameters to spread into the request's body
parameters dict
Review Comment:
It would be nice to point to original docs here, in case they change in
future releases.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
