agent-starter-pack 0.4.5__tar.gz → 0.5.1__tar.gz

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.4.5
+Version: 0.5.1
 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
@@ -56,7 +56,7 @@ Ready to build your AI agent? Simply run this command:
 
 ```bash
 # Create and activate a Python virtual environment
-python -m venv venv && source venv/bin/activate
+python -m venv .venv && source .venv/bin/activate
 
 # Install the agent starter pack
 pip install agent-starter-pack

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/README.md

@@ -31,7 +31,7 @@ Ready to build your AI agent? Simply run this command:
 
 ```bash
 # Create and activate a Python virtual environment
-python -m venv venv && source venv/bin/activate
+python -m venv .venv && source .venv/bin/activate
 
 # Install the agent starter pack
 pip install agent-starter-pack

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/agents/adk_base/template/.templateconfig.yaml

@@ -17,6 +17,6 @@ example_question: "What's the weather in San Francisco?"
 settings:
   requires_data_ingestion: false
   deployment_targets: ["agent_engine", "cloud_run"]
-  extra_dependencies: ["google-adk~=0.5.0"]
+  extra_dependencies: ["google-adk~=1.1.0"]
   tags: ["adk"]
   frontend_type: "None"

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/agents/adk_base/tests/integration/test_agent.py

@@ -29,7 +29,7 @@ def test_agent_stream() -> None:
 
     session_service = InMemorySessionService()
 
