在 Azure AI 搜索中创建矢量查询

项目
09/28/2024

在 Azure AI 搜索中，如果你有矢量索引，可以参阅本文来了解如何执行以下操作：

查询矢量字段
一次查询多个矢量字段
设置矢量权重
使用集成矢量化进行查询
设置阈值以排除低分结果（预览版）

本文使用 REST 进行说明。有关其他语言的代码示例，请参阅 azure-search-vector-samples GitHub 存储库，了解包含矢量查询的端到端解决方案。

也可以在 Azure 门户中使用搜索资源管理器。

先决条件

在任何区域内、任何层级上的 Azure AI 搜索。
Azure AI 搜索上的矢量索引。若要确认矢量索引，请检查你的索引中是否有 vectorSearch 部分。
（可选）向索引中添加矢量器，以便在查询期间进行内置文本到矢量转换或图像到矢量转换。
如果要自行运行这些示例，使用 Visual Studio Code 以及 REST 客户端和示例数据。若要开始使用 REST 客户端，请参阅快速入门：使用 REST 的 Azure AI 搜索。

将查询字符串输入转换为矢量

若要查询矢量字段，查询本身必须是矢量。

将用户的文本查询字符串转换为矢量表示的一种方法是在应用程序代码中调用嵌入库或 API。最佳做法是始终使用与在源文档中生成嵌入时相同的嵌入模型。可以在 azure-search-vector-samples 存储库中找到显示如何生成嵌入的代码示例。

另一种方法是使用集成矢量化（目前已正式发布），让 Azure AI 搜索处理查询矢量化输入和输出。

下面是提交到 Azure OpenAI 嵌入模型部署的查询字符串的 REST API 示例：

POST https://{{openai-service-name}}.openai.azure.com/openai/deployments/{{openai-deployment-name}}/embeddings?api-version={{openai-api-version}}
Content-Type: application/json
api-key: {{admin-api-key}}
{
    "input": "what azure services support generative AI'"
}

如果成功调用了部署的模型，则预期的响应为 202。

响应正文中的“embedding”字段是查询字符串“input”的矢量表示形式。要进行测试，可以使用接下来几个部分中所示的语法将“embedding”数组的值复制到查询请求中的“vectorQueries.vector”。

对已部署模型的这种 POST 调用的实际响应包含 1536 个嵌入，为了便于阅读，此处修剪了响应内容，仅包含前几个矢量。

{
    "object": "list",
    "data": [
        {
            "object": "embedding",
            "index": 0,
            "embedding": [
                -0.009171937,
                0.018715322,
                ...
                -0.0016804502
            ]
        }
    ],
    "model": "ada",
    "usage": {
        "prompt_tokens": 7,
        "total_tokens": 7
    }
}

在此方法中，应用程序代码负责连接到模型、生成嵌入和处理响应。

矢量查询请求

本部分介绍矢量查询的基本结构。可以使用 Azure 门户、REST API 或 Azure SDK 来构建矢量查询。如果要从 2023-07-01-Preview 迁移，则存在中断性变更。有关详细信息，请参阅升级到最新的 REST API。

2024-07-01 是搜索 POST 的稳定 REST API 版本。此版本支持：

vectorQueries 是矢量搜索的构造。
对于矢量数组，vectorQueries.kind 设置为 vector；如果输入是字符串且你有矢量器，则其设置为 text。
vectorQueries.vector 是查询（文本或图像的矢量表示形式）。
vectorQueries.weight（可选）指定搜索操作中包含的每个矢量查询的相对权重（请参阅矢量权重）。
exhaustive（可选）在查询时调用穷举 KNN，即使字段已根据 HNSW 编制了索引。

在以下示例中，矢量是字符串“what Azure services support full text search”的表示形式。查询面向 contentVector 字段。查询返回 k 结果。实际矢量有 1536 个嵌入，因此在本示例中对其进行了修剪以便于阅读。

POST https://{{search-service-name}}.search.azure.cn/indexes/{{index-name}}/docs/search?api-version=2024-07-01
Content-Type: application/json
api-key: {{admin-api-key}}
{
    "count": true,
    "select": "title, content, category",
    "vectorQueries": [
        {
            "kind": "vector",
            "vector": [
                -0.009154141,
                0.018708462,
                . . . 
                -0.02178128,
                -0.00086512347
            ],
            "exhaustive": true,
            "fields": "contentVector",
            "weight": 0.5,
            "k": 5
        }
    ]
}

2024-05-01-preview 是搜索 - POST 的最新预览 API 版本。它支持与 2024-07-01 相同的矢量查询语法，具有用于混合搜索的额外参数和用于排除较弱结果的最小阈值。

此预览添加了：

