agent-starter-pack 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

{agent_starter_pack-0.1.4.dist-info → agent_starter_pack-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.1.4
+Version: 0.1.5
 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

{agent_starter_pack-0.1.4.dist-info → agent_starter_pack-0.1.5.dist-info}/RECORD

@@ -29,9 +29,9 @@ agents/multimodal_live_api/tests/integration/test_server_e2e.py,sha256=D2VETDIyT
 agents/multimodal_live_api/tests/load_test/load_test.py,sha256=HHZyfC4gqiQtZVF_CbbxENGgWQccMLpwMv0IdoQ6cbQ,1275
 agents/multimodal_live_api/tests/unit/test_server.py,sha256=_TjlgQgNkjerIaBGnu8P8_KB8ZlSolDcivALpUOn_Rw,4786
 src/base_template/.gitignore,sha256=mJKTZIcVdAFiIUQicRfPNGUg6WvwcfTEC2xbmAaU34g,2579
-src/base_template/Makefile,sha256=PNmyhdIRf0p6eRLYqWHfCskLmuDMG673YPEeu51XLVQ,1720
-src/base_template/README.md,sha256=A2AsGhPVfbMTjiPnuuEui7V8UVWa7hE5m0zut--QbbU,8196
-src/base_template/pyproject.toml,sha256=Pv4nC4uh_9_sGA90Tom0ITJUAbGolhseCA9-AOaEIpY,2799
+src/base_template/Makefile,sha256=e2NsR1SLXZJjDcNFo1dRcoQCAoz7NHJap1OS82jIclY,2555
+src/base_template/README.md,sha256=gXEXkrnU06x-x2PpeCyfPPly8xXTiCqwsrdmSqPPPx8,8602
+src/base_template/pyproject.toml,sha256=JpwwlTf8PYgp68HlHDsu3arWoHdryUm7iex19H5fHyc,2797
 src/base_template/app/utils/tracing.py,sha256=ovp6tgwgo_tqibpbhhHl8cWIqqUWVYfV0AL7XPDEPpM,5661
 src/base_template/app/utils/typing.py,sha256=UUFm74qiKGtT8pLuE_3yRpcmntRCt91wH_Q-LKdjQuQ,3194
 src/base_template/deployment/README.md,sha256=8AorSeF2Aj0kvPmuQLFrmAAUWVbdTzg0S-rVBbfdirQ,4835
@@ -57,7 +57,7 @@ src/base_template/deployment/terraform/dev/vars/env.tfvars,sha256=xz5cYG29ZIgqM1
 src/base_template/deployment/terraform/vars/env.tfvars,sha256=2dxWRDzeolN2f7eKSHXcBiBt9fRcIMO6VH-2MAZevLQ,1633
 src/base_template/tests/unit/test_utils/test_tracing_exporter.py,sha256=JAb0vIB7wNFPm_kaDaHcxtPKNReypDHjdsMQyzpuePQ,4687
 src/cli/main.py,sha256=vNEV1Fdl1BhJAFbKHcN3MN1v7dPRUcAgXMnB3uCnlI0,1591
-src/cli/commands/create.py,sha256=38wc5DjUOelLtmCnkVLIq9aVc6c1EK10O0fd-refh6I,18963
+src/cli/commands/create.py,sha256=e4366_oC4NQSDMLnB2sB-G1oHDuHz3_0m9sDW7z5y1Q,20338
 src/cli/commands/setup_cicd.py,sha256=EgVkDdA94npDHHPTuYDmWNIiRscnsXtN7CbTuelrTVU,28347
 src/cli/utils/__init__.py,sha256=x-k6jfcv68_zYS88LrA1mr0yB5y5SekJ0nc5Tvt6iDA,1064
 src/cli/utils/cicd.py,sha256=nrb46lJ3bR_yVJRlUPsBN-To4v-xo5qO5WmnIi6qcYM,27076
@@ -127,32 +127,32 @@ src/frontends/live_api_react/frontend/src/utils/store-logger.ts,sha256=YxS0TjiGn
 src/frontends/live_api_react/frontend/src/utils/utils.ts,sha256=qQIhLzfyCBLecU0ksQCKIbD3cIflb0hxt0SPZGdYFEo,2457
 src/frontends/live_api_react/frontend/src/utils/worklets/audio-processing.ts,sha256=ULgnXphZUfbHkRhGoPT_670WHjzaXJwWgYU0ISQRSXI,1979
 src/frontends/live_api_react/frontend/src/utils/worklets/vol-meter.ts,sha256=DEXn9ywn7N_tqKBhQ9eUxvWCWqteyfT-Q0dwjbvjjpY,1780
-src/frontends/streamlit/frontend/side_bar.py,sha256=HWMu8TRz7GZUMIoppUdAGO57IvhzTbO-P8HJLFIhTtc,8827
+src/frontends/streamlit/frontend/side_bar.py,sha256=kZOTtHhr-BvzqD_pRN3h2Xu4HfFH2dcAO6XOSMBPnsA,9013
 src/frontends/streamlit/frontend/streamlit_app.py,sha256=O-mF9-0nSL220mZSohKK6QrdSiaRsUJL1o_XeKdj8zo,9868
 src/frontends/streamlit/frontend/style/app_markdown.py,sha256=4H9GGlaOIbG_4T4QvHyTMfoeumJze7cZ2VuGIv39wFo,1050
 src/frontends/streamlit/frontend/utils/chat_utils.py,sha256=Z0OYQu-14_d9tDmH9Z4Vvi8xpSUEm9vuPZ8wmFgl4sQ,2262
 src/frontends/streamlit/frontend/utils/local_chat_history.py,sha256=9wc8L8j4tk10DBPQdV64kdZvqE1fHxC2esK8szid0l8,4741
 src/frontends/streamlit/frontend/utils/message_editing.py,sha256=YWoPe2KeWMuL3YVTm0am6MK3kzjEIYVmdkdwTQpmGdQ,2263
 src/frontends/streamlit/frontend/utils/multimodal_utils.py,sha256=v6YbCkz_YcnEo-9YvRjwBNt0SzU4M39bYxJGmKk69vE,7313
-src/frontends/streamlit/frontend/utils/stream_handler.py,sha256=PZxffzmJqj_u_5DDx8TFuXFl9C756-oK_m6VAhlo4rg,11971
+src/frontends/streamlit/frontend/utils/stream_handler.py,sha256=afw-CnLx1LAsZWrQIivA0l2JVamlo2Ez88QoL5mFsN4,12091
 src/frontends/streamlit/frontend/utils/title_summary.py,sha256=B0cadS_KPW-tsbABauI4J681aqjEtuKFDa25e9R1WKc,3030
 src/resources/containers/data_processing/Dockerfile,sha256=uXjqDiSrr7YpoX-vtzaIJIVXZ2qeUpu1Nzju9qWZ9oI,859
 src/resources/containers/e2e-tests/Dockerfile,sha256=Q_aTyX_iaFY8j06XZkpMuggJnNO5daiLmmrvqaZHMxw,1611
-src/resources/locks/uv-agentic_rag_vertexai_search-agent_engine.lock,sha256=6muWlgt4dduqGDiGUvTg-d2zzY5rgW9zP9DbDnt3krc,474353
-src/resources/locks/uv-agentic_rag_vertexai_search-cloud_run.lock,sha256=ViUzp35RGWZiiQvvVi6ntFTT-bEbHg4_HDDQznD4ML8,756574
-src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=TOICJcW3ajjycUfK-bzY6hEZaMRiGf521z1e9n0BwQI,568962
-src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=FUwOcG2_RzxKvKHSxA833EXkVC4P8geVYH-3GWI2QXI,878413
-src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=DihpnjNFvnfPP9k-YmQVpEHWbE0Zx-aYbiqjZNi0LU0,469496
-src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=nMsiSn0weaVWrMxrH8-SnNtuKAEe3byvMOjrQ-UcS4I,751717
-src/resources/locks/uv-multimodal_live_api-cloud_run.lock,sha256=gLLVXP01wVom5fnvpFoYmue6Ynx3qiFbb1FOfDDQy34,751312
+src/resources/locks/uv-agentic_rag_vertexai_search-agent_engine.lock,sha256=G_RE_Qwt2eb9qoaWngVh1VMmeU85Sp3viF5DAaVDixQ,598272
+src/resources/locks/uv-agentic_rag_vertexai_search-cloud_run.lock,sha256=7_J9EJLrNqYxq274H9cCMLN8ENTmSKgVUziyvesD5UE,751772
+src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=b1WmObO8ClSTUGoXMS8u3R4V22-TN4UBAu-573mECT8,713850
+src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=JrYcrsUngizvPGDCyjmkhVoo41E7QYV-jLyeLTLWgTU,880790
+src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=xkO_5lKi7rH5IevTeXynxxOTTtWJslTBXFpPN7m2YwQ,593415
+src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=7WFFXeWYvr2FhJGhd9S_ezjW91ZwPIC2Kh-oy5gbr_Q,746915
+src/resources/locks/uv-multimodal_live_api-cloud_run.lock,sha256=_MRLUX5DQUlH6vDdaBQAj430i-OVUbtoDfmVMmHK48k,746510
 src/resources/setup_cicd/cicd_variables.tf,sha256=OHNKglkpN2TSTnfgpd-fM-eNUbWhs6grNwBIKHoLyfU,1097
 src/resources/setup_cicd/github.tf,sha256=g7gMQqPZI69XteCSruNfiT00GBGawuXANHj-F0fc-GE,2727
 src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5jB7QBfF0,1122
 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=IdbTLReYknTXXu5KWPz3sc7Zg5HTEMS5UVlMSnuhs5E,6067
-agent_starter_pack-0.1.4.dist-info/METADATA,sha256=fp6UjfA1O-M8Mo_vN41tSmjPxTYseeun5WOi5OzYI74,7701
-agent_starter_pack-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.1.4.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.1.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.1.4.dist-info/RECORD,,
+agent_starter_pack-0.1.5.dist-info/METADATA,sha256=lhkhYJnZ84HD-Xm2G2VJDJIOptv0otO2_qbsxTqf4K4,7701
+agent_starter_pack-0.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.1.5.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.1.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.1.5.dist-info/RECORD,,

src/base_template/Makefile

@@ -31,6 +31,24 @@ ui:
 {%- endif %}
 {%- endif %}
 
+setup-dev-env:
+	@if [ -z "$$PROJECT_ID" ]; then echo "Error: PROJECT_ID environment variable is not set"; exit 1; fi
+	(cd deployment/terraform/dev && terraform init && terraform apply --var-file vars/env.tfvars --var dev_project_id=$$PROJECT_ID --auto-approve)
+
+{%- if cookiecutter.data_ingestion%}
+data-ingestion:
+	@if [ -z "$$PROJECT_ID" ]; then echo "Error: PROJECT_ID environment variable is not set"; exit 1; fi
+	$(MAKE) install
+	(cd data_ingestion && uv run data_ingestion_pipeline/submit_pipeline.py \
+		--project-id=$$PROJECT_ID \
+		--data-store-id="sample-datastore" \
+		--region="us-central1" \
+		--data-store-region="us" \
+		--service-account="vertexai-pipelines-sa@$$PROJECT_ID.iam.gserviceaccount.com" \
+		--pipeline-root="gs://$$PROJECT_ID-pipeline-artifacts" \
+		--pipeline-name="data-ingestion-pipeline")
+{%- endif %}
+
 lint:
 	uv run codespell
 	uv run ruff check . --diff

src/base_template/README.md

@@ -113,6 +113,15 @@ For full command options and usage, refer to the [Makefile](Makefile).
 3.  **Interact with the Agent**
     Once both the backend and frontend are running, click the play button in the frontend UI to establish a connection with the backend. You can now interact with the Multimodal Live Agent! You can try asking questions such as "Using the tool you have, define Governance in the context MLOPs" to allow the agent to use the documentation it was provided to.
 
+**Remote access**:
+Use [Cloud Run proxy](https://cloud.google.com/sdk/gcloud/reference/run/services/proxy) for local access. The backend will be accessible at `http://localhost:8000`:
+
+   ```bash
+   gcloud run services proxy genai-app-sample --port 8000 --project $PROJECT_ID --region $REGION
+   ```
+
+   You can then use the same frontend setup described above to interact with your Cloud Run deployment.
+
 <details>
 <summary><b>Cloud Shell usage</b></summary>
 

src/base_template/pyproject.toml

@@ -22,9 +22,9 @@ dependencies = [
 {%- endif %}
 ]
 {% if cookiecutter.deployment_target == 'cloud_run' %}
-requires-python = ">=3.10,<=3.13"
+requires-python = ">=3.10,<3.14"
 {% elif cookiecutter.deployment_target == 'agent_engine' %}
-requires-python = ">=3.10,<=3.11"
+requires-python = ">=3.10,<3.13"
 {%- endif %}
 
 [dependency-groups]

src/cli/commands/create.py

@@ -163,11 +163,12 @@ def create(
         # GCP Setup
         logging.debug("Setting up GCP...")
 
+        creds_info = {}
         # Check for uv installation if not skipping checks
         if not skip_checks:
             # Set up GCP environment
             try:
-                setup_gcp_environment(
+                creds_info = setup_gcp_environment(
                     auto_approve=auto_approve,
                     skip_checks=skip_checks,
                     region=region,
@@ -211,17 +212,29 @@ def create(
         )
 
         project_path = destination_dir / project_name
-        console.print("\n> 👍 Done. Execute the following command to get started:")
-        if output_dir:
-            # If output_dir was specified, use the absolute path
-            console.print(
-                f"[bold bright_green]cd {project_path} && make install && make playground[/]"
-            )
-        else:
-            # If no output_dir specified, just use project name
+        # Determine the correct path to display to the user
+        display_path = str(project_path) if output_dir else project_name
+
+        if include_data_ingestion:
+            project_id = creds_info.get("project", "")
             console.print(
-                f"[bold bright_green]cd {project_name} && make install && make playground[/]"
+                f"\n[bold white]===== DATA INGESTION SETUP =====[/bold white]\n"
+                f"This agent uses a datastore for grounded responses.\n"
+                f"The agent will work without data, but for optimal results:\n"
+                f"1. Set up dev environment:\n"
+                f"   [white italic]`export PROJECT_ID={project_id} && cd {display_path} && make setup-dev-env`[/white italic]\n\n"
+                f"   See deployment/README.md for more info\n"
+                f"2. Run the data ingestion pipeline:\n"
+                f"   [white italic]`export PROJECT_ID={project_id} && cd {display_path} && make data-ingestion`[/white italic]\n\n"
+                f"   See data_ingestion/README.md for more info\n"
+                f"[bold white]=================================[/bold white]\n"
             )
+        console.print("\n> 👍 Done. Execute the following command to get started:")
+
+        # Show the command to get started
+        console.print(
+            f"[bold bright_green]cd {display_path} && make install && make playground[/]"
+        )
 
         # Replace region in all files if a different region was specified
         if region != "us-central1":
@@ -473,7 +486,16 @@ def replace_region_in_files(
         )
 
     # Define allowed file extensions
-    allowed_extensions = {".md", ".py", ".tfvars", ".yaml", ".tf", ".yml"}
+    allowed_extensions = {
+        ".md",
+        ".py",
+        ".tfvars",
+        ".yaml",
+        ".tf",
+        ".yml",
+        "Makefile",
+        "makefile",
+    }
 
     # Skip directories that shouldn't be modified
     skip_dirs = {".git", "__pycache__", "venv", ".venv", "node_modules"}
@@ -491,7 +513,10 @@ def replace_region_in_files(
         if (
             file_path.is_dir()
             or any(skip_dir in file_path.parts for skip_dir in skip_dirs)
-            or file_path.suffix not in allowed_extensions
+            or (
+                file_path.suffix not in allowed_extensions
+                and file_path.name not in allowed_extensions
+            )
         ):
             continue
 
@@ -522,6 +547,13 @@ def replace_region_in_files(
                     'data_store_region="us"', f'data_store_region="{data_store_region}"'
                 )
                 modified = True
+            elif 'data-store-region="us"' in content:
+                if debug:
+                    logging.debug(f"Replacing data-store-region in {file_path}")
+                content = content.replace(
+                    'data-store-region="us"', f'data-store-region="{data_store_region}"'
+                )
+                modified = True
             elif "_DATA_STORE_REGION: us" in content:
                 if debug:
                     logging.debug(f"Replacing _DATA_STORE_REGION in {file_path}")

src/frontends/streamlit/frontend/side_bar.py

@@ -56,10 +56,11 @@ class SideBar:
             )
             use_agent_path = self.st.selectbox(
                 "Select Agent Type",
-                ["Local Agent", "Remote Engine ID", "Remote URL"],
+                ["Local Agent", "Remote Agent Engine ID", "Remote URL"],
                 index=["Local Agent", "Remote Engine ID", "Remote URL"].index(
                     default_agent_type
                 ),
+                help="'Local Agent' uses a local implementation, 'Remote Agent Engine ID' connects to a deployed Vertex AI agent, and 'Remote URL' connects to a custom endpoint.",
             )
 
             if use_agent_path == "Local Agent":

src/frontends/streamlit/frontend/utils/stream_handler.py

@@ -32,6 +32,8 @@ from vertexai.preview import reasoning_engines
 
 from frontend.utils.multimodal_utils import format_content
 
+st.cache_resource.clear()
+
 
 @st.cache_resource
 def get_remote_agent(remote_agent_engine_id: str) -> Any:
@@ -65,6 +67,16 @@ def get_remote_url_config(url: str, authenticate_request: bool) -> dict[str, Any
     }
 
 
+@st.cache_resource()
+def get_local_agent(agent_callable_path: str) -> Any:
+    """Get cached local agent instance."""
+    module_path, class_name = agent_callable_path.rsplit(".", 1)
+    module = importlib.import_module(module_path)
+    agent = getattr(module, class_name)()
+    agent.set_up()
+    return agent
+
+
 class Client:
     """A client for streaming events from a server."""
 
@@ -94,14 +106,10 @@ class Client:
             self.agent = get_remote_agent(remote_agent_engine_id)
             self.url = None
         else:
-            # Force reload all submodules to get latest changes
             self.url = None
             if agent_callable_path is None:
                 raise ValueError("agent_callable_path cannot be None")
-            module_path, class_name = agent_callable_path.rsplit(".", 1)
-            module = importlib.import_module(module_path)
-            self.agent = getattr(module, class_name)()
-            self.agent.set_up()
+            self.agent = get_local_agent(agent_callable_path)
 
     def log_feedback(self, feedback_dict: dict[str, Any], run_id: str) -> None:
         """Log user feedback for a specific run."""