Guida alla migrazione dell'SDK Agent Runtime

Nella versione v1.112.0 dell'SDK Vertex AI, il agent_engines modulo all'interno dell'SDK Vertex AI Python è stato sottoposto a refactoring per un design basato su client. La pagina descrive le modifiche principali al modulo e come eseguire la migrazione del codice esistente al design basato su client. Per informazioni generali su Agent Runtime, vedi Panoramica.

Modifiche principali

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

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

viene sostituito da

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

I seguenti spazi dei nomi per Agent Runtime nell'SDK Vertex AI sono in fase di ritiro. Utilizza gli spazi dei nomi equivalenti dell'SDK Vertex AI basato su client, che ha una parità di funzionalità completa con i moduli e i pacchetti ritirati.

Spazio dei nomi dell'SDK Vertex AI	Codice interessato	Sostituzione dell'SDK Vertex AI basato su client
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.agent_engines.create client.agent_engines.get client.agent_engines.list client.agent_engines.update client.agent_engines.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.agent_engines.memories.create client.agent_engines.memories.delete client.agent_engines.memories.generate client.agent_engines.memories.get client.agent_engines.memories.list client.agent_engines.memories.retrieve client.agent_engines.sessions.create client.agent_engines.sessions.delete client.agent_engines.sessions.get client.agent_engines.sessions.list client.agent_engines.sessions.events.append client.agent_engines.sessions.events.list

Eseguire la migrazione al design basato su client

Questa sezione contiene snippet di codice che mostrano come eseguire la migrazione del codice Agent Runtime esistente al design basato su client. Nota: gli esempi potrebbero omettere importazioni, dipendenze e altro codice boilerplate per migliorare la leggibilità.

Creare 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 vertexai
client = vertexai.Client(
  project=PROJECT,
  location=LOCATION,
)
client.agent_engines.create(
    agent=local_agent,
    config={
        "staging_bucket": STAGING_BUCKET,
        "requirements": REQUIREMENTS,
        "extra_packages": EXTRA_PACKAGES,
        # ...
    },
)

Aggiornare 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 vertexai
client = vertexai.Client(
  project=PROJECT,
  location=LOCATION,
)
client.agent_engines.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 vertexai
client = vertexai.Client(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = client.agent_engines.get(name=resource_name)

Elencare le 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 vertexai
client = vertexai.Client(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = client.agent_engines.list()

Eliminare un'istanza di Agent Runtime

Prima

agent_engine.delete(
    force=True,  # Optional
)

In alternativa, py 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

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

In alternativa, py import vertexai client = vertexai.Client( project=PROJECT, location=LOCATION, ) agent_engine = client.agent_engines.delete( name=resource_name, # Required. force=True, # Optional. )