Function main

contributing/samples/integrations/sandbox_computer_use/main.py:85–161

Main function to run the sandbox computer use agent.

()

async def main():
  """Main function to run the sandbox computer use agent."""
  # Validate environment
  project_id = os.environ.get("GOOGLE_CLOUD_PROJECT")
  service_account = os.environ.get("VMAAS_SERVICE_ACCOUNT")

  if not project_id:
    print("ERROR: GOOGLE_CLOUD_PROJECT environment variable is not set.")
    print("Please set it to your GCP project ID.")
    return

  if not service_account:
    print("ERROR: VMAAS_SERVICE_ACCOUNT environment variable is not set.")
    print(
        "Please set it to your service account email with"
        " roles/iam.serviceAccountTokenCreator permission."
    )
    return

  print("=" * 60)
  print("Sandbox Computer Use Agent Demo")
  print("=" * 60)
  print(f"Project: {project_id}")
  print(f"Service Account: {service_account}")
  print("=" * 60)

  app_name = "sandbox_computer_use_demo"
  user_id = "demo_user"

  # Create runner and session
  runner = InMemoryRunner(
      agent=agent.root_agent,
      app_name=app_name,
  )
  session = await runner.session_service.create_session(
      app_name=app_name, user_id=user_id
  )

  print(f"\nSession created: {session.id}")
  print("\nStarting agent interaction...")

  start_time = time.time()

  # Example interaction: Navigate and describe
  await run_prompt(
      runner,
      session,
      user_id,
      "Navigate to https://www.google.com and tell me what you see.",
  )

  # Example interaction: Search for something
  await run_prompt(
      runner,
      session,
      user_id,
      "Search for 'Vertex AI Agent Engine' and tell me the first result.",
  )