Manual Deploy — Agent Engine 수동 배포

"""
Agent Engine manual deploy.
요구: google-adk>=1.19, google-cloud-aiplatform[adk,agent_engines]
"""
import vertexai
from vertexai import agent_engines
from vertexai.preview.reasoning_engines import AdkApp

from google.adk.agents import LlmAgent
from google.adk.tools import agent_tool, url_context, VertexAiSearchTool
from google.adk.tools.google_search_tool import GoogleSearchTool

# ── 설정 ──
PROJECT = "<YOUR_PROJECT>"
LOCATION = "us-west1"                       # us-central1 / asia-northeast3 도 가능
STAGING_BUCKET = "gs://<staging-bucket>"

DATA_STORE = (
    "projects/<YOUR_PROJECT>/locations/global/collections/"
    "default_collection/dataStores/<YOUR_DATA_STORE_ID>"
)

vertexai.init(project=PROJECT, location=LOCATION, staging_bucket=STAGING_BUCKET)

# ── Sub-agents ──
google_search_agent = LlmAgent(
    name="google_search_agent",
    model="gemini-2.5-pro",
    description="Agent specialized in performing Google searches.",
    instruction="Use the GoogleSearchTool to find information on the web.",
    tools=[GoogleSearchTool()],
)
url_context_agent = LlmAgent(
    name="url_context_agent",
    model="gemini-2.5-pro",
    description="Agent specialized in fetching content from URLs.",
    instruction="Use the UrlContextTool to retrieve content from provided URLs.",
    tools=[url_context],
)
vertex_ai_search_agent = LlmAgent(
    name="vertex_ai_search_agent",
    model="gemini-2.5-pro",
    description="Agent specialized in performing Vertex AI Search.",
    instruction="Use the VertexAISearchTool to find information using Vertex AI Search.",
    tools=[VertexAiSearchTool(data_store_id=DATA_STORE)],
)

# ── Root agent ──
root_agent = LlmAgent(
    name="my_root_agent",
    model="gemini-2.5-pro",
    description="자동차 세일즈 에이전트 데모",
    instruction=(
        "사용자의 질문에 답변할 때 반드시 참조한 문서의 파일명과 관련 내용을 함께 명시하세요.\n"
        "1. vertex_ai_search 도구의 모든 검색 결과에는 `uri` (또는 `link`) 필드에 "
        "GCS 경로가 포함됩니다. 반드시 GCS 경로 자체를 참고로 나열하세요."
    ),
    tools=[
        agent_tool.AgentTool(agent=google_search_agent),
        agent_tool.AgentTool(agent=url_context_agent),
        agent_tool.AgentTool(agent=vertex_ai_search_agent),
    ],
)

# ── 핵심: AdkApp wrap + 최신 SDK pin ──
app = AdkApp(agent=root_agent, enable_tracing=True)

print("=== 배포 시작 (10~15분 소요) ===")
remote = agent_engines.create(
    app,
    display_name="My Agent (manual)",
    requirements=[
        "google-adk>=1.19",
        "google-cloud-aiplatform[adk,agent_engines]",
    ],
)

print()
print("=== 배포 완료 ===")
print(f"resource_name: {remote.resource_name}")

remote = agent_engines.update(
    resource_name="projects/<PROJECT_NUMBER>/locations/<LOCATION>/reasoningEngines/<ID>",
    agent_engine=app,
    requirements=[...],
)