oci 2.150.0__py3-none-any.whl → 2.150.2__py3-none-any.whl

oci/generative_ai_inference/generative_ai_inference_client.py

@@ -122,6 +122,103 @@ class GenerativeAiInferenceClient(object):
         self.retry_strategy = kwargs.get('retry_strategy')
         self.circuit_breaker_callback = kwargs.get('circuit_breaker_callback')
 
+    def apply_guardrails(self, apply_guardrails_details, **kwargs):
+        """
+        Applies guardrails to the input text, including content moderation, PII detection, and prompt injection protection.
+
+
+        :param oci.generative_ai_inference.models.ApplyGuardrailsDetails apply_guardrails_details: (required)
+            Details for applying guardrails to the input text.
+
+        :param str opc_retry_token: (optional)
+            A token that uniquely identifies a request so it can be retried in case of a timeout or
+            server error without risk of executing that same action again. Retry tokens expire after 24
+            hours, but can be invalidated before that, in case of conflicting operations. For example, if a resource is deleted and purged from the system, then a retry of the original creation request
+            is rejected.
+
+        :param str opc_request_id: (optional)
+            The client request ID for tracing.
+
+        :param obj retry_strategy: (optional)
+            A retry strategy to apply to this specific operation/call. This will override any retry strategy set at the client-level.
+
+            This should be one of the strategies available in the :py:mod:`~oci.retry` module. This operation uses :py:data:`~oci.retry.DEFAULT_RETRY_STRATEGY` as default if no retry strategy is provided.
+            The specifics of the default retry strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/retries.html>`__.
+
+            To have this operation explicitly not perform any retries, pass an instance of :py:class:`~oci.retry.NoneRetryStrategy`.
+
+        :param bool allow_control_chars: (optional)
+            allow_control_chars is a boolean to indicate whether or not this request should allow control characters in the response object.
+            By default, the response will not allow control characters in strings
+
+        :return: A :class:`~oci.response.Response` object with data of type :class:`~oci.generative_ai_inference.models.ApplyGuardrailsResult`
+        :rtype: :class:`~oci.response.Response`
+
+        :example:
+        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.2/generativeaiinference/apply_guardrails.py.html>`__ to see an example of how to use apply_guardrails API.
+        """
+        # Required path and query arguments. These are in camelCase to replace values in service endpoints.
+        required_arguments = []
+        resource_path = "/actions/applyGuardrails"
+        method = "POST"
+        operation_name = "apply_guardrails"
+        api_reference_link = "https://docs.oracle.com/iaas/api/#/en/generative-ai-inference/20231130/ApplyGuardrailsResult/ApplyGuardrails"
+
+        # Don't accept unknown kwargs
+        expected_kwargs = [
+            "allow_control_chars",
+            "retry_strategy",
+            "opc_retry_token",
+            "opc_request_id"
+        ]
+        extra_kwargs = [_key for _key in six.iterkeys(kwargs) if _key not in expected_kwargs]
+        if extra_kwargs:
+            raise ValueError(
+                f"apply_guardrails got unknown kwargs: {extra_kwargs!r}")
+
+        header_params = {
+            "accept": "application/json",
+            "content-type": "application/json",
+            "opc-retry-token": kwargs.get("opc_retry_token", missing),
+            "opc-request-id": kwargs.get("opc_request_id", missing)
+        }
+        header_params = {k: v for (k, v) in six.iteritems(header_params) if v is not missing and v is not None}
+
+        retry_strategy = self.base_client.get_preferred_retry_strategy(
+            operation_retry_strategy=kwargs.get('retry_strategy'),
+            client_retry_strategy=self.retry_strategy
+        )
+        if retry_strategy is None:
+            retry_strategy = retry.DEFAULT_RETRY_STRATEGY
+
+        if retry_strategy:
+            if not isinstance(retry_strategy, retry.NoneRetryStrategy):
+                self.base_client.add_opc_retry_token_if_needed(header_params)
+                self.base_client.add_opc_client_retries_header(header_params)
+                retry_strategy.add_circuit_breaker_callback(self.circuit_breaker_callback)
+            return retry_strategy.make_retrying_call(
+                self.base_client.call_api,
+                resource_path=resource_path,
+                method=method,
+                header_params=header_params,
+                body=apply_guardrails_details,
+                response_type="ApplyGuardrailsResult",
+                allow_control_chars=kwargs.get('allow_control_chars'),
+                operation_name=operation_name,
+                api_reference_link=api_reference_link,
+                required_arguments=required_arguments)
+        else:
+            return self.base_client.call_api(
+                resource_path=resource_path,
+                method=method,
+                header_params=header_params,
+                body=apply_guardrails_details,
+                response_type="ApplyGuardrailsResult",
+                allow_control_chars=kwargs.get('allow_control_chars'),
+                operation_name=operation_name,
+                api_reference_link=api_reference_link,
+                required_arguments=required_arguments)
+
     def chat(self, chat_details, **kwargs):
         """
         Creates a response for the given conversation.
@@ -155,7 +252,7 @@ class GenerativeAiInferenceClient(object):
         :rtype: :class:`~oci.response.Response`
 
         :example:
-        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.0/generativeaiinference/chat.py.html>`__ to see an example of how to use chat API.
+        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.2/generativeaiinference/chat.py.html>`__ to see an example of how to use chat API.
         """
         # Required path and query arguments. These are in camelCase to replace values in service endpoints.
         required_arguments = []
