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

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

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

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

@@ -26,16 +26,16 @@ agents/langgraph_base_react/app/agent.py,sha256=srxiOT-NIM87WnUvP0PXuZGryEG9S33Z
 agents/langgraph_base_react/notebooks/evaluating_langgraph_agent.ipynb,sha256=PlW41fL-ji9NKkimW6u25YRLGWN2pgoX2G8w5nzIEPE,58737
 agents/langgraph_base_react/template/.templateconfig.yaml,sha256=NE_EJzZPtdbJRFdp5k9fovc6zUA6Wvm1MBUBvaZ6gV0,1050
 agents/langgraph_base_react/tests/integration/test_agent.py,sha256=wg2Y8V4l7dh7rhWy7Z8zioDW5ILRsz7RZJ-mz0AhK3U,1619
-agents/live_api/README.md,sha256=RbWUT9W9yAsrmjcGUbefZVIRfJvi9N5J16K6zWRy3TU,2904
-agents/live_api/app/agent.py,sha256=6ts3RGqVVfz-gOt3QUTvKz1lFL2imgPn2iJ5ZgtOMzc,2322
-agents/live_api/app/server.py,sha256=MtUeSP9U3-aWK2D82WOS95vvh5i-_7qvUpWtW7DZR7o,7552
+agents/live_api/README.md,sha256=c3lu5WguOUBc5hO_UtNZfVhfYxvSzMXuGFICn9M38mU,2897
+agents/live_api/app/agent.py,sha256=FBsmCYI_dmBIbFD1uojcPxJfozGNc-0rg1fW2PhEyFw,2298
+agents/live_api/app/server.py,sha256=A2frpeM-mHoRiUAlvJKZAdU_Q8-GOwcCh3iODdRmbr4,7652
 agents/live_api/template/.templateconfig.yaml,sha256=xN4ZNa_-qoBWNh4oMh8WBu-KmxUAYtJdF7qlYbHg-FM,964
 agents/live_api/tests/integration/test_server_e2e.py,sha256=D2VETDIyTD2fQyQ6DXwLr-POSYyVzzzcpIFLt5Ppyx4,8398
 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=lvcSmYD0S8dcQNOTKk3ZZG6_92bx8aWWDYStI4bZ8J8,4518
-src/base_template/README.md,sha256=ocMt-AFgd4vJda7ABIiR4ajhf4RxTg1jV4k7Lwq8-No,10143
+src/base_template/Makefile,sha256=O65aYaRQlCOZHfCaAdWbK3eEobw5x83A8nEjK88jtpQ,4538
+src/base_template/README.md,sha256=fZ2qe3LS9bA2pBzvBvAqclPTmdjBzRr84O5P3QoNk0I,10143
 src/base_template/pyproject.toml,sha256=V2Df010o_Q8smOaS1K9fx-pVe9X4_KZT4vpIRaB0aFs,2959
 src/base_template/app/__init__.py,sha256=UyAgc2l8nkVIUPUzL9hKR7EgzCsc2jSYcOcjHNNmpMU,59
 src/base_template/app/utils/tracing.py,sha256=2rv1Ukh2jTBENDwoghCItJ28l-Sjz9gMlzdojlVgJa4,6052
@@ -63,7 +63,7 @@ src/base_template/deployment/terraform/dev/vars/env.tfvars,sha256=LoQMjh1AAMR-MG
 src/base_template/deployment/terraform/vars/env.tfvars,sha256=Nze8q1x2Aj6ZUeWC2hDeZWqNUkLp13DgzA_LFmmOzCo,1216
 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/create.py,sha256=g0cSZUM2MabAg2h7GomuHHr7n5uMVjXBKebt1_1u17o,24145
 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=m2BxwV39323rVzJCjJjB37IPdSNQdxVgXGrgnhdbb84,26040
@@ -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.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,,
+agent_starter_pack-0.4.4.dist-info/METADATA,sha256=9f-WXcPpX22tOh8Qn9Jp1a1FK_EMX5Ix0lUcvnFqMVk,10452
+agent_starter_pack-0.4.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.4.4.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.4.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.4.4.dist-info/RECORD,,

agents/live_api/README.md

@@ -6,7 +6,7 @@ This pattern showcases a real-time conversational agent powered by Google Gemini
 
 **Key components:**
 
