feat: Migrate vertex ai session service to use agent engine sdk

PiperOrigin-RevId: 816259798
2026-03-30 10:57:20 -07:00 · 2025-10-07 09:57:53 -07:00
parent 3be9cd844c
commit 90d4c19c51
3 changed files with 427 additions and 375 deletions
@@ -105,8 +105,8 @@ def get_fast_api_app(
      agent_engine_id = agent_engine_id_or_resource_name.split("/")[-1]
    else:
      envs.load_dotenv_for_agent("", agents_dir)
-      project = os.environ["GOOGLE_CLOUD_PROJECT"]
-      location = os.environ["GOOGLE_CLOUD_LOCATION"]
+      project = os.environ.get("GOOGLE_CLOUD_PROJECT", None)
+      location = os.environ.get("GOOGLE_CLOUD_LOCATION", None)
      agent_engine_id = agent_engine_id_or_resource_name
    return project, location, agent_engine_id