agent-starter-pack 0.3.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

{agent_starter_pack-0.3.1.dist-info → agent_starter_pack-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.3.1
+Version: 0.3.2
 Summary: CLI tool to create GCP-based AI agent projects from templates
 Author-email: Google LLC <agent-starter-pack@google.com>
 License: Apache-2.0
@@ -70,7 +70,7 @@ agent-starter-pack create my-agent -d agent_engine -a adk_base
 
 | Agent Name                  | Description                                                                                                                       |
 |-----------------------------|-----------------------------------------------------------------------------------------------------------------------------------|
-| `adk_base`      | An agent implementing a base ReAct agent using Google's [Agent Development Kit](https://github.com/google/adk-python)
+| `adk_base`      | A base ReAct agent implemented using Google's [Agent Development Kit](https://github.com/google/adk-python) |
 | `agentic_rag` | A RAG agent for document retrieval and Q&A. Supporting [Vertex AI Search](https://cloud.google.com/generative-ai-app-builder/docs/enterprise-search-introduction) and [Vector Search](https://cloud.google.com/vertex-ai/docs/vector-search/overview).       |
 | `langgraph_base_react`      | An agent implementing a base ReAct agent using LangGraph |
 | `crewai_coding_crew`       | A multi-agent system implemented with CrewAI created to support coding activities       |

{agent_starter_pack-0.3.1.dist-info → agent_starter_pack-0.3.2.dist-info}/RECORD

@@ -1,13 +1,13 @@
-agents/adk_base/README.md,sha256=4_VlWQI0HU_rvaKZ59Qs7P5VYklWoEJr9pSV-m_BADc,852
+agents/adk_base/README.md,sha256=eLf-F9Z4u_mQOchAnUaSrOAHXRSx8wUQN29kkwz1SF4,894
 agents/adk_base/app/agent.py,sha256=Dm7dcELj0RbALNI20R9-uEay_RHgej0dPOdQRHA7E-8,2187
-agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=7qZmhssPoJCfT0DA2WpT3uwHgLBIu1EGoFyUDkjaVio,8198
-agents/adk_base/template/.templateconfig.yaml,sha256=a0GO7kgM2g5fxPW_KZtdfBS8l4Sl0p7G_vzAvKQXtv0,847
+agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=sHaW9s1PiclJILKBK5j-W6nx_-g60UVnvITKgZFraME,8591
+agents/adk_base/template/.templateconfig.yaml,sha256=m8Qkm1maDIQRDk5GmJlk7_tZT_1ZCn7Cxzo00TzMYmI,836
 agents/adk_base/tests/integration/test_agent.py,sha256=1VyshBu44p_PUAQQxV0QROmv6OUGt_m0nDLyaTI77po,1964
 agents/agentic_rag/README.md,sha256=mxMLkq6tAghFaDmc2sc2sEQ5vn9kGWdiRxfTykwpLz0,2268
 agents/agentic_rag/app/agent.py,sha256=EkbMIjEeyXqK49wIkoW6uWotxyXCIfJpeLCizw2QDZM,3959
 agents/agentic_rag/app/retrievers.py,sha256=qfxwSEixMddvAu4FvUTZAFh1Gdp3RPY65s0Wp6gEFVk,4561
 agents/agentic_rag/app/templates.py,sha256=8N_uRchwiN7hlH9Y8rdJ1CAtOhpHUWZkXZZ7IU-YJJk,867
-agents/agentic_rag/notebooks/adk_app_testing.ipynb,sha256=7qZmhssPoJCfT0DA2WpT3uwHgLBIu1EGoFyUDkjaVio,8198
+agents/agentic_rag/notebooks/adk_app_testing.ipynb,sha256=sHaW9s1PiclJILKBK5j-W6nx_-g60UVnvITKgZFraME,8591
 agents/agentic_rag/template/.templateconfig.yaml,sha256=6wznpR27bt_brW0DCT7rq_J5AaLVo6M--dcoG_uFtsU,1111
 agents/agentic_rag/tests/integration/test_agent.py,sha256=HbEv2gzC77jrse3RZ0ra_Bcu6zW_7YlrxYWcYfei63M,2107
 agents/crewai_coding_crew/README.md,sha256=4No7sfHQVHELARGiT3fWANJzDR0NP48ow4ct3FoPwlA,1672
@@ -80,7 +80,7 @@ src/data_ingestion/data_ingestion_pipeline/submit_pipeline.py,sha256=tWGqL0zUB3V
 src/data_ingestion/data_ingestion_pipeline/components/ingest_data.py,sha256=QbPmk6V7JPGo5Yc1BhL-1W_t9CXEN0iY5NKaGWlfSbM,10667
 src/data_ingestion/data_ingestion_pipeline/components/process_data.py,sha256=3D4CaUV9pBaIU10MogJMQAR00JPVzqjFHN6eL7WM_5Y,22045
 src/deployment_targets/agent_engine/deployment_metadata.json,sha256=G_t_n-UNrFsBgH1Xrw5-ZqYzUGwZ4X6ipsIm_yiSq-w,72
-src/deployment_targets/agent_engine/app/agent_engine_app.py,sha256=hRzFSAjaxMJzahJ0CYypWjV1SMAFok3UsAly4jMScTk,11334
+src/deployment_targets/agent_engine/app/agent_engine_app.py,sha256=ZBER9DVyVfG6R4IyukgFaVVL-AsbqLmvvBsXAwgv2jk,12032
 src/deployment_targets/agent_engine/app/utils/gcs.py,sha256=jKblaWOGQEigw3esaawcfX178shhZi2Fyk0fJSA4T68,1501
 src/deployment_targets/agent_engine/notebooks/intro_agent_engine.ipynb,sha256=RE6Gp-bu4bHtNBngfkt_CF5emy5YcOWQ5rMYW_Cv8_w,48809
 src/deployment_targets/agent_engine/tests/integration/test_agent_engine_app.py,sha256=3mEHdnenqT1RrQpndnQWZSMnn8kSLH85rPDhbxH6_lk,5978
@@ -166,8 +166,8 @@ src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5
 src/utils/generate_locks.py,sha256=xu5IAhGGBPkVQGSJX4kk7_JNDwWJUEaXAHbmaQIohbg,4386
 src/utils/lock_utils.py,sha256=_QdzQtgIbCmJ87s046_i1g966slVNmvr3bJDeHbRQSM,2419
 src/utils/watch_and_rebuild.py,sha256=vP4yIiA7E_lj5sfQdJUl8TXas6V7msDg8XWUutAC05Q,6679
-agent_starter_pack-0.3.1.dist-info/METADATA,sha256=-LVL1SWUzRiDRsyD5ZYC0gsNtEa0GJAVrdkerxBhXoY,8026
-agent_starter_pack-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.3.1.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.3.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.3.1.dist-info/RECORD,,
+agent_starter_pack-0.3.2.dist-info/METADATA,sha256=JDgsTYFzAC-s_GsOXtQe6IcaUsUnZoeQFvI5hUCE1R4,8018
+agent_starter_pack-0.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.3.2.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.3.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.3.2.dist-info/RECORD,,

agents/adk_base/README.md

@@ -1,6 +1,6 @@
 # ADK: Minimal Agent Example
 
-[![ADK Logo](./../../docs/images/adk_logo.png)](https://google.github.io/adk-docs/)
+<img src="https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/images/adk_logo.png?raw=true" width="400">
 
 A basic agent built using the **[Google Agent Development Kit (ADK)](https://google.github.io/adk-docs/)**. This example demonstrates core ADK concepts like agent creation and tool integration in a minimal setup.
 

agents/adk_base/notebooks/adk_app_testing.ipynb

@@ -106,6 +106,22 @@
     "    print(event)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "remote_agent_engine.register_feedback(\n",
+    "    feedback={\n",
+    "        \"score\": 5,\n",
+    "        \"text\": \"Great response!\",\n",
+    "        \"invocation_id\": \"test-invocation-123\",\n",
+    "        \"user_id\": \"test\",\n",
+    "    }\n",
+    ")"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},

agents/adk_base/template/.templateconfig.yaml

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-description: "An agent implementing a base ReAct agent using Google's Agent Development Kit"
+description: "A base ReAct agent built with Google's Agent Development Kit (ADK)"
 settings:
   requires_data_ingestion: false
   deployment_targets: ["agent_engine", "cloud_run"]

agents/agentic_rag/notebooks/adk_app_testing.ipynb

@@ -106,6 +106,22 @@
     "    print(event)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "remote_agent_engine.register_feedback(\n",
+    "    feedback={\n",
+    "        \"score\": 5,\n",
+    "        \"text\": \"Great response!\",\n",
+    "        \"invocation_id\": \"test-invocation-123\",\n",
+    "        \"user_id\": \"test\",\n",
+    "    }\n",
+    ")"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},

src/deployment_targets/agent_engine/app/agent_engine_app.py

@@ -14,9 +14,11 @@
 
 # mypy: disable-error-code="attr-defined"
 {%- if "adk" in cookiecutter.tags %}
+import copy
 import datetime
 import json
 import logging
+import os
 from collections.abc import Mapping, Sequence
 from typing import Any
 
@@ -41,7 +43,11 @@ class AgentEngineApp(AdkApp):
         logging_client = google_cloud_logging.Client()
         self.logger = logging_client.logger(__name__)
         provider = TracerProvider()
-        processor = export.BatchSpanProcessor(CloudTraceLoggingSpanExporter())
+        processor = export.BatchSpanProcessor(
+            CloudTraceLoggingSpanExporter(
+                project_id=os.environ.get("GOOGLE_CLOUD_PROJECT")
+            )
+        )
         provider.add_span_processor(processor)
         trace.set_tracer_provider(provider)
 
@@ -58,6 +64,19 @@ class AgentEngineApp(AdkApp):
         operations = super().register_operations()
         operations[""] = operations[""] + ["register_feedback"]
         return operations
+
+    def clone(self) -> "AgentEngineApp":
+        """Returns a clone of the ADK application."""
+        template_attributes = self._tmpl_attrs
+        return self.__class__(
+            agent=copy.deepcopy(template_attributes.get("agent")),
+            enable_tracing=template_attributes.get("enable_tracing"),
+            session_service_builder=template_attributes.get("session_service_builder"),
+            artifact_service_builder=template_attributes.get(
+                "artifact_service_builder"
+            ),
+            env_vars=template_attributes.get("env_vars"),
+        )
 {%- else %}
 import datetime
 import json
@@ -211,7 +230,8 @@ def deploy_agent_engine_app(
         requirements = f.read().strip().split("\n")
 {% if "adk" in cookiecutter.tags %}
     agent_engine = AgentEngineApp(
-        agent=root_agent, env_vars=env_vars, enable_tracing=True
+        agent=root_agent,
+        env_vars=env_vars,
     )
 {% else %}
     agent_engine = AgentEngineApp(project_id=project, env_vars=env_vars)