Add Azure OAI secrets to gh actions (#142)

Adds required env vars for azure openai support.

Author: tylerhutcherson

.github/workflows/run_tests.yml

@@ -60,6 +60,10 @@ jobs:
         GCP_LOCATION: ${{ secrets.GCP_LOCATION }}
         GCP_PROJECT_ID: ${{ secrets.GCP_PROJECT_ID }}
         COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
+        AZURE_OPENAI_API_KEY: ${{secrets.AZURE_OPENAI_API_KEY}}
+        AZURE_OPENAI_ENDPOINT: ${{secrets.AZURE_OPENAI_ENDPOINT}}
+        AZURE_OPENAI_DEPLOYMENT_NAME: ${{secrets.AZURE_OPENAI_DEPLOYMENT_NAME}}
+        OPENAI_API_VERSION: ${{secrets.OPENAI_API_VERSION}}
       run: |
         make test-cov
 
@@ -75,6 +79,10 @@ jobs:
         GCP_LOCATION: ${{ secrets.GCP_LOCATION }}
         GCP_PROJECT_ID: ${{ secrets.GCP_PROJECT_ID }}
         COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
+        AZURE_OPENAI_API_KEY: ${{secrets.AZURE_OPENAI_API_KEY}}
+        AZURE_OPENAI_ENDPOINT: ${{secrets.AZURE_OPENAI_ENDPOINT}}
+        AZURE_OPENAI_DEPLOYMENT_NAME: ${{secrets.AZURE_OPENAI_DEPLOYMENT_NAME}}
+        OPENAI_API_VERSION: ${{secrets.OPENAI_API_VERSION}}
       run: |
         cd docs/ && treon -v --exclude="./examples/openai_qna.ipynb"
 

docs/user_guide/vectorizers_04.ipynb

@@ -194,8 +194,10 @@
    "outputs": [],
    "source": [
     "# additionally to the API Key, setup the API endpoint and version\n",
+    "api_key = os.environ.get(\"AZURE_OPENAI_API_KEY\") or getpass.getpass(\"Enter your AzureOpenAI API key: \")\n",
     "api_version = os.environ.get(\"OPENAI_API_VERSION\") or getpass.getpass(\"Enter your AzureOpenAI API version: \")\n",
-    "azure_endpoint = os.environ.get(\"AZURE_OPENAI_ENDPOINT\") or getpass.getpass(\"Enter your AzureOpenAI API endpoint: \")"
+    "azure_endpoint = os.environ.get(\"AZURE_OPENAI_ENDPOINT\") or getpass.getpass(\"Enter your AzureOpenAI API endpoint: \")\n",
+    "deployment_name = os.environ.get(\"AZURE_OPENAI_DEPLOYMENT_NAME\", \"text-embedding-ada-002\")\n"
    ]
   },
   {
@@ -235,7 +237,7 @@
     "\n",
     "# create a vectorizer\n",
     "az_oai = AzureOpenAITextVectorizer(\n",
-    "    model=\"text-embedding-ada-002\", # Must be your custom deployment name\n",
+    "    model=deployment_name, # Must be your CUSTOM deployment name\n",
     "    api_config={\n",
     "        \"api_key\": api_key,\n",
     "        \"api_version\": api_version,\n",

tests/integration/test_vectorizers.py

@@ -41,7 +41,7 @@ def vectorizer(request, skip_vectorizer):
     elif request.param == CohereTextVectorizer:
         return request.param()
     elif request.param == AzureOpenAITextVectorizer:
-        return request.param()
+        return request.param(model=os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME", "text-embedding-ada-002"))
 
 
 def test_vectorizer_embed(vectorizer, skip_vectorizer):