如何在Python中使用Google的Gen AI多轮搜索API

Question

目前我正在使用 Google Discovery 引擎常规搜索，但我想使用这种多轮搜索，它可以保存我之前查询的上下文，但我无法找到有关如何在 python 中实现它的任何文档，链接：https:// /cloud.google.com/generative-ai-app-builder/docs/multi-turn-search

我目前的实施：

def genappbuilder_search(
        search_query: str = "",
        project_id: str = project_id,
        location: str = location,
        search_engine_id: str = search_engine_id,
        serving_config_id: str = serving_config_id,
) -> List[discoveryengine.SearchResponse.SearchResult]:

    client = discoveryengine.SearchServiceClient()

    serving_config = client.serving_config_path(
        project=project_id,
        location=location,
        data_store=search_engine_id,
        serving_config=serving_config_id,
    )

    request = discoveryengine.SearchRequest(
        serving_config=serving_config, query=search_query, page_size=5
    )

    response = client.search(request)
    return response.results

Answer 1

您可以使用此存储库中的对话笔记本在 Python 中使用多轮（对话）搜索。

https://github.com/rasalt/snippets_es/

官方代码示例应该很快就会添加到文档中。