Agent Runtime SDK 마이그레이션 가이드

Vertex AI SDK 버전 v1.112.0에서 Vertex AI SDK for Python 내의 agent_engines 모듈이 클라이언트 기반 설계로 리팩터링되었습니다. 이 페이지에서는 모듈의 주요 변경사항과 클라이언트 기반 설계로 기존 코드를 마이그레이션하는 방법을 설명합니다. 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의 동등한 네임스페이스를 사용하세요.

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