-- **Python Backend** (in `app/` folder): A production-ready server built with [FastAPI](https://fastapi.tiangolo.com/) and [google-genai](https://googleapis.github.io/python-python-genai/) that features:
+- **Python Backend** (in `app/` folder): A production-ready server built with [FastAPI](https://fastapi.tiangolo.com/) and [google-genai](https://googleapis.github.io/python-genai/) that features:
 
   - **Real-time bidirectional communication** via WebSockets between the frontend and Gemini model
   - **Integrated tool calling** with a weather information tool for demonstrating external data retrieval

agents/live_api/app/agent.py

@@ -22,8 +22,7 @@ from google.genai import types
 # Constants
 VERTEXAI = os.getenv("VERTEXAI", "true").lower() == "true"
 LOCATION = "us-central1"
-EMBEDDING_MODEL = "text-embedding-004"
-MODEL_ID = "gemini-2.0-flash-exp"
+MODEL_ID = "gemini-2.0-flash-live-preview-04-09"
 
 # Initialize Google Cloud clients
 credentials, project_id = google.auth.default()

agents/live_api/app/server.py

@@ -70,6 +70,7 @@ class GeminiSession:
         while True:
             try:
                 data = await self.websocket.receive_json()
+
                 if isinstance(data, dict) and (
                     "realtimeInput" in data or "clientContent" in data
                 ):
@@ -98,13 +99,7 @@ class GeminiSession:
     async def _handle_tool_call(
         self, session: Any, tool_call: LiveServerToolCall
     ) -> None:
-        """Process tool calls from Gemini and send back responses.
-
-        Args:
-            session: The Gemini session
-            tool_call: Tool call request from Gemini
-        """
-        # Handle case where function_calls might be None
+        """Process tool calls from Gemini and send back responses."""
         if tool_call.function_calls is None:
             logging.debug("No function calls in tool_call")
             return
@@ -116,7 +111,14 @@ class GeminiSession:
                 logging.error(f"Function {fc.name} not found")
                 continue
             args = fc.args if fc.args is not None else {}
-            response = func(**args)
+
+            # Handle both async and sync functions appropriately
+            if asyncio.iscoroutinefunction(func):
+                # Function is already async
+                response = await func(**args)
+            else:
+                # Run sync function in a thread pool to avoid blocking
+                response = await asyncio.to_thread(func, **args)
 
             tool_response = types.LiveClientToolResponse(
                 function_responses=[
@@ -127,18 +129,15 @@ class GeminiSession:
             await session.send(input=tool_response)
 
     async def receive_from_gemini(self) -> None:
-        """Listen for and process messages from Gemini.
-
-        Continuously receives messages from Gemini, forwards them to the client,
-        and handles any tool calls. Handles connection errors gracefully.
-        """
+        """Listen for and process messages from Gemini without blocking."""
         while result := await self.session._ws.recv(decode=False):
             await self.websocket.send_bytes(result)
             raw_message = json.loads(result)
             if "toolCall" in raw_message:
                 message = types.LiveServerMessage.model_validate(raw_message)
                 tool_call = LiveServerToolCall.model_validate(message.tool_call)
-                await self._handle_tool_call(self.session, tool_call)
+                # Create a separate task to handle the tool call without blocking
+                asyncio.create_task(self._handle_tool_call(self.session, tool_call))
 
 
 def get_connect_and_run_callable(websocket: WebSocket) -> Callable:

src/base_template/Makefile

@@ -22,7 +22,7 @@ playground:
 	uv run uvicorn app.server:app --host 0.0.0.0 --port 8000 --reload &
 {%- endif %}
 {%- if cookiecutter.agent_name == 'live_api' %}
-	npm --prefix frontend start
+	PORT=8501 npm --prefix frontend start
 {%- else %}
 	{% if cookiecutter.deployment_target == 'agent_engine' %}PYTHONPATH=. {% endif %}uv run streamlit run frontend/streamlit_app.py --browser.serverAddress=localhost --server.enableCORS=false --server.enableXsrfProtection=false
 {%- endif %}
@@ -54,7 +54,7 @@ local-backend:
 {%- if cookiecutter.agent_name == 'live_api' %}
 
 ui:
-	npm --prefix frontend start
+	PORT=8501 npm --prefix frontend start
 {%- endif %}
 {%- endif %}
 

src/base_template/README.md

@@ -110,7 +110,7 @@ Here’s the recommended workflow for local development:
     This launches the Streamlit application, which connects to the backend server (by default at `http://localhost:8000`).
 
 4.  **Interact and Iterate:**
-    *   Open the Streamlit UI in your browser (usually `http://localhost:3000` or `http://localhost:3001`).
+    *   Open the Streamlit UI in your browser (usually `http://localhost:8501` or `http://localhost:3001`).
     *   Click the play button in the UI to connect to the backend.
     *   Interact with the agent! Try prompts like: *"Using the tool you have, define Governance in the context MLOPs"*
     *   Modify the agent logic in `app/agent.py`. The backend server (FastAPI with `uvicorn --reload`) should automatically restart when you save changes. Refresh the frontend if needed to see behavioral changes.

src/cli/commands/create.py

@@ -134,7 +134,7 @@ def create(
         console.print("\n=== GCP Agent Starter Pack :rocket:===", style="bold blue")
         console.print("Welcome to the Agent Starter Pack!")
         console.print(
-            "This tool will help you create an end-to-end production-ready AI agent in GCP!\n"
+            "This tool will help you create an end-to-end production-ready AI agent in Google Cloud!\n"
         )
         # Validate project name
         if len(project_name) > 26:
@@ -317,9 +317,9 @@ def create(
 
         console.print("\n> Success! Your agent project is ready.")
         console.print(
-            "\n📖 For more information on project structure, usage, and deployment, check out the README:"
+            f"\n📖 Project README: [cyan]cat {cd_path}/README.md[/]"
+            "\n   Online Development Guide: [cyan][link=https://goo.gle/asp-dev]https://goo.gle/asp-dev[/link][/cyan]"
         )
-        console.print(f"   [cyan]cat {cd_path}/README.md[/]")
         # Determine the correct path to display based on whether output_dir was specified
         console.print("\n🚀 To get started, run the following command:")
         console.print(