@@ -254,7 +351,7 @@ class GenerativeAiInferenceClient(object):
         :rtype: :class:`~oci.response.Response`
 
         :example:
-        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.0/generativeaiinference/embed_text.py.html>`__ to see an example of how to use embed_text API.
+        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.2/generativeaiinference/embed_text.py.html>`__ to see an example of how to use embed_text API.
         """
         # Required path and query arguments. These are in camelCase to replace values in service endpoints.
         required_arguments = []
@@ -351,7 +448,7 @@ class GenerativeAiInferenceClient(object):
         :rtype: :class:`~oci.response.Response`
 
         :example:
-        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.0/generativeaiinference/generate_text.py.html>`__ to see an example of how to use generate_text API.
+        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.2/generativeaiinference/generate_text.py.html>`__ to see an example of how to use generate_text API.
         """
         # Required path and query arguments. These are in camelCase to replace values in service endpoints.
         required_arguments = []
@@ -450,7 +547,7 @@ class GenerativeAiInferenceClient(object):
         :rtype: :class:`~oci.response.Response`
 
         :example:
-        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.0/generativeaiinference/rerank_text.py.html>`__ to see an example of how to use rerank_text API.
+        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.2/generativeaiinference/rerank_text.py.html>`__ to see an example of how to use rerank_text API.
         """
         # Required path and query arguments. These are in camelCase to replace values in service endpoints.
         required_arguments = []
@@ -547,7 +644,7 @@ class GenerativeAiInferenceClient(object):
         :rtype: :class:`~oci.response.Response`
 
         :example:
-        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.0/generativeaiinference/summarize_text.py.html>`__ to see an example of how to use summarize_text API.
+        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/2.150.2/generativeaiinference/summarize_text.py.html>`__ to see an example of how to use summarize_text API.
         """
         # Required path and query arguments. These are in camelCase to replace values in service endpoints.
         required_arguments = []

oci/generative_ai_inference/models/__init__.py

@@ -6,9 +6,12 @@
 
 from __future__ import absolute_import
 
+from .apply_guardrails_details import ApplyGuardrailsDetails
+from .apply_guardrails_result import ApplyGuardrailsResult
 from .assistant_message import AssistantMessage
 from .base_chat_request import BaseChatRequest
 from .base_chat_response import BaseChatResponse
+from .category_score import CategoryScore
 from .chat_choice import ChatChoice
 from .chat_content import ChatContent
 from .chat_details import ChatDetails
@@ -31,6 +34,8 @@ from .cohere_tool_call import CohereToolCall
 from .cohere_tool_message import CohereToolMessage
 from .cohere_tool_result import CohereToolResult
 from .cohere_user_message import CohereUserMessage
+from .content_moderation_configuration import ContentModerationConfiguration
+from .content_moderation_result import ContentModerationResult
 from .dedicated_serving_mode import DedicatedServingMode
 from .document import Document
 from .document_rank import DocumentRank
@@ -43,6 +48,10 @@ from .generate_text_result import GenerateTextResult
 from .generated_text import GeneratedText
 from .generic_chat_request import GenericChatRequest
 from .generic_chat_response import GenericChatResponse
