Agent Engine 이전 가이드

Vertex AI SDK 버전 v1.112.0에서 Python용 Vertex AI SDK 내의 agent_engines 모듈이 클라이언트 기반 설계로 리팩터링되었습니다.

이 페이지에서는 모듈의 주요 변경사항과 클라이언트 기반 설계로 기존 코드를 이전하는 방법을 설명합니다.

Agent Engine에 대한 일반 정보는 개요를 참고하세요.

주요 변경사항

개략적으로 서비스 클라이언트 매개변수는 클라이언트별로 초기화되며 클라이언트에는 서비스 상호작용과 관련된 모듈이 포함됩니다.

다음이 이에 해당됩니다.

import vertexai
from vertexai import agent_engines

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

다음으로 대체됩니다.

import vertexai

client = vertexai.Client(project=GCP_PROJECT, location=GCP_REGION)
client.agent_engines.create(...)

Vertex AI SDK의 Vertex AI Agent Engine에 대한 다음 네임스페이스는 지원 중단 단계에 있습니다. 지원 중단된 모듈 및 패키지와 기능 패리티가 완전히 동일한 클라이언트 기반 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 Engine 코드를 클라이언트 기반 설계로 이전하는 방법을 보여주는 코드 스니펫이 포함되어 있습니다.

참고: 가독성을 높이기 위해 예시에서 가져오기, 종속 항목, 기타 상용구 코드가 생략될 수 있습니다.

Agent Engine 인스턴스 만들기

이전

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 Engine 인스턴스 업데이트

이전

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 Engine 인스턴스 가져오기

이전

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 Engine 인스턴스 나열

이전

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

에이전트 엔진 인스턴스 삭제

이전

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