Add Azure Cognitive Search vector connector (#1103)

Azure Cognitive Search integration with Python SK
microsoft · Jun 14, 2023 · 2d52965 · 2d52965
1 parent 560033b
commit 2d52965
Show file tree

Hide file tree

Showing 8 changed files with 842 additions and 151 deletions.
diff --git a/.github/workflows/python-integration-tests.yml b/.github/workflows/python-integration-tests.yml
@@ -39,13 +39,15 @@ jobs:
         export HNSWLIB_NO_NATIVE=1
         python -m pip install --upgrade pip setuptools wheel
         python -m pip install poetry pytest
-        cd python && poetry install --with hugging_face --with chromadb --with weaviate
+        poetry add azure-search-documents==11.4.0a20230509004 --source azure-sdk-dev
+        cd python && poetry install --with hugging_face --with chromadb --with weaviate --with azure_cognitive_search
     - name: Install dependencies wtih hnswlib native enabled
       if: matrix.os != 'macos-latest' || matrix.python-version != '3.11'
       run: |
         python -m pip install --upgrade pip setuptools wheel
         python -m pip install poetry pytest
-        cd python && poetry install --with hugging_face --with chromadb --with weaviate
+         poetry add azure-search-documents==11.4.0a20230509004 --source azure-sdk-dev
+        cd python && poetry install --with hugging_face --with chromadb --with weaviate --with azure_cognitive_search
     - name: Run Integration Tests
       shell: bash
       env: # Set Azure credentials secret as an input

diff --git a/python/.env.example b/python/.env.example
@@ -3,3 +3,5 @@ OPENAI_ORG_ID=""
 AZURE_OPENAI_DEPLOYMENT_NAME=""
 AZURE_OPENAI_ENDPOINT=""
 AZURE_OPENAI_API_KEY=""
+AZURE_SEARCH_SERVICE_ENDPOINT=""
+AZURE_SEARCH_ADMIN_KEY=""