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: commits-unsubscr...@airflow.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org