o
    Á¿i=  ã                   @   s   d Z dS )a(  
The Vertex AI integration instruments the Vertex Generative AI SDK for Python for requests made to Google models.

All traces submitted from the Vertex AI integration are tagged by:

- ``service``, ``env``, ``version``: see the `Unified Service Tagging docs <https://docs.datadoghq.com/getting_started/tagging/unified_service_tagging>`_.
- ``vertexai.request.provider``: LLM provider used in the request (e.g. ``google`` for Google models).
- ``vertexai.request.model``: Google model used in the request.


Enabling
~~~~~~~~

The Vertex AI integration is enabled automatically when you use
:ref:`ddtrace-run<ddtracerun>` or :ref:`import ddtrace.auto<ddtraceauto>`.

Alternatively, use :func:`patch() <ddtrace.patch>` to manually enable the Vertex AI integration::

    from ddtrace import config, patch

    patch(vertexai=True)


Configuration
~~~~~~~~~~~~~

.. py:data:: ddtrace.config.vertexai["service"]

   The service name reported by default for Vertex AI requests.

   Alternatively, set this option with the ``DD_VERTEXAI_SERVICE`` environment variable.
N)Ú__doc__© r   r   ú^/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/vertexai/__init__.pyÚ<module>   s    