-    session = session_service.create_session(user_id="test_user", app_name="test")
+    session = session_service.create_session_sync(user_id="test_user", app_name="test")
     runner = Runner(agent=root_agent, session_service=session_service, app_name="test")
 
     message = types.Content(

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/agents/agentic_rag/template/.templateconfig.yaml

@@ -18,7 +18,7 @@ settings:
   requires_data_ingestion: true
   deployment_targets: ["agent_engine", "cloud_run"]
   extra_dependencies: [
-    "google-adk~=0.5.0",
+    "google-adk~=1.1.0",
     "langchain-google-vertexai~=2.0.7",
     "langchain~=0.3.24",
     "langchain-core~=0.3.55",

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/agents/agentic_rag/tests/integration/test_agent.py

@@ -35,7 +35,7 @@ def test_agent_stream(mock_retrieve: MagicMock) -> None:
 
     session_service = InMemorySessionService()
 
-    session = session_service.create_session(user_id="test_user", app_name="test")
+    session = session_service.create_session_sync(user_id="test_user", app_name="test")
     runner = Runner(agent=root_agent, session_service=session_service, app_name="test")
 
     message = types.Content(

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/agents/live_api/app/server.py

@@ -60,6 +60,7 @@ class GeminiSession:
         self.run_id = "n/a"
         self.user_id = "n/a"
         self.tool_functions = tool_functions
+        self._tool_tasks: list[asyncio.Task] = []
 
     async def receive_from_client(self) -> None:
         """Listen for and process messages from the client.
@@ -137,7 +138,10 @@ class GeminiSession:
                 message = types.LiveServerMessage.model_validate(raw_message)
                 tool_call = LiveServerToolCall.model_validate(message.tool_call)
                 # Create a separate task to handle the tool call without blocking
-                asyncio.create_task(self._handle_tool_call(self.session, tool_call))
+                task = asyncio.create_task(
+                    self._handle_tool_call(self.session, tool_call)
+                )
+                self._tool_tasks.append(task)
 
 
 def get_connect_and_run_callable(websocket: WebSocket) -> Callable:

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/docs/cli/setup_cicd.md

@@ -4,21 +4,23 @@ The `setup-cicd` command is a powerful utility provided by `agent-starter-pack`
 
 **⚡️ Quick Start Example:**
 
-The command is now even simpler to get started with. You can run it without any arguments, and it will prompt you for the required project IDs:
+Getting started is straightforward. We recommend using `uvx` to execute the command, as `uv` (a prerequisite, see below) will handle fetching and running the `agent-starter-pack` for you, even if it's not installed globally:
 
 ```bash
-agent-starter-pack setup-cicd
+uvx agent-starter-pack setup-cicd
 ```
 *(You will be prompted for Staging and Production project IDs)*
 
 Alternatively, you can provide the project IDs directly as flags:
 
 ```bash
-agent-starter-pack setup-cicd \
+uvx agent-starter-pack setup-cicd \
   --staging-project=your-staging-project-id \
   --prod-project=your-prod-project-id
 ```
 
+**💡 Tip:** Using `uvx` (e.g., `uvx agent-starter-pack ...`) ensures you are able to run the command without needing to install the `agent-starter-pack` package globally or in your current environment. `uv` itself must be installed.
+
 **⚠️ Important Considerations:**
 
 *   **Experimental:** This command is under active development. Use it with caution and report any issues.
@@ -29,6 +31,7 @@ agent-starter-pack setup-cicd \
 
 1.  **Run from Project Root:** Execute the command from the root directory of your `agent-starter-pack` project (where `pyproject.toml` is located).
 2.  **Install Tools:**
+    *   `uv`: The command will be run using `uvx`. Ensure `uv` is installed. (See [uv installation guide](https://github.com/astral-sh/uv#installation))
     *   Terraform
     *   `gh` CLI (GitHub CLI): Install and authenticate using `gh auth login`.
     *   `gcloud` CLI (Google Cloud SDK): Authenticate using `gcloud auth application-default login`.
@@ -51,7 +54,7 @@ The `setup-cicd` command automates the following:
 ## Running the Command
 
 ```bash
-agent-starter-pack setup-cicd \
+uvx agent-starter-pack setup-cicd \
     [--staging-project <YOUR_STAGING_PROJECT_ID>] \
     [--prod-project <YOUR_PROD_PROJECT_ID>] \
     [--cicd-project <YOUR_CICD_PROJECT_ID>] \
@@ -76,7 +79,7 @@ agent-starter-pack setup-cicd \
 *   `--auto-approve`: (Optional) Skip interactive prompts (including project ID prompts if flags are omitted). Use carefully.
 *   `--debug`: (Optional) Enable verbose logging.
 
-*(For advanced/programmatic use with pre-existing connections, see options like `--github-pat`, `--github-app-installation-id`, `--host-connection-name` by running `agent-starter-pack setup-cicd --help`)*
+*(For advanced/programmatic use with pre-existing connections, see options like `--github-pat`, `--github-app-installation-id`, `--host-connection-name` by running `uvx agent-starter-pack setup-cicd --help`)*
 
 ## After Running the Command
 

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/docs/guide/deployment.md

@@ -1,7 +1,7 @@
 # Deployment
 
 ::: tip ⭐ Streamlined Deployment
-For a streamlined one-command deployment of the entire CI/CD pipeline and infrastructure using Terraform, you can use the [`agent-starter-pack setup-cicd` CLI command](../cli/setup_cicd). Currently only supporting Github.
+For a streamlined one-command deployment of the entire CI/CD pipeline and infrastructure using Terraform, you can use the [`uvx agent-starter-pack setup-cicd` CLI command](../cli/setup_cicd). Currently only supporting Github.
 :::
 
 The templated agent leverages [**Terraform**](http://terraform.io) to define and provision the underlying infrastructure, while [**Cloud Build**](https://cloud.google.com/build/) orchestrates the continuous integration and continuous deployment (CI/CD) pipeline.

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/docs/guide/development-guide.md

@@ -54,12 +54,12 @@ For reliable, automated deployments to staging and production, a CI/CD pipeline
 **Option 1: One-Command CI/CD Setup (Recommended for GitHub)**
 The `agent-starter-pack` CLI streamlines CI/CD setup with GitHub:
 ```bash
-agent-starter-pack setup-cicd
+uvx agent-starter-pack setup-cicd
 ```
 This automates creating a GitHub repository, connecting to Cloud Build, setting up staging/production infrastructure with Terraform, and configuring CI/CD triggers.
 
 Follow the interactive prompts. For critical systems needing granular control, consider the manual setup.
-See the [`agent-starter-pack setup-cicd` CLI reference](https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/cli/setup_cicd.md) for details. *(Note: Automated setup currently supports GitHub only).*
+See the [`agent-starter-pack setup-cicd` CLI reference](../cli/setup_cicd) for details. *(Note: Automated setup currently supports GitHub only).*
 
 **Option 2: Manual CI/CD Setup**
 For full control and compatibility with other Git providers, refer to the [manual deployment setup guide](./deployment.md).

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/docs/guide/getting-started.md

@@ -13,12 +13,13 @@ This guide quickly walks you through setting up your first agent project.
 
 ```bash
 # Create and activate a Python virtual environment (Recommended)
-python -m venv venv
-source venv/bin/activate  # On Windows: venv\Scripts\activate
+python -m venv .venv
+source .venv/bin/activate  # On Windows: .venv\Scripts\activate
 
 # Install the package
 pip install agent-starter-pack
 ```
+Check the [Installation Guide](/guide/installation) for alternative installation methods.
 
 ### 2. Create Your Agent Project
 
@@ -62,10 +63,8 @@ Inside your new project directory (`my-awesome-agent`), you'll find:
 
 ### Next Steps
 
-You're ready to go!
+You're ready to go! See the [Development Guide](/guide/development-guide) for detailed instructions on extending, customizing and deploying your agent.
 
-*   **Customize:** Modify the code in `app/` to edit prompts, configure tools, adjust API endpoints, customize business logic, extend functionality!
-*   **Deploy in minutes:** Use the [setup-cicd](/cli/setup_cicd) command.
-*   **Customized the deployment:** Use Terraform in `deployment/` to deploy to Google Cloud or follow the deployment guide ([Deployment Guide](/guide/deployment)). 
-*   **Add Data (RAG):** Configure [Data Ingestion](/guide/data-ingestion).
-*   **Monitor:** Explore [Observability](/guide/observability) features.
+- **Add Data (RAG):** Configure [Data Ingestion](/guide/data-ingestion) for knowledge-based agents.
+- **Monitor Performance:** Explore [Observability](/guide/observability) features for production monitoring.
+- **Deploy to Production:** Follow the [Deployment Guide](/guide/deployment) to deploy your agent to Google Cloud.

agent_starter_pack-0.5.1/docs/guide/installation.md

@@ -0,0 +1,151 @@
+# Installation
+
+There are several ways to install the Agent Starter Pack. Choose the method that works best for your workflow.
+
+**Want zero setup?** 👉 [Try 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 in [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)
+
+## `uvx` for Quick Project Creation
+
+If you have [uv](https://astral.sh/uv) installed, you can create projects without a permanent installation:
+```bash
+uvx agent-starter-pack create my-awesome-agent
+```
+
+## Virtual Environment Installation
+
+Installs into an isolated Python environment.
+
+```bash
+# Create and activate venv
+python -m venv .venv && source .venv/bin/activate # source .venv/Scripts/activate for Windows Git Bash
+
+# Install using pip or uv
+pip install agent-starter-pack
+```
+
+## Persistent CLI Installation
+
+Installs the `agent-starter-pack` command globally.
+
+### With `pipx` (Isolated Global Tool)
+```bash
+# Install pipx (if needed)
+python3 -m pip install --user pipx && python3 -m pipx ensurepath
+
+# Install Agent Starter Pack
+pipx install agent-starter-pack
+```
+
+### With `uv tool install` (Fast, Isolated Global Tool)
+Requires `uv` (see `uvx` section for install).
+```bash
+uv tool install agent-starter-pack
+```
+
+## Create Project (After Persistent/Venv Install)
+
+If you installed via `pipx`, `uv tool install`, or in a virtual environment:
+```bash
+agent-starter-pack create my-awesome-agent
+```
+
+## Managing Installation
+
+### Upgrading
+*   **`uvx`:** Not needed (always uses latest).
+*   **`pipx`:** `pipx upgrade agent-starter-pack`
+*   **`uv tool`:** `uv tool install agent-starter-pack` (this upgrades)
+*   **`pip`/`uv pip` (in .venv):** `(uv) pip install --upgrade agent-starter-pack`
+
+### Uninstalling
+*   **`uvx`:** Not applicable.
+*   **`pipx`:** `pipx uninstall agent-starter-pack`
+*   **`uv tool`:** `uv tool uninstall agent-starter-pack`
+*   **`pip`/`uv pip` (in .venv):** `(uv) pip uninstall agent-starter-pack`
+
+## Troubleshooting Common Installation Issues
+
+### Command Not Found After Installation
+
+If you encounter "command not found" errors after installation:
+
+1.  **Check your PATH**: Ensure that the Python scripts directory is in your PATH:
+    ```bash
+    echo $PATH
+    ```
+2.  **Verify installation location**: Check where the package was installed:
+    ```bash
+    pip show agent-starter-pack
+    ```
+3.  **Manual path addition**: If needed, add the scripts directory to your PATH:
+    ```bash
+    export PATH="$HOME/.local/bin:$PATH"
+    # For user installations
+    ```
+    Add this line to your `~/.bashrc` or `~/.zshrc` for persistence.
+
+### Permission Errors During Installation
+
+If you encounter permission errors:
+
+1.  **Use user installation mode**:
+    ```bash
+    pip install --user agent-starter-pack
+    ```
+2.  **Check directory permissions**:
+    ```bash
+    ls -la ~/.local/bin
+    ```
+3.  **Fix permissions if needed**:
+    ```bash
+    chmod +x ~/.local/bin/agent-starter-pack
+    ```
+
+### Python Version Compatibility Issues
+
+If you encounter Python version errors:
+
+1.  **Check your Python version**:
+    ```bash
+    python --version
+    ```
+2.  **Install a compatible Python version** if needed (3.10 or newer is required).
+3.  **Create a virtual environment with the correct Python version**:
+    ```bash
+    python3.10 -m venv .venv
+    source .venv/bin/activate
+    ```
+
+### Package Dependency Conflicts
+
+If you encounter dependency conflicts:
+
+1.  **Use a clean virtual environment**:
+    ```bash
+    python -m venv .venv
+    source .venv/bin/activate
+    pip install agent-starter-pack
+    ```
+2.  **Update pip and setuptools**:
+    ```bash
+    pip install --upgrade pip setuptools
+    ```
+3.  **Install with verbose output to identify conflicts**:
+    ```bash
+    pip install -v agent-starter-pack
+    ```
+
+### Installation Verification
+
+To verify your installation is working correctly:
+
+1.  **Check the installed version**:
+    ```bash
+    agent-starter-pack --version
+    ```
+2.  **Run the help command**:
+    ```bash
+    agent-starter-pack --help
+    ```
+
+If you continue to experience issues, please [file an issue](https://github.com/GoogleCloudPlatform/agent-starter-pack/issues) with details about your environment and the specific error messages you're encountering.

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.4.5"
+version = "0.5.1"
 description = "CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates."
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/base_template/README.md

@@ -153,13 +153,13 @@ This template follows a "bring your own agent" approach - you focus on your busi
 1. **Prototype:** Build your Generative AI Agent using the intro notebooks in `notebooks/` for guidance. Use Vertex AI Evaluation to assess performance.
 2. **Integrate:** Import your agent into the app by editing `app/agent.py`.
 3. **Test:** Explore your agent functionality using the Streamlit playground with `make playground`. The playground offers features like chat history, user feedback, and various input types, and automatically reloads your agent on code changes.
-4. **Deploy:** Set up and initiate the CI/CD pipelines, customizing tests as necessary. Refer to the [deployment section](#deployment) for comprehensive instructions. For streamlined infrastructure deployment, simply run `agent-starter-pack setup-cicd`. Check out the [`agent-starter-pack setup-cicd` CLI command](https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/cli/setup_cicd.md). Currently only supporting Github.
+4. **Deploy:** Set up and initiate the CI/CD pipelines, customizing tests as necessary. Refer to the [deployment section](#deployment) for comprehensive instructions. For streamlined infrastructure deployment, simply run `uvx agent-starter-pack setup-cicd`. Check out the [`agent-starter-pack setup-cicd` CLI command](https://googlecloudplatform.github.io/agent-starter-pack/cli/setup_cicd.html). Currently only supporting Github.
 5. **Monitor:** Track performance and gather insights using Cloud Logging, Tracing, and the Looker Studio dashboard to iterate on your application.
 {% endif %}
 
 ## Deployment
 
-> **Note:** For a streamlined one-command deployment of the entire CI/CD pipeline and infrastructure using Terraform, you can use the [`agent-starter-pack setup-cicd` CLI command](https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/cli/setup_cicd.md). Currently only supporting Github.
+> **Note:** For a streamlined one-command deployment of the entire CI/CD pipeline and infrastructure using Terraform, you can use the [`agent-starter-pack setup-cicd` CLI command](https://googlecloudplatform.github.io/agent-starter-pack/cli/setup_cicd.html). Currently only supporting Github.
 
 ### Dev Environment
 

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/base_template/deployment/README.md

@@ -29,7 +29,7 @@ The application leverages [**Terraform**](http://terraform.io) to define and pro
 
 ## Setup
 
-> **Note:** For a streamlined one-command deployment of the entire CI/CD pipeline and infrastructure using Terraform, you can use the [`agent-starter-pack setup-cicd` CLI command](https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/cli/setup_cicd.md). Currently only supporting Github.
+> **Note:** For a streamlined one-command deployment of the entire CI/CD pipeline and infrastructure using Terraform, you can use the [`uvx agent-starter-pack setup-cicd` CLI command](https://googlecloudplatform.github.io/agent-starter-pack/cli/setup_cicd.html). Currently only supporting Github.
 
 **Prerequisites:**
 

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/base_template/pyproject.toml

@@ -16,11 +16,11 @@ dependencies = [
 {%- endif %}
     "google-cloud-logging~=3.11.4",
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-    "google-cloud-aiplatform[evaluation]~=1.91.0",
+    "google-cloud-aiplatform[evaluation]~=1.95.1",
     "fastapi~=0.115.8",
     "uvicorn~=0.34.0"
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-    "google-cloud-aiplatform[evaluation,agent-engines]~=1.91.0"
+    "google-cloud-aiplatform[evaluation,agent-engines]~=1.95.1"
 {%- endif %}
 ]
 {% if cookiecutter.deployment_target == 'cloud_run' %}

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/cli/commands/create.py

@@ -467,11 +467,19 @@ def _handle_credential_verification(creds_info: dict) -> dict:
     # Check if running in Cloud Shell
     if os.environ.get("CLOUD_SHELL") == "true":
         if creds_info["project"] == "":
-            console.print("> It looks like you are running in Cloud Shell.")
             console.print(
-                "> You need to set up a project ID to continue, but you haven't setup a project yet."
+                "> It looks like you are running in Cloud Shell.", style="bold blue"
             )
-            new_project = Prompt.ask("\n> Enter a project ID")
+            console.print(
+                "> You need to set up a project ID to continue, but you haven't setup a project yet.",
+                style="bold blue",
+            )
+            new_project = Prompt.ask("\n> Enter a project ID", default=None)
+            while not new_project:
+                console.print(
+                    "> Project ID cannot be empty. Please try again.", style="bold red"
+                )
+                new_project = Prompt.ask("\n> Enter a project ID", default=None)
             creds_info["project"] = new_project
             set_gcp_project(creds_info["project"], set_quota_project=False)
         return creds_info

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/deployment_targets/agent_engine/tests/load_test/README.md

@@ -19,8 +19,8 @@ Follow these steps to execute load tests:
 
    ```bash
    # Create and activate virtual environment
-   python3 -m venv locust_env
-   source locust_env/bin/activate
+   python3 -m venv .locust_env
+   source .locust_env/bin/activate
    
    # Install required packages
    pip install locust==2.31.1 "google-cloud-aiplatform[langchain,reasoningengine]>=1.77.0"

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/deployment_targets/cloud_run/app/server.py

@@ -32,7 +32,7 @@ provider.add_span_processor(processor)
 trace.set_tracer_provider(provider)
 
 AGENT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-app: FastAPI = get_fast_api_app(agent_dir=AGENT_DIR, web=False)
+app: FastAPI = get_fast_api_app(agents_dir=AGENT_DIR, web=True)
 
 app.title = "{{cookiecutter.project_name}}"
 app.description = "API for interacting with the Agent {{cookiecutter.project_name}}"

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/deployment_targets/cloud_run/tests/load_test/README.md

@@ -18,7 +18,7 @@ uv run uvicorn app.server:app --host 0.0.0.0 --port 8000 --reload
 Using another terminal tab, This is suggested to avoid conflicts with the existing application python environment.
 
 ```bash
-python3 -m venv locust_env && source locust_env/bin/activate && pip install locust==2.31.1
+python3 -m venv .locust_env && source locust_env/bin/activate && pip install locust==2.31.1
 ```
 
 **3. Execute the Load Test:**
@@ -69,7 +69,7 @@ export _ID_TOKEN=$(gcloud auth print-identity-token -q)
 **3. Execute the Load Test:**
 Create virtual environment with Locust:
 ```bash
-python3 -m venv locust_env && source locust_env/bin/activate && pip install locust==2.31.1
+python3 -m venv .locust_env && source locust_env/bin/activate && pip install locust==2.31.1
 ```
 
 Execute load tests. The following command executes the same load test parameters as the local test but targets your remote Cloud Run instance.

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/frontends/streamlit/frontend/utils/stream_handler.py

@@ -253,10 +253,19 @@ class EventProcessor:
                         msg = f"\n\nTool response: `{content}`"
                         self.stream_handler.new_status(msg)
 
-                    # Handle AI responses
-                    elif content := message.get("content"):
-                        self.final_content += content
-                        self.stream_handler.new_token(content)
+                    # Handle incremental AI response chunks
+                    # These are partial content pieces that need to be accumulated
+                    elif (
+                        message.get("content")
+                        and message.get("type") == "AIMessageChunk"
+                    ):
+                        self.final_content += message.get("content")
+                        self.stream_handler.new_token(message.get("content"))
+
+                    # Handle complete AI responses
+                    # This is used when receiving a full message rather than chunks
+                    elif message.get("content") and message.get("type") == "ai":
+                        self.final_content = message.get("content")
 
         # Handle end of stream
         if self.final_content:

{agent_starter_pack-0.4.5 → agent_starter_pack-0.5.1}/src/resources/idx/.idx/dev.nix

@@ -7,9 +7,7 @@
 
   # Use https://search.nixos.org/packages to find packages
   packages = [
-    pkgs.python311
-    pkgs.python311Packages.pip
-    pkgs.python311Packages.uv
+    pkgs.uv
     pkgs.gnumake
     pkgs.terraform
     pkgs.gh
@@ -36,13 +34,8 @@
         echo "Setting gcloud project..."
         gcloud config set project $GOOGLE_CLOUD_PROJECT
 
-        echo "Creating Python virtual environment and installing packages..."
-        uv venv && uv pip install agent-starter-pack
-        echo "alias agent-starter-pack=\"~/$WS_NAME/.venv/bin/agent-starter-pack\"" >> ~/.bashrc
-        source ~/.bashrc
-
         echo "Running agent starter pack creation..."
-        uv run agent-starter-pack create $AGENT_NAME
+        uvx agent-starter-pack create $AGENT_NAME
         code ~/$WS_NAME/$AGENT_NAME/README.md
         exec bash
         '';