encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat: Migrate vertex_ai_session_service to use Agent Engine SDK

Browse Source 
PiperOrigin-RevId: 814948921
This commit is contained in:
Shangjie Chen
2025-10-03 21:00:15 -07:00
committed by Copybara-Service
parent 960eda3d1f
commit 97b950b36b
3 changed files with 421 additions and 379 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/cli/fast_api.py
+2 -2
View File
@@ -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
src/google/adk/sessions/vertex_ai_session_service.py
+169 -231
View File
File diff suppressed because it is too large Load Diff
tests/unittests/sessions/test_vertex_ai_session_service.py
+250 -146
View File
File diff suppressed because it is too large Load Diff