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.
- {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/METADATA +2 -2
- {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/RECORD +23 -23
- agents/adk_base/template/.templateconfig.yaml +1 -1
- agents/agentic_rag/template/.templateconfig.yaml +1 -1
- agents/crewai_coding_crew/template/.templateconfig.yaml +1 -1
- agents/langgraph_base_react/template/.templateconfig.yaml +1 -1
- src/base_template/README.md +1 -1
- src/data_ingestion/data_ingestion_pipeline/components/process_data.py +1 -1
- src/deployment_targets/agent_engine/app/agent_engine_app.py +17 -5
- src/deployment_targets/cloud_run/app/server.py +11 -2
- src/resources/locks/uv-adk_base-agent_engine.lock +360 -369
- src/resources/locks/uv-adk_base-cloud_run.lock +453 -459
- src/resources/locks/uv-agentic_rag-agent_engine.lock +623 -631
- src/resources/locks/uv-agentic_rag-cloud_run.lock +849 -849
- src/resources/locks/uv-crewai_coding_crew-agent_engine.lock +873 -744
- src/resources/locks/uv-crewai_coding_crew-cloud_run.lock +1126 -960
- src/resources/locks/uv-langgraph_base_react-agent_engine.lock +716 -593
- src/resources/locks/uv-langgraph_base_react-cloud_run.lock +949 -800
- src/resources/locks/uv-live_api-cloud_run.lock +759 -697
- {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/WHEEL +0 -0
- {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/entry_points.txt +0 -0
- {agent_starter_pack-0.5.1.dist-info → agent_starter_pack-0.5.3.dist-info}/licenses/LICENSE +0 -0
- /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.
|
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
|
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=
|
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=
|
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=
|
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=
|
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=
|
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=
|
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=
|
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=
|
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=
|
158
|
-
src/resources/locks/uv-adk_base-cloud_run.lock,sha256=
|
159
|
-
src/resources/locks/uv-agentic_rag-agent_engine.lock,sha256=
|
160
|
-
src/resources/locks/uv-agentic_rag-cloud_run.lock,sha256=
|
161
|
-
src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=
|
162
|
-
src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=
|
163
|
-
src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=
|
164
|
-
src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=
|
165
|
-
src/resources/locks/uv-live_api-cloud_run.lock,sha256=
|
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.
|
173
|
-
agent_starter_pack-0.5.
|
174
|
-
agent_starter_pack-0.5.
|
175
|
-
agent_starter_pack-0.5.
|
176
|
-
agent_starter_pack-0.5.
|
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.
|
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.
|
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",
|
src/base_template/README.md
CHANGED
@@ -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`
|
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-
|
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
|
-
|
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=
|
218
|
+
bucket_name=artifacts_bucket_name, project=project, location=location
|
217
219
|
)
|
218
|
-
|
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(
|
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(
|
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 %}
|