Integre o MongoDB com o Google ADK

Definir configurações

MongoDbToolSettings é uma classe de configurações que configura o modelo de incorporação, nomes de índice e limites de pesquisa que o MongoDbToolset utiliza. Passe uma instância dessa classe para o parâmetro settings de MongoDbToolset para substituir seus padrões, conforme mostrado no exemplo a seguir:

from google.adk.integrations.mongodb import MongoDbToolsetfrom google.adk.integrations.mongodb import MongoDbToolSettings
toolset = MongoDbToolset(    connection_string="<connection_string>",    database_name="products_db",    settings=MongoDbToolSettings(default_limit=8, max_results=20),)

Incorporando modos

Antes de executar uma pesquisa vetorial, o conjunto de ferramentas incorpora seu texto de query. Selecione a aba Atlas Automated Embedding para permitir que o Atlas incorpore queries automaticamente ou selecione a aba Google embedding models para visualizar o comportamento de incorporação padrão:

from google.adk.integrations.mongodb import MongoDbToolsetfrom google.adk.integrations.mongodb import MongoDbToolSettings
toolset = MongoDbToolset(    connection_string="<connection_string>",    database_name="products_db",    settings=MongoDbToolSettings(        use_mongodb_auto_embedding=True,        mongodb_auto_embedding_model="voyage-4",  # optional    ),)

from google.adk.integrations.mongodb import MongoDbToolset
toolset = MongoDbToolset(    connection_string="<connection_string>",    database_name="products_db",)

Escolha da coleção

O modelo escolhe a collection em cada chamada, portanto, um conjunto de ferramentas vinculado a um banco de dados acessa cada collection nesse banco de dados que tenha um índice de pesquisa}. Defina o parâmetro database_name como um banco de dados que contenha somente dados que o agente tem permissão para ler e armazene quaisquer outros dados em um banco de dados separado.

O parâmetro tool_filter não restringe quais collections o modelo pode acessar, pois seleciona ferramentas inteiras e não recebe argumentos de chamada. Para impor uma lista de permissões de coleção, use o gancho before_tool_callback, que recebe os argumentos de chamada:

from google.adk.agents import Agentfrom google.adk.integrations.mongodb import MongoDbToolset

def allow_products_only(tool, args, tool_context):  if tool.name.startswith("mongodb_") and args.get("collection_name") != "products":    return {        "status": "ERROR",        "error_details": "Only the products collection may be searched.",    }  return None

agent = Agent(    name="product_search_agent",    instruction="Answer questions about products.",    tools=[        MongoDbToolset(            connection_string="<connection_string>",            database_name="products_db",        )    ],    before_tool_callback=allow_products_only,)

Se a chamada de resposta retornar um dicionário, o ADK nunca executará a ferramenta de pesquisa. Em vez disso, ele envia esse dicionário para o agente como se fosse a saída da própria ferramenta. Retornar None permite que a ferramenta de pesquisa seja executada normalmente.

Disposição do documento

MongoDbSessionService armazena dados nas seguintes coleções:

O Google ADK define o estado do aplicativo, do usuário ou da sessão atual com um prefixo app: ou user: na chave de estado. MongoDbSessionService armazena cada valor na coleção correspondente e remove o prefixo da chave armazenada, de modo que um valor definido com a chave app:theme seja armazenado na coleção app_states como theme.

Modelo de simultaneidade

O modelo de simultaneidade do serviço depende do seu tipo de sistema:

• Em um conjunto de réplicas, cluster fragmentado ou cluster do Atlas , cada escrita de vários documentos é executada em uma transação, portanto, ela é confirmada ou cancelada atomicamente.

• Em um mongod autônomo , ou em qualquer sistema sem suporte a transações, o serviço retorna às gravações sequenciais com uma verificação de revisão de concorrência otimista no documento da sessão.

Em um único processo, um bloqueio por sessão impede gravações simultâneas na mesma sessão. Esse bloqueio não tem efeito em vários processos, como quando várias instâncias do seu aplicação gravam na mesma sessão ao mesmo tempo.

Em vários processos, a verificação de revisão do documento da sessão evita a corrupção de dados. Se duas gravações competirem, a escrita que conclui primeiro aumenta o número de revisão e a outra escrita falhará na verificação de revisão. Seu aplicação deve detectar essa falha, recarregar a sessão para obter o estado atual e tentar novamente a gravação.

Considerações

Ao usar o serviço de memória, considere o seguinte comportamento:

• Quando você pesquisa a memória, o serviço retorna documentos cuja array de palavras-chave intersecciona as palavras-chave da query.

• A ingestão é idempotente. Se você consumir novamente uma sessão, o serviço substituirá suas memória em vez de duplicá-las.

• A recuperação realiza uma pesquisa por palavra-chave e não requer um modelo de incorporação, credenciais do Google ou um índice vetorial. Como resultado, a recuperação de memória funciona em qualquer sistema MongoDB .