Guida alla migrazione dell'SDK Agent Runtime

Nella versione 2.0.1 dell'SDK Agent Platform, il modulo agent_engines all'interno dell'SDK è stato sottoposto a refactoring e rinominato in runtimes nel pacchetto agentplatform autonomo. La pagina descrive le modifiche principali al modulo e come migrare il codice esistente alla nuova struttura dell'SDK. Per informazioni generali su Agent Runtime, consulta la panoramica.

Modifiche principali

A livello generale, i parametri del client di servizio vengono inizializzati in base al client e il client contiene i moduli pertinenti per le interazioni con il servizio. Vale a dire:

import vertexai
from vertexai import agent_engines
vertexai.init(project=GCP_PROJECT, location=GCP_REGION)
agent_engines.create(...)

è sostituito da

import agentplatform
# Mandatory for cross-project deployment and for use of the framework-specific templates.
client = agentplatform.Client(project=GCP_PROJECT, location=GCP_REGION)
client.runtimes.create(...)

I seguenti spazi dei nomi nell'SDK sono nella fase di ritiro. Utilizza gli spazi dei nomi equivalenti dell'SDK Agent Platform basato su client, che offre la stessa parità di funzionalità dei moduli e dei pacchetti ritirati.

Spazio dei nomi legacy	Codice interessato	Sostituzione (agentplatform)
vertexai.agent_engines	Metodi interessati: vertexai.agent_engines.create vertexai.agent_engines.get vertexai.agent_engines.list vertexai.agent_engines.update vertexai.agent_engines.delete	Sostituzione: client.runtimes.create client.runtimes.get client.runtimes.list client.runtimes.update client.runtimes.delete
client.agent_engines	Metodi interessati: client.agent_engines.create_memory client.agent_engines.delete_memory client.agent_engines.generate_memories client.agent_engines.get_memory client.agent_engines.list_memories client.agent_engines.retrieve_memories client.agent_engines.create_session client.agent_engines.delete_session client.agent_engines.get_session client.agent_engines.list_sessions client.agent_engines.append_session_event client.agent_engines.list_session_events	Sostituzione: client.memory_banks.memories.create client.memory_banks.memories.delete client.memory_banks.memories.generate client.memory_banks.memories.get client.memory_banks.memories.list client.memory_banks.memories.retrieve client.sessions.create client.sessions.delete client.sessions.get client.sessions.list client.sessions.events.append client.sessions.events.list

Eseguire la migrazione alla progettazione basata sul client

Questa sezione contiene snippet di codice che mostrano come eseguire la migrazione del codice Agent Runtime esistente alla progettazione dell'SDK agentplatform. Nota: gli esempi potrebbero omettere importazioni, dipendenze e altro codice boilerplate per migliorare la leggibilità.

Creazione di un'istanza di Agent Runtime

Prima

import vertexai
from vertexai import agent_engines
vertexai.init(
  project=PROJECT,
  location=LOCATION,
  staging_bucket=STAGING_BUCKET,
)
agent_engines.create(
    local_agent,
    requirements=REQUIREMENTS,
    extra_packages=EXTRA_PACKAGES,
    # ...
)

Dopo

import agentplatform
client = agentplatform.Client(
  project=PROJECT,
  location=LOCATION,
)
client.runtimes.create(
    agent=local_agent,
    config={
        "staging_bucket": STAGING_BUCKET,
        "requirements": REQUIREMENTS,
        "extra_packages": EXTRA_PACKAGES,
        # ...
    },
)

Aggiornamento di un'istanza di Agent Runtime

Prima

import vertexai
from vertexai import agent_engines
vertexai.init(
  project=PROJECT,
  location=LOCATION,
  staging_bucket=STAGING_BUCKET,
)
agent_engines.update(
    resource_name,
    agent_engine=local_agent,
    requirements=REQUIREMENTS,
    extra_packages=EXTRA_PACKAGES,
    # ...
)

Dopo

import agentplatform
client = agentplatform.Client(
  project=PROJECT,
  location=LOCATION,
)
client.runtimes.update(
    name=resource_name,
    agent=local_agent,
    config={
        "staging_bucket": STAGING_BUCKET,
        "requirements": REQUIREMENTS,
        "extra_packages": EXTRA_PACKAGES,
        # ...
    },
)

Ottenere un'istanza di Agent Runtime

Prima

import vertexai
from vertexai import agent_engines
vertexai.init(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = agent_engines.get(resource_name)

Dopo

import agentplatform
client = agentplatform.Client(
  project=PROJECT,
  location=LOCATION,
)
remote_agent = client.runtimes.get(name=resource_name)

Elenco delle istanze di Agent Runtime

Prima

import vertexai
from vertexai import agent_engines
vertexai.init(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = agent_engines.list()

Dopo

import agentplatform
client = agentplatform.Client(
  project=PROJECT,
  location=LOCATION,
)
runtimes = client.runtimes.list()

Eliminazione di un'istanza di Agent Runtime

Prima

agent_engine.delete(
    force=True,  # Optional
)

In alternativa,

import vertexai
from vertexai import agent_engines
vertexai.init(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = agent_engines.delete(
    resource_name,  # Required.
    force=True,  # Optional
)

Dopo

remote_agent.delete(
    force=True,     # Optional.
)

In alternativa,

import agentplatform
client = agentplatform.Client(
  project=PROJECT,
  location=LOCATION,
)
remote_agent = client.runtimes.delete(
    name=resource_name,  # Required.
    force=True,          # Optional.
)