Skip to content

Commit

Permalink
Improve docs for Azure OpenAI environment vars (mlflow#10441)
Browse files Browse the repository at this point in the history
Signed-off-by: Prithvi Kannan <prithvi.kannan@databricks.com>
Signed-off-by: swathi <konakanchi.swathi@gmail.com>
  • Loading branch information
prithvikannan authored and KonakanchiSwathi committed Nov 29, 2023
1 parent ee6fc01 commit 22c0a3a
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions docs/source/python_api/mlflow.metrics.rst
Expand Up @@ -204,3 +204,10 @@ You can also create your own generative AI :py:class:`EvaluationMetric <mlflow.m
When using generative AI :py:class:`EvaluationMetric <mlflow.metrics.EvaluationMetric>`\s, it is important to pass in an :py:class:`EvaluationExample <mlflow.metrics.genai.EvaluationExample>`

.. autoclass:: mlflow.metrics.genai.EvaluationExample

Users must set the appropriate environment variables for the LLM service they are using for
evaluation. For example, if you are using OpenAI's API, you must set the ``OPENAI_API_KEY``
environment variable. If using Azure OpenAI, you must also set the ``OPENAI_API_TYPE``,
``OPENAI_API_VERSION``, ``OPENAI_API_BASE``, and ``OPENAI_DEPLOYMENT_NAME`` environment variables.
See `Azure OpenAI documentation <https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/switching-endpoints>`_
Users do not need to set these environment variables if they are using a gateway route.

0 comments on commit 22c0a3a

Please sign in to comment.