From 160e0e092ab3b30809f9918c5fa4d3dc56d3e5a7 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 7 Oct 2023 13:36:35 -0700
Subject: [PATCH] (docs) azure openai

---
 docs/my-website/docs/providers/azure.md | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/docs/my-website/docs/providers/azure.md b/docs/my-website/docs/providers/azure.md
index a713676f260..ec2e65a00c9 100644
--- a/docs/my-website/docs/providers/azure.md
+++ b/docs/my-website/docs/providers/azure.md
@@ -1,20 +1,20 @@
-# Azure
-LiteLLM supports Azure Chat + Embedding calls. 
-
-### API KEYS
+# Azure OpenAI
+## API KEYS
 ```python
 import os
-
 os.environ["AZURE_API_KEY"] = ""
 os.environ["AZURE_API_BASE"] = ""
 os.environ["AZURE_API_VERSION"] = ""
 ```
 
+## Usage
+<a target="_blank" href="https://colab.research.google.com/github/BerriAI/litellm/blob/main/cookbook/LiteLLM_Azure_OpenAI.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
+</a>
 
-### Azure OpenAI Chat Completion Models
+### Completion - using .env variables
 
 ```python
-
 from litellm import completion
 
 ## set ENV variables
@@ -22,16 +22,19 @@ os.environ["AZURE_API_KEY"] = ""
 os.environ["AZURE_API_BASE"] = ""
 os.environ["AZURE_API_VERSION"] = ""
 
-messages = [{ "content": "Hello, how are you?","role": "user"}]
-
 # azure call
-response = completion("azure/<your_deployment_name>", messages)
+response = completion(
+    model = "azure/<your_deployment_name>", 
+    messages = [{ "content": "Hello, how are you?","role": "user"}]
+)
 ```
 
-### Set API Key, API Base, API Version in Completion()
+### Completion - using api_key, api_base, api_version
 
 ```python
 import litellm
+
+# azure call
 response = litellm.completion(
     model = "azure/<your deployment name>",             # model = azure/<your deployment name> 
     api_base = "",                                      # azure api base