用于排除低分结果的 threshold。
能够更好地控制混合查询输入的 Hybridsearch.MaxTextRecallSize。

POST https://{{search-service-name}}.search.azure.cn/indexes/{{index-name}}/docs/search?api-version=2024-05-01-preview
Content-Type: application/json
api-key: {{admin-api-key}}
{
    "count": true,
    "select": "title, content, category",
    "hybridSearch": {
        "maxTextRecallSize": 100,
        "countAndFacetMode": "countAllResults"
        }
    "vectorQueries": [
        {
            "kind": "vector",
            "vector": [
                -0.009154141,
                0.018708462,
                . . . 
                -0.02178128,
                -0.00086512347
            ],
            "fields": "contentVector",
            "k": 5,
            "exhaustive": true,
            "weight": 2,
            "threshold": {
                "kind": "vectorSimilarity",
                "value": 0.8
            },

        }
    ]

矢量查询响应

在 Azure AI 搜索中，查询响应默认包含所有 retrievable 字段。但是，通常通过将搜索结果列在 select 语句中，将搜索结果限制为 retrievable 字段的子集。

在矢量查询中，请仔细考虑是否需要在响应中使用矢量字段。矢量字段不是人类可读的，因此如果要将响应推送到网页，应选择能代表结果的非矢量字段。例如，如果查询针对 contentVector 执行，则可以改为返回 content。

如果确实希望在结果中包含矢量字段，下面是一个响应结构示例。 contentVector 是嵌入的字符串数组，为了简洁起见，此处进行了剪裁。搜索分数指示相关性。其他非矢量字段也包括在内，以提供上下文。

{
    "@odata.count": 3,
    "value": [
        {
            "@search.score": 0.80025613,
            "title": "Azure Search",
            "category": "AI + Machine Learning",
            "contentVector": [
                -0.0018343845,
                0.017952163,
                0.0025753193,
                ...
            ]
        },
        {
            "@search.score": 0.78856903,
            "title": "Azure Application Insights",
            "category": "Management + Governance",
            "contentVector": [
                -0.016821077,
                0.0037742127,
                0.016136652,
                ...
            ]
        },
        {
            "@search.score": 0.78650564,
            "title": "Azure Media Services",
            "category": "Media",
            "contentVector": [
                -0.025449317,
                0.0038463024,
                -0.02488436,
                ...
            ]
        }
    ]
}

要点：

k 决定返回的最近的邻域结果数，在本例中为 3。假设至少有 k 个文档存在，那么矢量查询总是返回 k 个结果，即使存在相似性较差的文档也是如此，因为该算法会找到查询矢量的任何 k 个最近的邻域。
@search.score 由矢量搜索算法确定。
搜索结果中的字段要么是所有 retrievable 字段，要么是 select 子句中的字段。在执行矢量查询时，仅根据矢量数据进行匹配。但是，响应可以包含索引中的任何 retrievable 字段。由于目前没有用于解码矢量字段结果的机制，所以包含非矢量文本字段对于展示易于人类理解的值非常有帮助。

多个矢量字段

可以将“vectorQueries.fields”属性设置为多个矢量字段。矢量查询针对你在 fields 列表中提供的每个矢量字段执行。在查询多个矢量字段时，确保每个矢量字段都包含来自同一嵌入模型的嵌入，并且查询也是由同一嵌入模型生成的。

POST https://{{search-service-name}}.search.azure.cn/indexes/{{index-name}}/docs/search?api-version=2024-07-01
Content-Type: application/json
api-key: {{admin-api-key}}
{
    "count": true,
    "select": "title, content, category",
    "vectorQueries": [
        {
            "kind": "vector",
            "vector": [
                -0.009154141,
                0.018708462,
                . . . 
                -0.02178128,
                -0.00086512347
            ],
            "exhaustive": true,
            "fields": "contentVector, titleVector",
            "k": 5
        }
    ]
}

多个矢量查询

多查询矢量搜索跨搜索索引中的多个矢量字段发送多个查询。此查询请求的一个常见示例是，将 CLIP 等模型用于多模式矢量搜索，其中同一模型可以向量化图像和文本内容。

以下查询示例在 myImageVector 和 myTextVector中查找相似性，但分别在两个不同的查询嵌入中发送，每个查询都并行执行。此查询生成使用倒数排名融合 (RRF) 评分的结果。

vectorQueries 提供矢量查询的数组。
vector 包含搜索索引中的图像矢量和文本矢量。每个实例都是单独的查询。
fields 指定要针对的矢量字段。
k 是要包含在结果中的最近邻域匹配项的数量。

{
    "count": true,
    "select": "title, content, category",
    "vectorQueries": [
        {
            "kind": "vector",
            "vector": [
                -0.009154141,
                0.018708462,
                . . . 
                -0.02178128,
                -0.00086512347
            ],
            "fields": "myimagevector",
            "k": 5
        },
        {
            "kind": "vector"
            "vector": [
                -0.002222222,
                0.018708462,
                -0.013770515,
            . . .
            ],
            "fields": "mytextvector",
            "k": 5
        }
    ]
}

假设搜索索引包含图像文件的字段（搜索索引不存储图像），搜索结果将包括文本和图像的组合。

使用集成矢量化进行查询

本节会介绍调用集成矢量化将文本转换为矢量的矢量查询。对于此功能，建议使用稳定的 2024-07-01 REST API、搜索资源管理器或较新的 Azure SDK 包。

先决条件是提供一个搜索索引，其中配置了矢量化器，并且该矢量化器已分配到矢量字段。矢量化器为查询时使用的嵌入模型提供连接信息。

Azure 门户
REST API

搜索资源管理器支持在查询时进行集成矢量化。如果索引包含矢量字段且有矢量器，则可以使用内置文本到矢量转换。

使用 Azure 帐户登录到 Azure 门户，然后转到 Azure AI 搜索服务。
在左侧菜单中，展开“搜索管理”>“索引”，然后选择索引。在索引页上，“搜索资源管理器”是第一个选项卡。
检查“矢量配置文件”以确认你有矢量器。
在搜索资源管理器中，可以在查询视图的默认搜索栏中输入文本字符串。内置矢量器将字符串转换为矢量，执行搜索并返回结果。

或者，你可以选择“视图”>“JSON 视图”来查看或修改查询。如果存在矢量，搜索资源管理器则会自动设置矢量查询。可以使用 JSON 视图选择搜索中使用的字段，并在响应中添加筛选器，或构造更高级的查询，例如混合查询。本部分的 REST API 选项卡中提供了一个 JSON 示例。

使用索引 - GET 返回索引定义并检查是否存在矢量器配置。在索引定义中查找 vectorizers。它应指定已部署的嵌入模型。
针对查询请求使用搜索 - POST。
- kind 必须设置为 text。
- text 必须有一个文本字符串。它将被传递给分配到矢量字段的矢量化器。
- fields 是要搜索的矢量字段。
- k 是要返回的矢量匹配项的数量。

下面是在查询时矢量化的查询的简单示例。文本字符串被矢量化，然后用于查询 descriptionVector 字段。

POST https://{{search-service}}.search.azure.cn/indexes/{{index}}/docs/search?api-version=2024-07-01
{
    "select": "title, genre, description",
    "vectorQueries": [
        {
            "kind": "text",
            "text": "mystery novel set in London",
            "fields": "descriptionVector",
            "k": 5
        }
    ]
}

下面是使用文本查询的集成矢量化的混合查询。此查询包括多个查询矢量字段、多个非矢量字段、一个筛选器和语义排名。同样，区别在于矢量查询的 kind，以及包含 text 字符串而不是 vector。

在本例中，搜索引擎对索引中分配给 descriptionVector、synopsisVector 和 authorBioVector 的矢量化程序进行了三次矢量化调用。生成的矢量用于根据相应字段检索文档。搜索引擎还对 search 查询“mystery novel set in London”执行了关键词搜索。

POST https://{{search-service}}.search.azure.cn/indexes/{{index}}/docs/search?api-version=2024-07-01
Content-Type: application/json
api-key: {{admin-api-key}}
{
    "search":"mystery novel set in London", 
    "searchFields":"description, synopsis", 
    "semanticConfiguration":"my-semantic-config", 
    "queryType":"semantic",
    "select": "title, author, synopsis",
    "filter": "genre eq 'mystery'",
    "vectorFilterMode": "postFilter",
    "vectorQueries": [
        {
            "kind": "text",
            "text": "mystery novel set in London",
            "fields": "descriptionVector, synopsisVector",
            "k": 5
        },
        {
            "kind": "text"
            "text": "living english author",
            "fields": "authorBioVector",
            "k": 5
        }
    ]
}

所有四个查询的评分结果均使用 RRF 排名进行融合。

注意

矢量化在索引编制和查询期间发生。如果索引中不需要数据分块和矢量化，则可以跳过创建索引器、技能组和数据源等步骤。在此工作流中，矢量化仅在查询时使用，以将文本字符串或图像转换为嵌入。可以针对此步骤在搜索索引中定义矢量器。

矢量查询响应中的排名结果数

矢量查询指定 k 参数，该参数确定在结果中返回多少个匹配项。搜索引擎始终返回 k 个匹配项。如果 k 大于索引中的文档数，则文档数决定了可以返回的上限。

如果你熟悉全文搜索的话，就会知道，当索引不包含要搜索的字词或短语时，预期会返回零个结果。但是在矢量搜索中，搜索操作标识最近的邻域，即使最近的邻域不是那么相似，它也始终返回 k 个结果。因此，对于无意义或偏离主题的查询，尤其是当你未使用提示设置边界时，可以获得结果。相关性较低的结果的相似性评分较差，但如果没有更接近的矢量，它们仍然是“最接近的”矢量。因此，不包含有意义结果的响应仍可能返回 k 个结果，但每个结果的相似性评分很低。

包含全文搜索的混合方法可以缓解此问题。另一种缓解方法是对搜索分数设置最小阈值，但仅限于查询是纯单矢量查询时。混合查询不适合最小阈值，因为 RRF 范围要小得多且不稳定。

影响结果计数的查询参数包括：

对于仅矢量查询，使用 "k": n 个结果
对于包含“search”参数的混合查询，使用 "top": n 个结果

“k”和“top”都是可选的。如果未指定，响应中的默认结果数为 50。可以设置“top”和“skip”以浏览更多结果或更改默认值。

矢量查询中使用的排名算法

结果排名是根据下列其中一项计算的：

相似性指标
如果存在多个搜索结果集，则采用倒数排名融合 (RRF)。

相似性指标

在仅矢量查询的索引 vectorSearch 节中指定的相似性指标。有效值为：cosine、euclidean 和 dotProduct。

Azure OpenAI 嵌入模型使用余弦相似性，因此如果你使用 Azure OpenAI 嵌入模型，则 cosine 是建议的指标。其他支持的排名指标包括 euclidean 和 dotProduct。

使用 RRF

在查询执行期间，矢量查询只能针对一个内部矢量索引。因此，对于多个矢量字段和多个矢量查询，搜索引擎会生成多个查询，它们针对每个字段的相应矢量索引。输出是为每个查询返回的一个排名结果集，这些结果使用 RRF 进行融合。有关详细信息，请参阅使用倒数排序融合 (RRF) 进行相关性评分。

矢量权重

添加 weight 查询参数来指定搜索操作中包含的每个矢量查询的相对权重。当合并由同一请求中的两个或多个矢量查询生成的多个排名列表的结果，或者合并来自混合查询矢量部分的结果时，将使用此值。

默认值为 1.0，该值必须是大于零的正数。

可使用权重来计算每个文档的倒数排序融合分数。计算结果是 weight 值与文档在其各自结果集中的排名分数的乘积。

以下示例是一个混合查询，其中包含两个矢量查询字符串和一个文本字符串。权重分配给矢量查询。第一个查询是 0.5 或一半的权重，其在请求中的重要性有所减少。第二个矢量查询的重要性达到两倍。

文本查询没有权重参数，但可通过设置 maxTextRecallSize 来增加或减少其重要性。

POST https://[service-name].search.azure.cn/indexes/[index-name]/docs/search?api-version=2024-07-01

    { 
      "vectorQueries": [ 
        { 
          "kind": "vector", 
          "vector": [1.0, 2.0, 3.0], 
          "fields": "my_first_vector_field", 
          "k": 10, 
          "weight": 0.5 
        },
        { 
          "kind": "vector", 
          "vector": [4.0, 5.0, 6.0], 
          "fields": "my_second_vector_field", 
          "k": 10, 
          "weight": 2.0
        } 
      ], 
      "search": "hello world" 
    }

设置阈值以排除低分结果（预览版）

由于最近的邻域搜索始终返回所请求的 k 邻域，因此，为了满足搜索结果的 k 数量要求，可能会显示多个低分匹配项。若要排除低分搜索结果，可以添加 threshold 查询参数，根据最低分数筛选出结果。筛选后将从不同的调用集中融合结果。

此参数仍以预览版提供。建议使用预览版 REST API 版本 2024-05-01-preview。

在此示例中，即使结果数低于 k，也会从矢量搜索结果中排除分数低于 0.8 的所有匹配项。

POST https://[service-name].search.azure.cn/indexes/[index-name]/docs/search?api-version=2024-05-01-preview 
    Content-Type: application/json 
    api-key: [admin key] 

    { 
      "vectorQueries": [ 
        { 
          "kind": "vector", 
          "vector": [1.0, 2.0, 3.0], 
          "fields": "my-cosine-field", 
          "threshold": { 
            "kind": "vectorSimilarity", 
            "value": 0.8 
          } 
        }
      ]
    }

混合搜索的 MaxTextSizeRecall（预览版）

矢量查询通常用于包含非矢量字段的混合构造中。如果发现 BM25 排名结果在混合查询结果中表现得过高或过低，则可以设置 maxTextRecallSize来增加或减少为混合排名提供的 BM25 排名结果。

只能在同时包含“search”和“vectorQueries”组成部分的混合请求中设置此属性。