+from .guardrail_configs import GuardrailConfigs
+from .guardrails_input import GuardrailsInput
+from .guardrails_results import GuardrailsResults
+from .guardrails_text_input import GuardrailsTextInput
 from .image_content import ImageContent
 from .image_url import ImageUrl
 from .llama_llm_inference_request import LlamaLlmInferenceRequest
@@ -52,6 +61,10 @@ from .llm_inference_response import LlmInferenceResponse
 from .logprobs import Logprobs
 from .message import Message
 from .on_demand_serving_mode import OnDemandServingMode
+from .personally_identifiable_information_configuration import PersonallyIdentifiableInformationConfiguration
+from .personally_identifiable_information_result import PersonallyIdentifiableInformationResult
+from .prompt_injection_configuration import PromptInjectionConfiguration
+from .prompt_injection_protection_result import PromptInjectionProtectionResult
 from .rerank_text_details import RerankTextDetails
 from .rerank_text_result import RerankTextResult
 from .search_query import SearchQuery
@@ -73,9 +86,12 @@ from .user_message import UserMessage
 
 # Maps type names to classes for generative_ai_inference services.
 generative_ai_inference_type_mapping = {
+    "ApplyGuardrailsDetails": ApplyGuardrailsDetails,
+    "ApplyGuardrailsResult": ApplyGuardrailsResult,
     "AssistantMessage": AssistantMessage,
     "BaseChatRequest": BaseChatRequest,
     "BaseChatResponse": BaseChatResponse,
+    "CategoryScore": CategoryScore,
     "ChatChoice": ChatChoice,
     "ChatContent": ChatContent,
     "ChatDetails": ChatDetails,
@@ -98,6 +114,8 @@ generative_ai_inference_type_mapping = {
     "CohereToolMessage": CohereToolMessage,
     "CohereToolResult": CohereToolResult,
     "CohereUserMessage": CohereUserMessage,
+    "ContentModerationConfiguration": ContentModerationConfiguration,
+    "ContentModerationResult": ContentModerationResult,
     "DedicatedServingMode": DedicatedServingMode,
     "Document": Document,
     "DocumentRank": DocumentRank,
@@ -110,6 +128,10 @@ generative_ai_inference_type_mapping = {
     "GeneratedText": GeneratedText,
     "GenericChatRequest": GenericChatRequest,
     "GenericChatResponse": GenericChatResponse,
+    "GuardrailConfigs": GuardrailConfigs,
+    "GuardrailsInput": GuardrailsInput,
+    "GuardrailsResults": GuardrailsResults,
+    "GuardrailsTextInput": GuardrailsTextInput,
     "ImageContent": ImageContent,
     "ImageUrl": ImageUrl,
     "LlamaLlmInferenceRequest": LlamaLlmInferenceRequest,
@@ -119,6 +141,10 @@ generative_ai_inference_type_mapping = {
     "Logprobs": Logprobs,
     "Message": Message,
     "OnDemandServingMode": OnDemandServingMode,
+    "PersonallyIdentifiableInformationConfiguration": PersonallyIdentifiableInformationConfiguration,
+    "PersonallyIdentifiableInformationResult": PersonallyIdentifiableInformationResult,
+    "PromptInjectionConfiguration": PromptInjectionConfiguration,
+    "PromptInjectionProtectionResult": PromptInjectionProtectionResult,
     "RerankTextDetails": RerankTextDetails,
     "RerankTextResult": RerankTextResult,
     "SearchQuery": SearchQuery,

oci/generative_ai_inference/models/apply_guardrails_details.py

@@ -0,0 +1,124 @@
+# coding: utf-8
+# Copyright (c) 2016, 2025, Oracle and/or its affiliates.  All rights reserved.
+# This software is dual-licensed to you under the Universal Permissive License (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl or Apache License 2.0 as shown at http://www.apache.org/licenses/LICENSE-2.0. You may choose either license.
+
+# NOTE: This class is auto generated by OracleSDKGenerator. DO NOT EDIT. API Version: 20231130
+
+
+from oci.util import formatted_flat_dict, NONE_SENTINEL, value_allowed_none_or_none_sentinel  # noqa: F401
+from oci.decorators import init_model_state_from_kwargs
+
+
+@init_model_state_from_kwargs
+class ApplyGuardrailsDetails(object):
+    """
+    Details for applying guardrails to the input text.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new ApplyGuardrailsDetails object with values from keyword arguments.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param input:
+            The value to assign to the input property of this ApplyGuardrailsDetails.
+        :type input: oci.generative_ai_inference.models.GuardrailsInput
+
+        :param guardrail_configs:
+            The value to assign to the guardrail_configs property of this ApplyGuardrailsDetails.
+        :type guardrail_configs: oci.generative_ai_inference.models.GuardrailConfigs
+
+        :param compartment_id:
+            The value to assign to the compartment_id property of this ApplyGuardrailsDetails.
+        :type compartment_id: str
+
+        """
+        self.swagger_types = {
+            'input': 'GuardrailsInput',
+            'guardrail_configs': 'GuardrailConfigs',
+            'compartment_id': 'str'
+        }
+        self.attribute_map = {
+            'input': 'input',
+            'guardrail_configs': 'guardrailConfigs',
+            'compartment_id': 'compartmentId'
+        }
+        self._input = None
+        self._guardrail_configs = None
+        self._compartment_id = None
+
+    @property
+    def input(self):
+        """
+        **[Required]** Gets the input of this ApplyGuardrailsDetails.
+
+        :return: The input of this ApplyGuardrailsDetails.
+        :rtype: oci.generative_ai_inference.models.GuardrailsInput
+        """
+        return self._input
+
+    @input.setter
+    def input(self, input):
+        """
+        Sets the input of this ApplyGuardrailsDetails.
+
+        :param input: The input of this ApplyGuardrailsDetails.
+        :type: oci.generative_ai_inference.models.GuardrailsInput
+        """
+        self._input = input
+
+    @property
+    def guardrail_configs(self):
+        """
+        **[Required]** Gets the guardrail_configs of this ApplyGuardrailsDetails.
+
+        :return: The guardrail_configs of this ApplyGuardrailsDetails.
+        :rtype: oci.generative_ai_inference.models.GuardrailConfigs
+        """
+        return self._guardrail_configs
+
+    @guardrail_configs.setter
+    def guardrail_configs(self, guardrail_configs):
+        """
+        Sets the guardrail_configs of this ApplyGuardrailsDetails.
+
+        :param guardrail_configs: The guardrail_configs of this ApplyGuardrailsDetails.
+        :type: oci.generative_ai_inference.models.GuardrailConfigs
+        """
+        self._guardrail_configs = guardrail_configs
+
+    @property
+    def compartment_id(self):
+        """
+        **[Required]** Gets the compartment_id of this ApplyGuardrailsDetails.
+        The OCID of the compartment to apply guardrails.
+
+
+        :return: The compartment_id of this ApplyGuardrailsDetails.
+        :rtype: str
+        """
+        return self._compartment_id
+
+    @compartment_id.setter
+    def compartment_id(self, compartment_id):
+        """
+        Sets the compartment_id of this ApplyGuardrailsDetails.
+        The OCID of the compartment to apply guardrails.
+
+
+        :param compartment_id: The compartment_id of this ApplyGuardrailsDetails.
+        :type: str
+        """
+        self._compartment_id = compartment_id
+
+    def __repr__(self):
+        return formatted_flat_dict(self)
+
+    def __eq__(self, other):
+        if other is None:
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other

oci/generative_ai_inference/models/apply_guardrails_result.py

@@ -0,0 +1,66 @@
+# coding: utf-8
+# Copyright (c) 2016, 2025, Oracle and/or its affiliates.  All rights reserved.
+# This software is dual-licensed to you under the Universal Permissive License (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl or Apache License 2.0 as shown at http://www.apache.org/licenses/LICENSE-2.0. You may choose either license.
+
+# NOTE: This class is auto generated by OracleSDKGenerator. DO NOT EDIT. API Version: 20231130
+
+
+from oci.util import formatted_flat_dict, NONE_SENTINEL, value_allowed_none_or_none_sentinel  # noqa: F401
+from oci.decorators import init_model_state_from_kwargs
+
+
+@init_model_state_from_kwargs
+class ApplyGuardrailsResult(object):
+    """
+    The result of applying guardrails to the input text.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new ApplyGuardrailsResult object with values from keyword arguments.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param results:
+            The value to assign to the results property of this ApplyGuardrailsResult.
+        :type results: oci.generative_ai_inference.models.GuardrailsResults
+
+        """
+        self.swagger_types = {
+            'results': 'GuardrailsResults'
+        }
+        self.attribute_map = {
+            'results': 'results'
+        }
+        self._results = None
+
+    @property
+    def results(self):
+        """
+        **[Required]** Gets the results of this ApplyGuardrailsResult.
+
+        :return: The results of this ApplyGuardrailsResult.
+        :rtype: oci.generative_ai_inference.models.GuardrailsResults
+        """
+        return self._results
+
+    @results.setter
+    def results(self, results):
+        """
+        Sets the results of this ApplyGuardrailsResult.
+
+        :param results: The results of this ApplyGuardrailsResult.
+        :type: oci.generative_ai_inference.models.GuardrailsResults
+        """
+        self._results = results
+
+    def __repr__(self):
+        return formatted_flat_dict(self)
+
+    def __eq__(self, other):
+        if other is None:
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other

oci/generative_ai_inference/models/assistant_message.py

@@ -13,6 +13,9 @@ from oci.decorators import init_model_state_from_kwargs
 class AssistantMessage(Message):
     """
     Represents a single instance of assistant message.
+
+    Note: When responding to a tool call, set `content` to `null` (not `\"\"`).
+    This ensures the model correctly processes tool call messages in multi-step interactions.
     """
 
     def __init__(self, **kwargs):

oci/generative_ai_inference/models/category_score.py

@@ -0,0 +1,101 @@
+# coding: utf-8
+# Copyright (c) 2016, 2025, Oracle and/or its affiliates.  All rights reserved.
+# This software is dual-licensed to you under the Universal Permissive License (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl or Apache License 2.0 as shown at http://www.apache.org/licenses/LICENSE-2.0. You may choose either license.
+
+# NOTE: This class is auto generated by OracleSDKGenerator. DO NOT EDIT. API Version: 20231130
+
+
+from oci.util import formatted_flat_dict, NONE_SENTINEL, value_allowed_none_or_none_sentinel  # noqa: F401
+from oci.decorators import init_model_state_from_kwargs
+
+
+@init_model_state_from_kwargs
+class CategoryScore(object):
+    """
+    A category with its score.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new CategoryScore object with values from keyword arguments.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param name:
+            The value to assign to the name property of this CategoryScore.
+        :type name: str
+
+        :param score:
+            The value to assign to the score property of this CategoryScore.
+        :type score: float
+
+        """
+        self.swagger_types = {
+            'name': 'str',
+            'score': 'float'
+        }
+        self.attribute_map = {
+            'name': 'name',
+            'score': 'score'
+        }
+        self._name = None
+        self._score = None
+
+    @property
+    def name(self):
+        """
+        **[Required]** Gets the name of this CategoryScore.
+        The name of the category.
+
+
+        :return: The name of this CategoryScore.
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name):
+        """
+        Sets the name of this CategoryScore.
+        The name of the category.
+
+
+        :param name: The name of this CategoryScore.
+        :type: str
+        """
+        self._name = name
+
+    @property
+    def score(self):
+        """
+        **[Required]** Gets the score of this CategoryScore.
+        The score of the category.
+
+
+        :return: The score of this CategoryScore.
+        :rtype: float
+        """
+        return self._score
+
+    @score.setter
+    def score(self, score):
+        """
+        Sets the score of this CategoryScore.
+        The score of the category.
+
+
+        :param score: The score of this CategoryScore.
+        :type: float
+        """
+        self._score = score
+
+    def __repr__(self):
+        return formatted_flat_dict(self)
+
+    def __eq__(self, other):
+        if other is None:
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other

oci/generative_ai_inference/models/cohere_chat_request.py

@@ -31,6 +31,18 @@ class CohereChatRequest(BaseChatRequest):
     #: This constant has a value of "FAST"
     CITATION_QUALITY_FAST = "FAST"
 
+    #: A constant which can be used with the safety_mode property of a CohereChatRequest.
+    #: This constant has a value of "CONTEXTUAL"
+    SAFETY_MODE_CONTEXTUAL = "CONTEXTUAL"
+
+    #: A constant which can be used with the safety_mode property of a CohereChatRequest.
+    #: This constant has a value of "STRICT"
+    SAFETY_MODE_STRICT = "STRICT"
+
+    #: A constant which can be used with the safety_mode property of a CohereChatRequest.
+    #: This constant has a value of "OFF"
+    SAFETY_MODE_OFF = "OFF"
+
     def __init__(self, **kwargs):
         """
         Initializes a new CohereChatRequest object with values from keyword arguments. The default value of the :py:attr:`~oci.generative_ai_inference.models.CohereChatRequest.api_format` attribute
@@ -136,6 +148,11 @@ class CohereChatRequest(BaseChatRequest):
             Allowed values for this property are: "ACCURATE", "FAST"
         :type citation_quality: str
 
+        :param safety_mode:
+            The value to assign to the safety_mode property of this CohereChatRequest.
+            Allowed values for this property are: "CONTEXTUAL", "STRICT", "OFF"
+        :type safety_mode: str
+
         """
         self.swagger_types = {
             'api_format': 'str',
@@ -161,7 +178,8 @@ class CohereChatRequest(BaseChatRequest):
             'is_force_single_step': 'bool',
             'stop_sequences': 'list[str]',
             'is_raw_prompting': 'bool',
-            'citation_quality': 'str'
+            'citation_quality': 'str',
+            'safety_mode': 'str'
         }
         self.attribute_map = {
             'api_format': 'apiFormat',
@@ -187,7 +205,8 @@ class CohereChatRequest(BaseChatRequest):
             'is_force_single_step': 'isForceSingleStep',
             'stop_sequences': 'stopSequences',
             'is_raw_prompting': 'isRawPrompting',
-            'citation_quality': 'citationQuality'
+            'citation_quality': 'citationQuality',
+            'safety_mode': 'safetyMode'
         }
         self._api_format = None
         self._message = None
@@ -213,6 +232,7 @@ class CohereChatRequest(BaseChatRequest):
         self._stop_sequences = None
         self._is_raw_prompting = None
         self._citation_quality = None
+        self._safety_mode = None
         self._api_format = 'COHERE'
 
     @property
@@ -809,6 +829,37 @@ class CohereChatRequest(BaseChatRequest):
             )
         self._citation_quality = citation_quality
 
+    @property
+    def safety_mode(self):
+        """
+        Gets the safety_mode of this CohereChatRequest.
+        Used to select the safety instruction inserted into the prompt. When selected CONTEXTUAL mode, It is appropriate for wide-ranging interactions with fewer constraints on output while maintaining core protections by rejecting harmful or illegal suggestions. When selected STRICT mode, it aims to avoid all sensitive topics, such as violent or sexual acts and profanity. When selected OFF, the safety instruction will be omitted. Note: This parameter is only compatible with models Command R 08-2024, Command R+ 08-2024 and newer. Also, command-r7b-12-2024 only supports \"CONTEXTUAL\" and \"STRICT\" modes.
+
+        Allowed values for this property are: "CONTEXTUAL", "STRICT", "OFF"
+
+
+        :return: The safety_mode of this CohereChatRequest.
+        :rtype: str
+        """
+        return self._safety_mode
+
+    @safety_mode.setter
+    def safety_mode(self, safety_mode):
+        """
+        Sets the safety_mode of this CohereChatRequest.
+        Used to select the safety instruction inserted into the prompt. When selected CONTEXTUAL mode, It is appropriate for wide-ranging interactions with fewer constraints on output while maintaining core protections by rejecting harmful or illegal suggestions. When selected STRICT mode, it aims to avoid all sensitive topics, such as violent or sexual acts and profanity. When selected OFF, the safety instruction will be omitted. Note: This parameter is only compatible with models Command R 08-2024, Command R+ 08-2024 and newer. Also, command-r7b-12-2024 only supports \"CONTEXTUAL\" and \"STRICT\" modes.
+
+
+        :param safety_mode: The safety_mode of this CohereChatRequest.
+        :type: str
+        """
+        allowed_values = ["CONTEXTUAL", "STRICT", "OFF"]
+        if not value_allowed_none_or_none_sentinel(safety_mode, allowed_values):
+            raise ValueError(
+                f"Invalid value for `safety_mode`, must be None or one of {allowed_values}"
+            )
+        self._safety_mode = safety_mode
+
     def __repr__(self):
         return formatted_flat_dict(self)