agent-starter-pack 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

{agent_starter_pack-0.4.0.dist-info → agent_starter_pack-0.4.2.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.4.0
-Summary: CLI tool to create GCP-based AI agent projects from templates
+Version: 0.4.2
+Summary: CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates.
 Author-email: Google LLC <agent-starter-pack@google.com>
 License: Apache-2.0
 License-File: LICENSE
@@ -38,7 +38,7 @@ Description-Content-Type: text/markdown
       alt="Try in Firebase Studio"
       src="https://cdn.firebasestudio.dev/btn/try_blue_20.svg">
   </picture>
-</a> ![Stars](https://img.shields.io/github/stars/GoogleCloudPlatform/agent-starter-pack?color=yellow)
+</a> [![Launch in Cloud Shell](https://img.shields.io/badge/Launch-in_Cloud_Shell-white)](https://shell.cloud.google.com/cloudshell/editor?cloudshell_git_repo=https%3A%2F%2Fgithub.com%2Feliasecchig%2Fasp-open-in-cloud-shell&cloudshell_print=open-in-cs) ![Stars](https://img.shields.io/github/stars/GoogleCloudPlatform/agent-starter-pack?color=yellow)
 
 
 The `agent-starter-pack` is a collection of production-ready Generative AI Agent templates built for Google Cloud. <br>
@@ -66,7 +66,7 @@ agent-starter-pack create my-awesome-agent
 ```
 
 **That's it!** You now have a fully functional agent project—complete with backend, frontend, and deployment infrastructure—ready for you to explore and customize.
-For more installation options, see the [Installation Guide](https://googlecloudplatform.github.io/agent-starter-pack/guide/installation). You can also [try it in Firebase Studio](https://studio.firebase.google.com/new?template=https%3A%2F%2Fgithub.com%2FGoogleCloudPlatform%2Fagent-starter-pack%2Ftree%2Fmain%2Fsrc%2Fresources%2Fidx) with zero setup.
+See [Installation Guide](https://googlecloudplatform.github.io/agent-starter-pack/guide/installation) for more options, or try with zero setup in [Firebase Studio](https://studio.firebase.google.com/new?template=https%3A%2F%2Fgithub.com%2FGoogleCloudPlatform%2Fagent-starter-pack%2Ftree%2Fmain%2Fsrc%2Fresources%2Fidx) or [Cloud Shell](https://shell.cloud.google.com/cloudshell/editor?cloudshell_git_repo=https%3A%2F%2Fgithub.com%2Feliasecchig%2Fasp-open-in-cloud-shell&cloudshell_print=open-in-cs).
 
 ---
 
@@ -129,13 +129,12 @@ Visit our [documentation site](https://googlecloudplatform.github.io/agent-start
 
 ### Video Walkthrough:
 
-## Video Tutorials
+- **[Exploring the Agent Starter Pack](https://www.youtube.com/watch?v=9zqwym-N3lg)**: A comprehensive tutorial demonstrating how to rapidly deploy AI Agents using the Agent Starter Pack, covering architecture, templates, and step-by-step deployment.
 
-- **April 2024**: A [6-minute introduction video](https://www.youtube.com/live/eZ-8UQ_t4YM?feature=shared&t=2791) Explaining the Agent Starter Pack and demonstrating its key features. Part of the Kaggle GenAI intensive course.
-- **March 6, 2025**: A [120 Minute livestream video demo](https://www.youtube.com/watch?v=yIRIT_EtALs&t=235s) of the new `agent-starter-pack` were we build 3 Agents under 30 minutes!
-- **Oct 29, 2024**: A [20-Minute Video Walkthrough](https://youtu.be/kwRG7cnqSu0) is available, showcasing the previous `agent-starter-pack`.
+- **[6-minute introduction](https://www.youtube.com/live/eZ-8UQ_t4YM?feature=shared&t=2791)** (April 2024): Explaining the Agent Starter Pack and demonstrating its key features. Part of the Kaggle GenAI intensive course.
+
+- **[120-minute livestream demo](https://www.youtube.com/watch?v=yIRIT_EtALs&t=235s)** (March 6, 2025): Watch us build 3 Agents in under 30 minutes using the `agent-starter-pack`!
 
-## Explore More Generative AI Resources
 
 Looking for more examples and resources for Generative AI on Google Cloud? Check out the [GoogleCloudPlatform/generative-ai](https://github.com/GoogleCloudPlatform/generative-ai) repository for notebooks, code samples, and more!
 

{agent_starter_pack-0.4.0.dist-info → agent_starter_pack-0.4.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 agents/adk_base/README.md,sha256=eLf-F9Z4u_mQOchAnUaSrOAHXRSx8wUQN29kkwz1SF4,894
 agents/adk_base/app/agent.py,sha256=Kt-0TQM4g9NLs7Q2KmRWPfPWWgyQEODVo0uCepLkqDw,2182
-agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=sHaW9s1PiclJILKBK5j-W6nx_-g60UVnvITKgZFraME,8591
+agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=WTxrxu-bMfjqR8SvwZxNuR2_4Qo20QUZ7F3aBCRxtA0,9613
 agents/adk_base/notebooks/evaluating_adk_agent.ipynb,sha256=OMTiT3gAFWPvE9pPL8z7tsDwOU-l8wem5wH-OXPPlvk,56805
 agents/adk_base/template/.templateconfig.yaml,sha256=x3mXlO-JkgXEOpbLoreJzOp78LQfhIWlV9rKAjHPwfU,893
 agents/adk_base/tests/integration/test_agent.py,sha256=1VyshBu44p_PUAQQxV0QROmv6OUGt_m0nDLyaTI77po,1964
@@ -8,7 +8,7 @@ agents/agentic_rag/README.md,sha256=mxMLkq6tAghFaDmc2sc2sEQ5vn9kGWdiRxfTykwpLz0,
 agents/agentic_rag/app/agent.py,sha256=thaU0njJ3W90DI54EVrFe4lwx9yMWZT4jSGRMnUSdY8,3987
 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=sHaW9s1PiclJILKBK5j-W6nx_-g60UVnvITKgZFraME,8591
+agents/agentic_rag/notebooks/adk_app_testing.ipynb,sha256=WTxrxu-bMfjqR8SvwZxNuR2_4Qo20QUZ7F3aBCRxtA0,9613
 agents/agentic_rag/notebooks/evaluating_adk_agent.ipynb,sha256=OMTiT3gAFWPvE9pPL8z7tsDwOU-l8wem5wH-OXPPlvk,56805
 agents/agentic_rag/template/.templateconfig.yaml,sha256=33QMS3Vi_gpYLCqxm2idbv_mZ8EMqruezzw0G-a7oyg,1223
 agents/agentic_rag/tests/integration/test_agent.py,sha256=HbEv2gzC77jrse3RZ0ra_Bcu6zW_7YlrxYWcYfei63M,2107
@@ -34,7 +34,7 @@ agents/live_api/tests/integration/test_server_e2e.py,sha256=D2VETDIyTD2fQyQ6DXwL
 agents/live_api/tests/load_test/load_test.py,sha256=HHZyfC4gqiQtZVF_CbbxENGgWQccMLpwMv0IdoQ6cbQ,1275
 agents/live_api/tests/unit/test_server.py,sha256=_TjlgQgNkjerIaBGnu8P8_KB8ZlSolDcivALpUOn_Rw,4786
 src/base_template/.gitignore,sha256=mJKTZIcVdAFiIUQicRfPNGUg6WvwcfTEC2xbmAaU34g,2579
-src/base_template/Makefile,sha256=8B4JJ_30k6J9MW-Kvr77LFErBSoHr09Us_8Nf5DIPQE,4544
+src/base_template/Makefile,sha256=lvcSmYD0S8dcQNOTKk3ZZG6_92bx8aWWDYStI4bZ8J8,4518
 src/base_template/README.md,sha256=ocMt-AFgd4vJda7ABIiR4ajhf4RxTg1jV4k7Lwq8-No,10143
 src/base_template/pyproject.toml,sha256=V2Df010o_Q8smOaS1K9fx-pVe9X4_KZT4vpIRaB0aFs,2959
 src/base_template/app/__init__.py,sha256=UyAgc2l8nkVIUPUzL9hKR7EgzCsc2jSYcOcjHNNmpMU,59
@@ -64,9 +64,9 @@ src/base_template/deployment/terraform/vars/env.tfvars,sha256=Nze8q1x2Aj6ZUeWC2h
 src/base_template/tests/unit/test_dummy.py,sha256=2exfCH8qhkZrLWvK04ZxNTO9MV3fdTbZkJN3uK6zvok,850
 src/cli/main.py,sha256=pMsSlNwkrFqHUHHA5U-WMZ4QRquaI_F7OXQt6yxuugE,1688
 src/cli/commands/create.py,sha256=2dvvkvwLAJDi1gG7SA3LXfriTY7T6fyWKeuBHHwxLIU,24117
-src/cli/commands/setup_cicd.py,sha256=usWoOPWxNbSDql6HJYKYvRue0xZd2fUZSINzMjlBBK0,32784
+src/cli/commands/setup_cicd.py,sha256=Lr0wLeKlRxROsRVFLAI3ygfIh_sTavvgQckDkzFEKMQ,31705
 src/cli/utils/__init__.py,sha256=_cTmsXGPqOtK0q8UW5164QTltbJRJFR_Efxq_BRL1-o,1311
-src/cli/utils/cicd.py,sha256=tlz50wI0unRosRZsDpZ5Oh5haMJmNjc9dV0-TPWawKE,27443
+src/cli/utils/cicd.py,sha256=m2BxwV39323rVzJCjJjB37IPdSNQdxVgXGrgnhdbb84,26040
 src/cli/utils/datastores.py,sha256=gv1V6eDcOEKx4MRNG5C3Y-VfixYq1AzQuaYMLp8QRNo,1058
 src/cli/utils/gcp.py,sha256=cnuCyN144eiyYc9aJNEK9JnyWN66rdevugoMdDYC1UU,4032
 src/cli/utils/logging.py,sha256=0lHe4EPi1A8sOx9xkA7gS4UNl0GsIyp2ahydkkuCzLY,1570
@@ -169,8 +169,8 @@ src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5
 src/utils/generate_locks.py,sha256=6V1B8V2BEuevWnXUsxZVTrLjXwFRII8UfsIGrQqZxVs,4320
 src/utils/lock_utils.py,sha256=RSE6n3hBkH64xNm3Q5wrR0Pqbeo-oc7xaTOhA9yzHjk,2275
 src/utils/watch_and_rebuild.py,sha256=vP4yIiA7E_lj5sfQdJUl8TXas6V7msDg8XWUutAC05Q,6679
-agent_starter_pack-0.4.0.dist-info/METADATA,sha256=KdGfhPZ1ppJODWICbU_dLPReKTCFCJpDr6scuxJLBsc,9991
-agent_starter_pack-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.4.0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.4.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.4.0.dist-info/RECORD,,
+agent_starter_pack-0.4.2.dist-info/METADATA,sha256=hI-T9Fi9ih7Cl32r5ng0SpqlDSH_LnBef6kYZIlDC1M,10452
+agent_starter_pack-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.4.2.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.4.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.4.2.dist-info/RECORD,,

agents/adk_base/notebooks/adk_app_testing.ipynb

@@ -16,58 +16,43 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Import libraries"
+    "## Install dependencies"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "import json\n",
-    "\n",
-    "import requests\n",
-    "import vertexai.agent_engines"
+    "!pip install google-cloud-aiplatform --upgrade"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## If you are using Agent Engine\n",
-    "See more documentation at [Agent Engine Overview](https://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/overview)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Local Testing\n",
-    "\n",
-    "You can import directly the AgentEngineApp class within your environment. "
+    "### Import libraries"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from app.agent import root_agent\n",
-    "from app.agent_engine_app import AgentEngineApp\n",
+    "import json\n",
     "\n",
-    "agent_engine = AgentEngineApp(agent=root_agent)"
+    "import requests\n",
+    "import vertexai.agent_engines"
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "metadata": {},
-   "outputs": [],
    "source": [
-    "for event in agent_engine.stream_query(message=\"hi!\", user_id=\"test\"):\n",
-    "    print(event)"
+    "## If you are using Agent Engine\n",
+    "See more documentation at [Agent Engine Overview](https://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/overview)"
    ]
   },
   {
@@ -122,6 +107,53 @@
     ")"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Local Testing\n",
+    "\n",
+    "You can import directly the AgentEngineApp class within your environment. \n",
+    "To run the agent locally, follow these steps:\n",
+    "1. Make sure all required packages are installed in your environment\n",
+    "2. The recommended approach is to use the same virtual environment created by the 'uv' tool\n",
+    "3. You can set up this environment by running 'make install' from your agent's root directory\n",
+    "4. Then select this kernel (.venv folder in your project) in your Jupyter notebook to ensure all dependencies are available"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Uncomment the following lines if you're not using the virtual environment created by uv\n",
+    "# import sys\n",
+    "# sys.path.append(\"../\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from app.agent import root_agent\n",
+    "from app.agent_engine_app import AgentEngineApp\n",
+    "\n",
+    "agent_engine = AgentEngineApp(agent=root_agent)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "for event in agent_engine.stream_query(message=\"hi!\", user_id=\"test\"):\n",
+    "    print(event)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -133,17 +165,39 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Local Testing\n",
+    "#### Remote Testing\n",
     "\n",
-    "> You can run the application locally via the `make backend` command."
+    "For more information about authenticating HTTPS requests to Cloud Run services, see:\n",
+    "[Cloud Run Authentication Documentation](https://cloud.google.com/run/docs/triggering/https-request)\n",
+    "\n",
+    "Remote testing involves using a deployed service URL instead of localhost.\n",
+    "\n",
+    "Authentication is handled using GCP identity tokens instead of local credentials."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 20,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "ID_TOKEN = get_ipython().getoutput(\"gcloud auth print-identity-token -q\")[0]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "SERVICE_URL = \"YOUR_SERVICE_URL_HERE\"  # Replace with your Cloud Run service URL"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### Create a session\n",
-    " Create a new session with user preferences and state information\n"
+    "You'll need to first create a Session"
    ]
   },
   {
@@ -156,8 +210,8 @@
     "session_id = \"test_session_456\"\n",
     "session_data = {\"state\": {\"preferred_language\": \"English\", \"visit_count\": 1}}\n",
     "\n",
-    "session_url = f\"http://127.0.0.1:8000/apps/app/users/{user_id}/sessions/{session_id}\"\n",
-    "headers = {\"Content-Type\": \"application/json\"}\n",
+    "session_url = f\"{SERVICE_URL}/apps/app/users/{user_id}/sessions/{session_id}\"\n",
+    "headers = {\"Content-Type\": \"application/json\", \"Authorization\": f\"Bearer {ID_TOKEN}\"}\n",
     "\n",
     "session_response = requests.post(session_url, headers=headers, json=session_data)\n",
     "print(f\"Session creation status code: {session_response.status_code}\")"
@@ -167,8 +221,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### Send a message\n",
-    "Send a message to the backend service and receive a streaming response\n"
+    "Then you will be able to send a message"
    ]
   },
   {
@@ -185,7 +238,7 @@
     "    \"streaming\": True,\n",
     "}\n",
     "\n",
-    "message_url = \"http://127.0.0.1:8000/run_sse\"\n",
+    "message_url = f\"{SERVICE_URL}/run_sse\"\n",
     "message_response = requests.post(\n",
     "    message_url, headers=headers, json=message_data, stream=True\n",
     ")\n",
@@ -206,39 +259,17 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### Remote Testing\n",
-    "\n",
-    "For more information about authenticating HTTPS requests to Cloud Run services, see:\n",
-    "[Cloud Run Authentication Documentation](https://cloud.google.com/run/docs/triggering/https-request)\n",
-    "\n",
-    "Remote testing involves using a deployed service URL instead of localhost.\n",
+    "### Local Testing\n",
     "\n",
-    "Authentication is handled using GCP identity tokens instead of local credentials."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 20,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "ID_TOKEN = get_ipython().getoutput(\"gcloud auth print-identity-token -q\")[0]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "SERVICE_URL = \"YOUR_SERVICE_URL_HERE\"  # Replace with your Cloud Run service URL"
+    "> You can run the application locally via the `make backend` command."
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "You'll need to first create a Session"
+    "#### Create a session\n",
+    " Create a new session with user preferences and state information\n"
    ]
   },
   {
@@ -251,8 +282,8 @@
     "session_id = \"test_session_456\"\n",
     "session_data = {\"state\": {\"preferred_language\": \"English\", \"visit_count\": 1}}\n",
     "\n",
-    "session_url = f\"{SERVICE_URL}/apps/app/users/{user_id}/sessions/{session_id}\"\n",
-    "headers = {\"Content-Type\": \"application/json\", \"Authorization\": f\"Bearer {ID_TOKEN}\"}\n",
+    "session_url = f\"http://127.0.0.1:8000/apps/app/users/{user_id}/sessions/{session_id}\"\n",
+    "headers = {\"Content-Type\": \"application/json\"}\n",
     "\n",
     "session_response = requests.post(session_url, headers=headers, json=session_data)\n",
     "print(f\"Session creation status code: {session_response.status_code}\")"
@@ -262,7 +293,8 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Then you will be able to send a message"
+    "#### Send a message\n",
+    "Send a message to the backend service and receive a streaming response\n"
    ]
   },
   {
@@ -279,7 +311,7 @@
     "    \"streaming\": True,\n",
     "}\n",
     "\n",
-    "message_url = f\"{SERVICE_URL}/run_sse\"\n",
+    "message_url = \"http://127.0.0.1:8000/run_sse\"\n",
     "message_response = requests.post(\n",
     "    message_url, headers=headers, json=message_data, stream=True\n",
     ")\n",
@@ -299,7 +331,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": ".venv",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -313,7 +345,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.8"
+   "version": "3.12.2"
   }
  },
  "nbformat": 4,

agents/agentic_rag/notebooks/adk_app_testing.ipynb

@@ -16,58 +16,43 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Import libraries"
+    "## Install dependencies"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "import json\n",
-    "\n",
-    "import requests\n",
-    "import vertexai.agent_engines"
+    "!pip install google-cloud-aiplatform --upgrade"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## If you are using Agent Engine\n",
-    "See more documentation at [Agent Engine Overview](https://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/overview)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "### Local Testing\n",
-    "\n",
-    "You can import directly the AgentEngineApp class within your environment. "
+    "### Import libraries"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from app.agent import root_agent\n",
-    "from app.agent_engine_app import AgentEngineApp\n",
+    "import json\n",
     "\n",
-    "agent_engine = AgentEngineApp(agent=root_agent)"
+    "import requests\n",
+    "import vertexai.agent_engines"
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "metadata": {},
-   "outputs": [],
    "source": [
-    "for event in agent_engine.stream_query(message=\"hi!\", user_id=\"test\"):\n",
-    "    print(event)"
+    "## If you are using Agent Engine\n",
+    "See more documentation at [Agent Engine Overview](https://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/overview)"
    ]
   },
   {
@@ -122,6 +107,53 @@
     ")"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Local Testing\n",
+    "\n",
+    "You can import directly the AgentEngineApp class within your environment. \n",
+    "To run the agent locally, follow these steps:\n",
+    "1. Make sure all required packages are installed in your environment\n",
+    "2. The recommended approach is to use the same virtual environment created by the 'uv' tool\n",
+    "3. You can set up this environment by running 'make install' from your agent's root directory\n",
+    "4. Then select this kernel (.venv folder in your project) in your Jupyter notebook to ensure all dependencies are available"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Uncomment the following lines if you're not using the virtual environment created by uv\n",
+    "# import sys\n",
+    "# sys.path.append(\"../\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from app.agent import root_agent\n",
+    "from app.agent_engine_app import AgentEngineApp\n",
+    "\n",
+    "agent_engine = AgentEngineApp(agent=root_agent)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "for event in agent_engine.stream_query(message=\"hi!\", user_id=\"test\"):\n",
+    "    print(event)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -133,17 +165,39 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Local Testing\n",
+    "#### Remote Testing\n",
     "\n",
-    "> You can run the application locally via the `make backend` command."
+    "For more information about authenticating HTTPS requests to Cloud Run services, see:\n",
+    "[Cloud Run Authentication Documentation](https://cloud.google.com/run/docs/triggering/https-request)\n",
+    "\n",
+    "Remote testing involves using a deployed service URL instead of localhost.\n",
+    "\n",
+    "Authentication is handled using GCP identity tokens instead of local credentials."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 20,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "ID_TOKEN = get_ipython().getoutput(\"gcloud auth print-identity-token -q\")[0]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "SERVICE_URL = \"YOUR_SERVICE_URL_HERE\"  # Replace with your Cloud Run service URL"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### Create a session\n",
-    " Create a new session with user preferences and state information\n"
+    "You'll need to first create a Session"
    ]
   },
   {
@@ -156,8 +210,8 @@
     "session_id = \"test_session_456\"\n",
     "session_data = {\"state\": {\"preferred_language\": \"English\", \"visit_count\": 1}}\n",
     "\n",
-    "session_url = f\"http://127.0.0.1:8000/apps/app/users/{user_id}/sessions/{session_id}\"\n",
-    "headers = {\"Content-Type\": \"application/json\"}\n",
+    "session_url = f\"{SERVICE_URL}/apps/app/users/{user_id}/sessions/{session_id}\"\n",
+    "headers = {\"Content-Type\": \"application/json\", \"Authorization\": f\"Bearer {ID_TOKEN}\"}\n",
     "\n",
     "session_response = requests.post(session_url, headers=headers, json=session_data)\n",
     "print(f\"Session creation status code: {session_response.status_code}\")"
@@ -167,8 +221,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### Send a message\n",
-    "Send a message to the backend service and receive a streaming response\n"
+    "Then you will be able to send a message"
    ]
   },
   {
@@ -185,7 +238,7 @@
     "    \"streaming\": True,\n",
     "}\n",
     "\n",
-    "message_url = \"http://127.0.0.1:8000/run_sse\"\n",
+    "message_url = f\"{SERVICE_URL}/run_sse\"\n",
     "message_response = requests.post(\n",
     "    message_url, headers=headers, json=message_data, stream=True\n",
     ")\n",
@@ -206,39 +259,17 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "#### Remote Testing\n",
-    "\n",
-    "For more information about authenticating HTTPS requests to Cloud Run services, see:\n",
-    "[Cloud Run Authentication Documentation](https://cloud.google.com/run/docs/triggering/https-request)\n",
-    "\n",
-    "Remote testing involves using a deployed service URL instead of localhost.\n",
+    "### Local Testing\n",
     "\n",
-    "Authentication is handled using GCP identity tokens instead of local credentials."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 20,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "ID_TOKEN = get_ipython().getoutput(\"gcloud auth print-identity-token -q\")[0]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "SERVICE_URL = \"YOUR_SERVICE_URL_HERE\"  # Replace with your Cloud Run service URL"
+    "> You can run the application locally via the `make backend` command."
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "You'll need to first create a Session"
+    "#### Create a session\n",
+    " Create a new session with user preferences and state information\n"
    ]
   },
   {
@@ -251,8 +282,8 @@
     "session_id = \"test_session_456\"\n",
     "session_data = {\"state\": {\"preferred_language\": \"English\", \"visit_count\": 1}}\n",
     "\n",
-    "session_url = f\"{SERVICE_URL}/apps/app/users/{user_id}/sessions/{session_id}\"\n",
-    "headers = {\"Content-Type\": \"application/json\", \"Authorization\": f\"Bearer {ID_TOKEN}\"}\n",
+    "session_url = f\"http://127.0.0.1:8000/apps/app/users/{user_id}/sessions/{session_id}\"\n",
+    "headers = {\"Content-Type\": \"application/json\"}\n",
     "\n",
     "session_response = requests.post(session_url, headers=headers, json=session_data)\n",
     "print(f\"Session creation status code: {session_response.status_code}\")"
@@ -262,7 +293,8 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Then you will be able to send a message"
+    "#### Send a message\n",
+    "Send a message to the backend service and receive a streaming response\n"
    ]
   },
   {
@@ -279,7 +311,7 @@
     "    \"streaming\": True,\n",
     "}\n",
     "\n",
-    "message_url = f\"{SERVICE_URL}/run_sse\"\n",
+    "message_url = \"http://127.0.0.1:8000/run_sse\"\n",
     "message_response = requests.post(\n",
     "    message_url, headers=headers, json=message_data, stream=True\n",
     ")\n",
@@ -299,7 +331,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": ".venv",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -313,7 +345,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.8"
+   "version": "3.12.2"
   }
  },
  "nbformat": 4,

src/base_template/Makefile

@@ -42,8 +42,8 @@ backend:
 		"COMMIT_SHA=$(shell git rev-parse HEAD){%- if cookiecutter.data_ingestion %}{%- if cookiecutter.datastore_type == "vertex_ai_search" %},DATA_STORE_ID={{cookiecutter.project_name}}-datastore,DATA_STORE_REGION=us{%- elif cookiecutter.datastore_type == "vertex_ai_vector_search" %},VECTOR_SEARCH_INDEX={{cookiecutter.project_name}}-vector-search,VECTOR_SEARCH_INDEX_ENDPOINT={{cookiecutter.project_name}}-vector-search-endpoint,VECTOR_SEARCH_BUCKET=$$PROJECT_ID-{{cookiecutter.project_name}}-vs{%- endif %}{%- endif %}"
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
 	# Export dependencies to requirements file using uv export.
-	uv export --no-hashes --no-sources --no-header --no-dev --no-emit-project --no-annotate --frozen > .requirements.txt 2>/dev/null || \
-	uv export --no-hashes --no-sources --no-header --no-dev --no-emit-project --frozen > .requirements.txt && uv run app/agent_engine_app.py
+	uv export --no-hashes --no-header --no-dev --no-emit-project --no-annotate --frozen > .requirements.txt 2>/dev/null || \
+	uv export --no-hashes --no-header --no-dev --no-emit-project --frozen > .requirements.txt && uv run app/agent_engine_app.py
 {%- endif %}
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
 

src/cli/commands/setup_cicd.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import json
 import logging
 import re
 import shutil
@@ -228,40 +227,18 @@ def prompt_for_repository_details(
                 repo_url = click.prompt(
                     "Enter existing repository URL (e.g., https://github.com/owner/repo)"
                 )
-                # Extract owner and name from URL
-                match = re.match(r"https://github\.com/([^/]+)/([^/]+)", repo_url)
+                # Extract owner and repo name from URL
+                match = re.search(r"github\.com/([^/]+)/([^/]+)", repo_url)
                 if match:
                     repository_owner = match.group(1)
-                    repository_name = match.group(2)
-
-                    # Verify repository exists and is empty
-                    try:
-                        result = run_command(
-                            [
-                                "gh",
-                                "repo",
-                                "view",
-                                f"{repository_owner}/{repository_name}",
-                                "--json",
-                                "isEmpty",
-                            ],
-                            capture_output=True,
-                        )
-                        if not json.loads(result.stdout).get("isEmpty", False):
-                            if not click.confirm(
-                                "Repository is not empty. Are you sure you want to use it?",
-                                default=False,
-                            ):
-                                continue
-                        break
-                    except subprocess.CalledProcessError:
-                        console.print(
-                            "❌ Repository not found or not accessible",
-                            style="bold red",
-                        )
-                        continue
+                    # Remove .git suffix if present
+                    repository_name = match.group(2).rstrip(".git")
+                    break
                 else:
-                    console.print("❌ Invalid repository URL format", style="bold red")
+                    console.print(
+                        "❌ Invalid repository URL format. Please use https://github.com/owner/repo",
+                        style="bold red",
+                    )
 
     if repository_name is None or repository_owner is None:
         raise ValueError("Repository name and owner must be provided")
@@ -570,15 +547,12 @@ def setup_cicd(
 
     if git_provider == "github" and detected_mode == "interactive":
         # First create the repository since we're in interactive mode
-        create_github_repository(repository_owner, repository_name)
+        if not repository_exists:
+            create_github_repository(repository_owner, repository_name)
 
         # Then create the connection
         oauth_token_secret_id, github_app_installation_id = create_github_connection(
-            project_id=cicd_project,
-            region=region,
-            connection_name=host_connection_name,
-            repository_name=repository_name,
-            repository_owner=repository_owner,
+            project_id=cicd_project, region=region, connection_name=host_connection_name
         )
         repository_exists = True
     elif git_provider == "github" and detected_mode == "programmatic":

src/cli/utils/cicd.py

@@ -89,11 +89,7 @@ def setup_repository_name(
 
 
 def create_github_connection(
-    project_id: str,
-    region: str,
-    connection_name: str,
-    repository_name: str,
-    repository_owner: str,
+    project_id: str, region: str, connection_name: str
 ) -> tuple[str, str]:
     """Create and verify GitHub connection using gcloud command.
 
@@ -101,53 +97,12 @@ def create_github_connection(
         project_id: GCP project ID
         region: GCP region
         connection_name: Name for the GitHub connection
-        repository_name: Name of repository to create
-        repository_owner: Owner of repository
 
     Returns:
         tuple[str, str]: The OAuth token secret ID and the app installation ID
     """
     console.print("\n🔗 Creating GitHub connection...")
 
-    # Create repository if details provided
-    try:
-        # Check if repo exists
-        result = run_command(
-            [
-                "gh",
-                "repo",
-                "view",
-                f"{repository_owner}/{repository_name}",
-                "--json",
-                "name",
-            ],
-            capture_output=True,
-            check=False,
-        )
-
-        if result.returncode != 0:
-            # Repository doesn't exist, create it
-            console.print(
-                f"\n📦 Creating GitHub repository: {repository_owner}/{repository_name}"
-            )
-            run_command(
-                [
-                    "gh",
-                    "repo",
-                    "create",
-                    f"{repository_owner}/{repository_name}",
-                    "--private",
-                    "--description",
-                    "Repository created by Terraform",
-                ]
-            )
-            console.print("✅ GitHub repository created")
-        else:
-            console.print("✅ Using existing GitHub repository")
-    except subprocess.CalledProcessError as e:
-        console.print(f"❌ Failed to create/check repository: {e!s}", style="bold red")
-        raise
-
     def try_create_connection() -> subprocess.CompletedProcess[str]:
         cmd = [
             "gcloud",