Agent Runtime SDK 移行ガイド

Vertex AI SDK のバージョン v1.112.0 では、agent_engines モジュールが Vertex AI SDK for Python 内で クライアントベースの設計にリファクタリングされました。 このページでは、モジュールの主な変更点と、クライアントベースの設計に既存のコードを移行する方法について説明します。 Agent Runtime の一般的な情報については、概要をご覧ください。

主な変更点

全体としては、サービス クライアント パラメータはクライアントごとに初期化され、クライアントにはサービス インタラクションに関連するモジュールが含まれます。 まとめると次のようになります。

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

次のもので置き換えられます。

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(...)

Vertex AI SDK の Agent Runtime の次の名前空間は、非推奨フェーズに入っています。クライアント ベースの Vertex AI SDK の同等の名前空間を使用します。この SDK には、非推奨のモジュールおよびパッケージと同等の機能がすべて備わっています。

Vertex AI SDK の名前空間	影響を受けるコード	クライアントベースの Vertex AI SDK の後継
vertexai.agent_engines	影響を受けるメソッド: vertexai.agent_engines.create vertexai.agent_engines.get vertexai.agent_engines.list vertexai.agent_engines.update vertexai.agent_engines.delete	後継: client.agent_engines.create client.agent_engines.get client.agent_engines.list client.agent_engines.update client.agent_engines.delete
client.agent_engines	影響を受けるメソッド: 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	後継: 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

クライアントベースの設計に移行する

このセクションでは、既存の Agent Runtime コードをクライアント ベースの設計に移行する方法を示すコード スニペットを紹介します。 注: 読みやすさを向上させるため、例ではインポート、依存関係、その他のボイラープレート コードが省略されている場合があります。

Agent Runtime インスタンスの作成

これより前

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,
    # ...
)

これから

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,
        # ...
    },
)

Agent Runtime インスタンスの更新

これより前

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,
    # ...
)

これから

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,
        # ...
    },
)

Agent Runtime インスタンスの取得

これより前

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

これから

import vertexai
client = vertexai.Client(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = client.agent_engines.get(name=resource_name)

Agent Runtime インスタンスの一覧表示

これより前

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

これから

import vertexai
client = vertexai.Client(
  project=PROJECT,
  location=LOCATION,
)
agent_engine = client.agent_engines.list()

Agent Runtime インスタンスの削除

これより前

agent_engine.delete(
    force=True,  # Optional
)

または、 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 )

これから

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

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