diff --git a/tools/sdk-ai-bots/Embeddings/embedding/embedding.py b/tools/sdk-ai-bots/Embeddings/embedding/embedding.py index 5dce2fbfe85..35964295576 100644 --- a/tools/sdk-ai-bots/Embeddings/embedding/embedding.py +++ b/tools/sdk-ai-bots/Embeddings/embedding/embedding.py @@ -4,7 +4,8 @@ SearchField, SearchFieldDataType, SimpleField, - HnswVectorSearchAlgorithmConfiguration + HnswAlgorithmConfiguration, + VectorSearch ) from azure.core.credentials import AzureKeyCredential from azure.search.documents import SearchClient @@ -71,11 +72,15 @@ def __init__(self): facetable=True, ) ] - algorithm = HnswVectorSearchAlgorithmConfiguration( + algorithm_configuration = HnswAlgorithmConfiguration( name="searchAlgorithm", + kind="hnsw", parameters={ "metric": "cosine", - } + } + ) + vector_search: VectorSearch = VectorSearch( + algorithms=[algorithm_configuration] ) azure_search: AzureSearch = AzureSearch( azure_search_endpoint=AZURE_SEARCH_ENDPOINT, @@ -83,7 +88,7 @@ def __init__(self): index_name=AZURE_SEARCH_INDEX_NAME, embedding_function=embedding_function, fields=fields, - algorithm_configurations=[algorithm] + vector_search=vector_search ) azure_search_client: SearchClient = SearchClient( endpoint=AZURE_SEARCH_ENDPOINT, diff --git a/tools/sdk-ai-bots/Embeddings/requirements.txt b/tools/sdk-ai-bots/Embeddings/requirements.txt index fae8977c477..abc1ce1bb34 100644 --- a/tools/sdk-ai-bots/Embeddings/requirements.txt +++ b/tools/sdk-ai-bots/Embeddings/requirements.txt @@ -1,6 +1,6 @@ azure-identity==1.13.0 -azure-search-documents==11.4.0b8 -langchain==0.1.0 +azure-search-documents==11.6.0b2 +langchain==0.1.8 openai==0.27.8 tiktoken==0.4.0 python-dotenv==0.19.1 \ No newline at end of file