agent-starter-pack 0.5.1__py3-none-any.whl → 0.5.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (23) hide show
  1. {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/METADATA +2 -2
  2. {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/RECORD +23 -23
  3. agents/adk_base/template/.templateconfig.yaml +1 -1
  4. agents/agentic_rag/template/.templateconfig.yaml +1 -1
  5. agents/crewai_coding_crew/template/.templateconfig.yaml +1 -1
  6. agents/langgraph_base_react/template/.templateconfig.yaml +1 -1
  7. src/base_template/README.md +1 -1
  8. src/data_ingestion/data_ingestion_pipeline/components/process_data.py +1 -1
  9. src/deployment_targets/agent_engine/app/agent_engine_app.py +17 -5
  10. src/deployment_targets/cloud_run/app/server.py +11 -2
  11. src/resources/locks/uv-adk_base-agent_engine.lock +360 -369
  12. src/resources/locks/uv-adk_base-cloud_run.lock +453 -459
  13. src/resources/locks/uv-agentic_rag-agent_engine.lock +623 -631
  14. src/resources/locks/uv-agentic_rag-cloud_run.lock +849 -849
  15. src/resources/locks/uv-crewai_coding_crew-agent_engine.lock +873 -744
  16. src/resources/locks/uv-crewai_coding_crew-cloud_run.lock +1126 -960
  17. src/resources/locks/uv-langgraph_base_react-agent_engine.lock +716 -593
  18. src/resources/locks/uv-langgraph_base_react-cloud_run.lock +949 -800
  19. src/resources/locks/uv-live_api-cloud_run.lock +759 -697
  20. {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/WHEEL +0 -0
  21. {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/entry_points.txt +0 -0
  22. {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/licenses/LICENSE +0 -0
  23. /src/{deployment_targets/agent_engine → base_template}/app/utils/gcs.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: agent-starter-pack
3
- Version: 0.5.1
3
+ Version: 0.5.3
4
4
  Summary: CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates.
5
5
  Author-email: Google LLC <agent-starter-pack@google.com>
6
6
  License: Apache-2.0
@@ -9,7 +9,7 @@ Requires-Python: >=3.10
9
9
  Requires-Dist: backoff>=2.2.1
10
10
  Requires-Dist: click~=8.1.7
11
11
  Requires-Dist: cookiecutter~=2.5.0
12
- Requires-Dist: google-cloud-aiplatform~=1.87.0
12
+ Requires-Dist: google-cloud-aiplatform>=1.87.0
13
13
  Requires-Dist: pyyaml~=6.0.1
14
14
  Requires-Dist: rich~=13.7.0
15
15
  Provides-Extra: jupyter
@@ -2,7 +2,7 @@ agents/adk_base/README.md,sha256=eLf-F9Z4u_mQOchAnUaSrOAHXRSx8wUQN29kkwz1SF4,894
2
2
  agents/adk_base/app/agent.py,sha256=Kt-0TQM4g9NLs7Q2KmRWPfPWWgyQEODVo0uCepLkqDw,2182
3
3
  agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=WTxrxu-bMfjqR8SvwZxNuR2_4Qo20QUZ7F3aBCRxtA0,9613
4
4
  agents/adk_base/notebooks/evaluating_adk_agent.ipynb,sha256=OMTiT3gAFWPvE9pPL8z7tsDwOU-l8wem5wH-OXPPlvk,56805
5
- agents/adk_base/template/.templateconfig.yaml,sha256=BT88Q8Bri6DYwv2HHjkpcL2mik8u8Gkwi3YV9_XW7PU,893
5
+ agents/adk_base/template/.templateconfig.yaml,sha256=d6f_11cA0m6VMUzVPXETSoCViQe5b49ImzJAlkhskHw,893
6
6
  agents/adk_base/tests/integration/test_agent.py,sha256=VypgpEWrQegJBgc58V-BlIsbqh-Hii-hwmkwcz5feFw,1969
7
7
  agents/agentic_rag/README.md,sha256=mxMLkq6tAghFaDmc2sc2sEQ5vn9kGWdiRxfTykwpLz0,2268
8
8
  agents/agentic_rag/app/agent.py,sha256=thaU0njJ3W90DI54EVrFe4lwx9yMWZT4jSGRMnUSdY8,3987
@@ -10,7 +10,7 @@ agents/agentic_rag/app/retrievers.py,sha256=qfxwSEixMddvAu4FvUTZAFh1Gdp3RPY65s0W
10
10
  agents/agentic_rag/app/templates.py,sha256=8N_uRchwiN7hlH9Y8rdJ1CAtOhpHUWZkXZZ7IU-YJJk,867
11
11
  agents/agentic_rag/notebooks/adk_app_testing.ipynb,sha256=WTxrxu-bMfjqR8SvwZxNuR2_4Qo20QUZ7F3aBCRxtA0,9613
12
12
  agents/agentic_rag/notebooks/evaluating_adk_agent.ipynb,sha256=OMTiT3gAFWPvE9pPL8z7tsDwOU-l8wem5wH-OXPPlvk,56805
13
- agents/agentic_rag/template/.templateconfig.yaml,sha256=fTsHlqDrmMcOA_yNWYAiQJoLMOQm21JeQMnxO1gYs1Q,1223
13
+ agents/agentic_rag/template/.templateconfig.yaml,sha256=k86nWTpBRnQPe3i9LS_KD1gK6vGXzllAlCHI6u4gQ0w,1223
14
14
  agents/agentic_rag/tests/integration/test_agent.py,sha256=mxsKFyZrwXzIvAoxlqZv2JndM3VTVTKcugRYdF3APr4,2112
15
15
  agents/crewai_coding_crew/README.md,sha256=4No7sfHQVHELARGiT3fWANJzDR0NP48ow4ct3FoPwlA,1672
16
16
  agents/crewai_coding_crew/app/agent.py,sha256=axa-prJ7N8Gi8ASUSjokLeWCE-Z_cU0A99WwuLGsZyQ,3264
@@ -19,12 +19,12 @@ agents/crewai_coding_crew/app/crew/config/agents.yaml,sha256=Cn9zTcP3C-8MvlV19cY
19
19
  agents/crewai_coding_crew/app/crew/config/tasks.yaml,sha256=SSFuiSQPhi4dM0V5HdNuMqtN7JJr5RUCMhhBzUkPBpc,1511
20
20
  agents/crewai_coding_crew/notebooks/evaluating_crewai_agent.ipynb,sha256=ifTn4RfrJeYQ8ell9n6VE65ZEfcnKGkMLFG8RBlPK70,57974
21
21
  agents/crewai_coding_crew/notebooks/evaluating_langgraph_agent.ipynb,sha256=PlW41fL-ji9NKkimW6u25YRLGWN2pgoX2G8w5nzIEPE,58737
22
- agents/crewai_coding_crew/template/.templateconfig.yaml,sha256=YCWXOcXzK1UJfFGGzNn5wz8tdIECsSP2N9uxpG1LVfA,1052
22
+ agents/crewai_coding_crew/template/.templateconfig.yaml,sha256=4SLcpHgDcf_6mjJCxPVciQ002QVz09V04uvbtMLZs4A,1051
23
23
  agents/crewai_coding_crew/tests/integration/test_agent.py,sha256=0uvfmM3FSdESIU7ICb2oqABsQqylQU2PE3UBwbpif-A,1629
24
24
  agents/langgraph_base_react/README.md,sha256=KZQ0e8q9CqnIEiItH9Tc2Ceb2mC3sBcyQUo-ZAmO-zs,482
25
25
  agents/langgraph_base_react/app/agent.py,sha256=srxiOT-NIM87WnUvP0PXuZGryEG9S33Z_4orHl3D_0A,2509
26
26
  agents/langgraph_base_react/notebooks/evaluating_langgraph_agent.ipynb,sha256=PlW41fL-ji9NKkimW6u25YRLGWN2pgoX2G8w5nzIEPE,58737
27
- agents/langgraph_base_react/template/.templateconfig.yaml,sha256=NE_EJzZPtdbJRFdp5k9fovc6zUA6Wvm1MBUBvaZ6gV0,1050
27
+ agents/langgraph_base_react/template/.templateconfig.yaml,sha256=jHh3WDNT70KMbA2zy3Da_J_jIWeGfeiqIKfUs5B8OBk,1049
28
28
  agents/langgraph_base_react/tests/integration/test_agent.py,sha256=wg2Y8V4l7dh7rhWy7Z8zioDW5ILRsz7RZJ-mz0AhK3U,1619
29
29
  agents/live_api/README.md,sha256=c3lu5WguOUBc5hO_UtNZfVhfYxvSzMXuGFICn9M38mU,2897
30
30
  agents/live_api/app/agent.py,sha256=FBsmCYI_dmBIbFD1uojcPxJfozGNc-0rg1fW2PhEyFw,2298
@@ -35,9 +35,10 @@ agents/live_api/tests/load_test/load_test.py,sha256=HHZyfC4gqiQtZVF_CbbxENGgWQcc
35
35
  agents/live_api/tests/unit/test_server.py,sha256=_TjlgQgNkjerIaBGnu8P8_KB8ZlSolDcivALpUOn_Rw,4786
36
36
  src/base_template/.gitignore,sha256=mJKTZIcVdAFiIUQicRfPNGUg6WvwcfTEC2xbmAaU34g,2579
37
37
  src/base_template/Makefile,sha256=O65aYaRQlCOZHfCaAdWbK3eEobw5x83A8nEjK88jtpQ,4538
38
- src/base_template/README.md,sha256=Bg-wux7NGMAhx1-enAn87dfMUchTi8D7-Gkx7TWDcs4,10119
38
+ src/base_template/README.md,sha256=nZFLjJwau3DJoJVhHOItlv4SkG7QFiROm5ZCygvFTgw,10119
39
39
  src/base_template/pyproject.toml,sha256=FmD5qLIWjMTjT6euiUegSw8PLeGvxO1bnSg-r0UjkJU,2959
40
40
  src/base_template/app/__init__.py,sha256=UyAgc2l8nkVIUPUzL9hKR7EgzCsc2jSYcOcjHNNmpMU,59
41
+ src/base_template/app/utils/gcs.py,sha256=jKblaWOGQEigw3esaawcfX178shhZi2Fyk0fJSA4T68,1501
41
42
  src/base_template/app/utils/tracing.py,sha256=2rv1Ukh2jTBENDwoghCItJ28l-Sjz9gMlzdojlVgJa4,6052
42
43
  src/base_template/app/utils/typing.py,sha256=DP5OZC3IGvqA1XbvWt8kI3gyAK3ZjzUSL5Ca17wNeLI,4249
43
44
  src/base_template/deployment/README.md,sha256=qWTCWyEj1bZuny3XyCelXPgljh6yu_GDwfPNUeFoXBY,5245
@@ -78,10 +79,9 @@ src/data_ingestion/uv.lock,sha256=HzSD6_IxS2urt49EefD9MvVxBwxW_bJ-k0XltTDT3Vc,14
78
79
  src/data_ingestion/data_ingestion_pipeline/pipeline.py,sha256=UAS6dxV954yARP0NdDbCf5kzap3NfmoX52eZ2mZtWXs,3331
79
80
  src/data_ingestion/data_ingestion_pipeline/submit_pipeline.py,sha256=tWGqL0zUB3V5JxR1SpX7LMjCW5r6JEGnvgF8tAxZ12s,7784
80
81
  src/data_ingestion/data_ingestion_pipeline/components/ingest_data.py,sha256=QbPmk6V7JPGo5Yc1BhL-1W_t9CXEN0iY5NKaGWlfSbM,10667
81
- src/data_ingestion/data_ingestion_pipeline/components/process_data.py,sha256=3D4CaUV9pBaIU10MogJMQAR00JPVzqjFHN6eL7WM_5Y,22045
82
+ src/data_ingestion/data_ingestion_pipeline/components/process_data.py,sha256=8vubxkykRpx-d89H1XYGO3yXz4hxhihhGDgDR3s-XZ4,22045
82
83
  src/deployment_targets/agent_engine/deployment_metadata.json,sha256=G_t_n-UNrFsBgH1Xrw5-ZqYzUGwZ4X6ipsIm_yiSq-w,72
83
- src/deployment_targets/agent_engine/app/agent_engine_app.py,sha256=j5xK6AQYyJmRubISxfgTXcbsV_4DvCSswJqRJTm2a-0,11825
84
- src/deployment_targets/agent_engine/app/utils/gcs.py,sha256=jKblaWOGQEigw3esaawcfX178shhZi2Fyk0fJSA4T68,1501
84
+ src/deployment_targets/agent_engine/app/agent_engine_app.py,sha256=Wpke3ClK5HHbPvIPrGfFf6MV30YQaSykxTX-RUOeDm8,12270
85
85
  src/deployment_targets/agent_engine/notebooks/intro_agent_engine.ipynb,sha256=RE6Gp-bu4bHtNBngfkt_CF5emy5YcOWQ5rMYW_Cv8_w,48809
86
86
  src/deployment_targets/agent_engine/tests/integration/test_agent_engine_app.py,sha256=3mEHdnenqT1RrQpndnQWZSMnn8kSLH85rPDhbxH6_lk,5978
87
87
  src/deployment_targets/agent_engine/tests/load_test/README.md,sha256=qaVE9qR0MvZcaABwf4GS6cAxi7un5ql_vh-zbgh3LmA,1443
@@ -89,7 +89,7 @@ src/deployment_targets/agent_engine/tests/load_test/load_test.py,sha256=FFCo_LjI
89
89
  src/deployment_targets/agent_engine/tests/load_test/.results/.placeholder,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
90
90
  src/deployment_targets/cloud_run/Dockerfile,sha256=vedZX0JJW2bymESqB-0LumnkKyr7hcog8NlR0DX8Q74,843
91
91
  src/deployment_targets/cloud_run/uv.lock,sha256=2RYrR89xDS1FHVb-g2JGZq8zZ-rzbm_jRvehVCYuXRQ,887957
92
- src/deployment_targets/cloud_run/app/server.py,sha256=6xgR99amWepWlOP-uENJzENA7p3ekFl93ivl7-PtG7E,4793
92
+ src/deployment_targets/cloud_run/app/server.py,sha256=EFPdzu7K1KvV6J4WpVbYzs-6j3-QszVcGeGXeiBhTL8,5122
93
93
  src/deployment_targets/cloud_run/tests/integration/test_server_e2e.py,sha256=dKgFE8ksZz-6IDqMfJjTJcwgywJcN9gpSd9Ni97NjPg,7784
94
94
  src/deployment_targets/cloud_run/tests/load_test/README.md,sha256=HgWizsGHyh_2i0xOG2-1IN_Eav3GiGNKwZVOffilo2I,2806
95
95
  src/deployment_targets/cloud_run/tests/load_test/load_test.py,sha256=cOlGaNxx6X3C-eWD0LulXyggqTrTeWZPjnsVn7FtZIs,4082
@@ -154,23 +154,23 @@ src/resources/containers/e2e-tests/Dockerfile,sha256=Q_aTyX_iaFY8j06XZkpMuggJnNO
154
154
  src/resources/idx/idx-template.json,sha256=07OQZCPp45Iqor2O7Tm1e1Gmsdd-AmmUofSvA7y-oRs,793
155
155
  src/resources/idx/idx-template.nix,sha256=sesHGev_PYtVDg0J5tHkg0OO7IR1Bz2iAtl_if3Ar3M,892
156
156
  src/resources/idx/.idx/dev.nix,sha256=XQ4T1xt5I-8SVwnI37v56RFFlxa58Ko-a9G2-oelYAI,1602
157
- src/resources/locks/uv-adk_base-agent_engine.lock,sha256=5C0qM0h4QtYmFJKLq3PFddfSzjQWFEjOiO82vHxAUdQ,354900
158
- src/resources/locks/uv-adk_base-cloud_run.lock,sha256=1IGYBSh0vFV2zgxKkv97RsfSffrO1eqTgqxXPGanRT8,439843
159
- src/resources/locks/uv-agentic_rag-agent_engine.lock,sha256=bUYvGPpldusxjM59ZmkiGx5TD0uVYR_q6Ae4zX_RDMc,520793
160
- src/resources/locks/uv-agentic_rag-cloud_run.lock,sha256=jf_-4Fy3Ru-8Sv1FT2wezM68VpJza4mY9vnnASJRRYg,676939
161
- src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=D7ChXJvgpv9tx04oySxtKNYV7c75fFmgjkHkfY-9t7c,752871
162
- src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=gAo8kIqxFo7ce9d3gtgvjqj4uW5qWdyA-rfSFw8wPGM,954380
163
- src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=4MaAtU7WsOBcr2sKFQxFLxBAI3sbmTItcwheO--juIY,641971
164
- src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=LZChkvjJas5yiajNLar3ejaZ8ZmdzwPuyKZ4mAXtrT8,832580
165
- src/resources/locks/uv-live_api-cloud_run.lock,sha256=FJ7T5-oVajY_zjLSHlq5IUVDMg0SKy6g7bwvlMLueSs,773265
157
+ src/resources/locks/uv-adk_base-agent_engine.lock,sha256=hPplULv_naT_erWwTmbg0Y3EFh1wZ_ZglOZ44jPB7_s,352622
158
+ src/resources/locks/uv-adk_base-cloud_run.lock,sha256=idBIkXAchi2j-a-tngAoSXmNvu9fiuQlwNCMJIMMYcI,433449
159
+ src/resources/locks/uv-agentic_rag-agent_engine.lock,sha256=aOekEzuougaxd108vLUonodT2ypJ75NJFjJPN1Wptdo,518546
160
+ src/resources/locks/uv-agentic_rag-cloud_run.lock,sha256=_vMiI-0buziUak0_WRQo4a3Sxye0dBL-r1Ob44fBnRE,671999
161
+ src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=XdHrWOj0AjJz1KLXyw2YQprtJAmds-wZ_IKKmeRbotc,778953
162
+ src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=b7_Df5UYJDDNWUtqWNuJGxLkc3phCzLkDfLpzYLGwls,990359
163
+ src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=fi_wRRdrAP7VQtGDSG-XSnC_aM68zQxf-S6g7tLqd5g,666366
164
+ src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=73Ni3DG0WPhZnXnJRk5M5Ul7S9XXAmL9ZNTsY2sWNwc,863721
165
+ src/resources/locks/uv-live_api-cloud_run.lock,sha256=mixMuU4nEEZpFPzr-mv8b8NItjmYt_gDT4C9Pf2OKqc,785851
166
166
  src/resources/setup_cicd/cicd_variables.tf,sha256=PMflYe1TzQi63LORHkmeCktTYzXFplJgxffNH4DtuAQ,1244
167
167
  src/resources/setup_cicd/github.tf,sha256=scTBgeZlCM74N-pzhVKsnTN0PX9a5GboNl1HN3-LlCM,2791
168
168
  src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5jB7QBfF0,1122
169
169
  src/utils/generate_locks.py,sha256=6V1B8V2BEuevWnXUsxZVTrLjXwFRII8UfsIGrQqZxVs,4320
170
170
  src/utils/lock_utils.py,sha256=RSE6n3hBkH64xNm3Q5wrR0Pqbeo-oc7xaTOhA9yzHjk,2275
171
171
  src/utils/watch_and_rebuild.py,sha256=vP4yIiA7E_lj5sfQdJUl8TXas6V7msDg8XWUutAC05Q,6679
172
- agent_starter_pack-0.5.1.dist-info/METADATA,sha256=2GCZDXrJusKMveFpA8rFwssXfaekQfwPsRZzM5G587o,10454
173
- agent_starter_pack-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
174
- agent_starter_pack-0.5.1.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
175
- agent_starter_pack-0.5.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
176
- agent_starter_pack-0.5.1.dist-info/RECORD,,
172
+ agent_starter_pack-0.5.3.dist-info/METADATA,sha256=xLUqZpFRecNARUHlGZV8EvdFtFtoRPQKz-g265-KMO4,10454
173
+ agent_starter_pack-0.5.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
174
+ agent_starter_pack-0.5.3.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
175
+ agent_starter_pack-0.5.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
176
+ agent_starter_pack-0.5.3.dist-info/RECORD,,
@@ -17,6 +17,6 @@ example_question: "What's the weather in San Francisco?"
17
17
  settings:
18
18
  requires_data_ingestion: false
19
19
  deployment_targets: ["agent_engine", "cloud_run"]
20
- extra_dependencies: ["google-adk~=1.1.0"]
20
+ extra_dependencies: ["google-adk~=1.3.0"]
21
21
  tags: ["adk"]
22
22
  frontend_type: "None"
@@ -18,7 +18,7 @@ settings:
18
18
  requires_data_ingestion: true
19
19
  deployment_targets: ["agent_engine", "cloud_run"]
20
20
  extra_dependencies: [
21
- "google-adk~=1.1.0",
21
+ "google-adk~=1.3.0",
22
22
  "langchain-google-vertexai~=2.0.7",
23
23
  "langchain~=0.3.24",
24
24
  "langchain-core~=0.3.55",
@@ -21,7 +21,7 @@ settings:
21
21
  "langchain~=0.3.14",
22
22
  "langchain-community~=0.3.17",
23
23
  "langchain-openai~=0.3.5",
24
- "langgraph~=0.2.63",
24
+ "langgraph~=0.4.8",
25
25
  "crewai~=0.119.0"
26
26
  ]
27
27
  example_question: "How can I implement a function to sort a list in Python?"
@@ -19,7 +19,7 @@ settings:
19
19
  extra_dependencies: [
20
20
  "langchain-google-vertexai~=2.0.7",
21
21
  "langchain~=0.3.14",
22
- "langgraph~=0.2.63",
22
+ "langgraph~=0.4.8",
23
23
  "langchain-google-vertexai~=2.0.22",
24
24
  "langchain~=0.3.14",
25
25
  "langchain-community~=0.3.17",
@@ -49,7 +49,7 @@ make install && make playground
49
49
  {%- if cookiecutter.deployment_target == 'cloud_run' %}
50
50
  | `make playground` | Launch local development environment with backend and frontend{%- if "adk" in cookiecutter.tags %} - leveraging `adk web` command. {%- endif %}|
51
51
  | `make backend` | Deploy agent to Cloud Run |
52
- | `make local-backend` | Deploy agent to Cloud Run |
52
+ | `make local-backend` | Launch local development server |
53
53
  {%- if cookiecutter.deployment_target == 'cloud_run' %}
54
54
  {%- if cookiecutter.agent_name == 'live_api' %}
55
55
  | `make ui` | Launch Agent Playground front-end only |
@@ -15,7 +15,7 @@
15
15
 
16
16
  """
17
17
  This component is derived from the notebook:
18
- https://github.com/GoogleCloudPlatform/generative-ai/blob/main/gemini/use-cases/retrieval-augmented_generation/scalable_rag_with_bigframes.ipynb
18
+ https://github.com/GoogleCloudPlatform/generative-ai/blob/main/gemini/use-cases/retrieval-augmented-generation/scalable_rag_with_bigframes.ipynb
19
19
 
20
20
  It leverages BigQuery for data processing. We also suggest looking at remote functions for enhanced scalability.
21
21
  """
@@ -24,6 +24,7 @@ from typing import Any
24
24
 
25
25
  import google.auth
26
26
  import vertexai
27
+ from google.adk.artifacts import GcsArtifactService
27
28
  from google.cloud import logging as google_cloud_logging
28
29
  from opentelemetry import trace
29
30
  from opentelemetry.sdk.trace import TracerProvider, export
@@ -210,18 +211,29 @@ def deploy_agent_engine_app(
210
211
  ) -> agent_engines.AgentEngine:
211
212
  """Deploy the agent engine app to Vertex AI."""
212
213
 
213
- staging_bucket = f"gs://{project}-agent-engine"
214
-
214
+ staging_bucket_uri = f"gs://{project}-agent-engine"
215
+ {%- if "adk" in cookiecutter.tags %}
216
+ artifacts_bucket_name = f"{project}-{{cookiecutter.project_name}}-logs-data"
215
217
  create_bucket_if_not_exists(
216
- bucket_name=staging_bucket, project=project, location=location
218
+ bucket_name=artifacts_bucket_name, project=project, location=location
217
219
  )
218
- vertexai.init(project=project, location=location, staging_bucket=staging_bucket)
220
+ {%- endif %}
221
+ create_bucket_if_not_exists(
222
+ bucket_name=staging_bucket_uri, project=project, location=location
223
+ )
224
+
225
+ vertexai.init(project=project, location=location, staging_bucket=staging_bucket_uri)
219
226
 
220
227
  # Read requirements
221
228
  with open(requirements_file) as f:
222
229
  requirements = f.read().strip().split("\n")
223
230
  {% if "adk" in cookiecutter.tags %}
224
- agent_engine = AgentEngineApp(agent=root_agent)
231
+ agent_engine = AgentEngineApp(
232
+ agent=root_agent,
233
+ artifact_service_builder=lambda: GcsArtifactService(
234
+ bucket_name=artifacts_bucket_name
235
+ ),
236
+ )
225
237
  {% else %}
226
238
  agent_engine = AgentEngineApp(project_id=project)
227
239
  {% endif %}
@@ -14,26 +14,35 @@
14
14
  {% if "adk" in cookiecutter.tags %}
15
15
  import os
16
16
 
17
+ import google.auth
17
18
  from fastapi import FastAPI
18
19
  from google.adk.cli.fast_api import get_fast_api_app
19
20
  from google.cloud import logging as google_cloud_logging
20
21
  from opentelemetry import trace
21
22
  from opentelemetry.sdk.trace import TracerProvider, export
22
23
 
24
+ from app.utils.gcs import create_bucket_if_not_exists
23
25
  from app.utils.tracing import CloudTraceLoggingSpanExporter
24
26
  from app.utils.typing import Feedback
25
27
 
28
+ _, project_id = google.auth.default()
26
29
  logging_client = google_cloud_logging.Client()
27
30
  logger = logging_client.logger(__name__)
28
31
 
32
+ bucket_name = f"gs://{project_id}-{{cookiecutter.project_name}}-logs-data"
33
+ create_bucket_if_not_exists(
34
+ bucket_name=bucket_name, project=project_id, location="us-central1"
35
+ )
36
+
29
37
  provider = TracerProvider()
30
38
  processor = export.BatchSpanProcessor(CloudTraceLoggingSpanExporter())
31
39
  provider.add_span_processor(processor)
32
40
  trace.set_tracer_provider(provider)
33
41
 
34
42
  AGENT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
35
- app: FastAPI = get_fast_api_app(agents_dir=AGENT_DIR, web=True)
36
-
43
+ app: FastAPI = get_fast_api_app(
44
+ agents_dir=AGENT_DIR, web=True, artifact_service_uri=bucket_name
45
+ )
37
46
  app.title = "{{cookiecutter.project_name}}"
38
47
  app.description = "API for interacting with the Agent {{cookiecutter.project_name}}"
39
48
  {%- else %}