Add Azure OAI secrets to gh actions (#142)

Adds required env vars for azure openai support.
redis · May 6, 2024 · 9a1a5b4 · 9a1a5b4
1 parent 998d4ec
commit 9a1a5b4
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/.github/workflows/run_tests.yml b/.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"
 

diff --git a/docs/user_guide/vectorizers_04.ipynb b/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",

diff --git a/tests/integration/test_vectorizers.py b/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):