Azure AI Search(旧Cognitive Search) ベクトル クエリの python ライブラリバージョンアップ対応

背景

• Azure AI Search でベクトルクエリができる
• python で実行する場合、azure-search-documents ライブラリをバージョン指定してベータ版をインストールしていた
  • ドキュメントでも記載あり
  • https://learn.microsoft.com/ja-jp/azure/search/vector-search-how-to-query?tabs=python-vector-query%2Cfilter-2023-11-01
• b が取れた 11.4.0 がリリースされたので、ライブラリのバージョンアップや API の変更点に対応していく
  • https://pypi.org/project/azure-search-documents/11.4.0/

ライブラリアップデート

• ライブラリをアンインストール
  • pip uninstall azure-search-documents
• バージョン指定してライブラリをインストール
  • pip install azure-search-documents==11.4.0

プログラム修正

11.4.0 のインタフェースに合わせて修正する
参考までに過去バージョンサンプルも載せる

11.4.0

from azure.search.documents import SearchClient
from azure.search.documents.models import VectorizedQuery
:
:
results = search_client.search(
    search_text=tmp_search_text,
    vector_queries=[
        VectorizedQuery(
            kind="vector", vector=tmp_vector, k_nearest_neighbors=3, fields="embedding"
        )
    ],
)

11.4.0b08

from azure.search.documents import SearchClient
from azure.search.documents.models import Vector
:
:
results = search_client.search(
    search_text=tmp_search_text,
    vectors=[Vector(value=tmp_vector, k=3, fields="embedding")],
)

11.4.0b06

from azure.search.documents import SearchClient
:
:
results = search_client.search(
    search_text=tmp_search_text,
    vector=tmp_vector,
    vector_fields="embedding",
    top_k=3)

参考

以下ソースの tag を切り替えるとインタフェースを確認できる
https://github.com/Azure/azure-sdk-for-python/blob/azure-search-documents_11.4.0/sdk/search/azure-search-documents/azure/search/documents/_search_client.py