agent-starter-pack 0.3.7__tar.gz → 0.3.8__tar.gz

{agent_starter_pack-0.3.7 → agent_starter_pack-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.3.7
+Version: 0.3.8
 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.3.7 → agent_starter_pack-0.3.8}/agents/agentic_rag/template/.templateconfig.yaml

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-description: "A RAG agent for document retrieval and Q&A. Includes a data pipeline for ingesting and indexing documents into Vertex AI Search or Vector Search."
+description: "ADK RAG agent for document retrieval and Q&A. Includes a data pipeline for ingesting and indexing documents into Vertex AI Search or Vector Search."
 settings:
   requires_data_ingestion: true
   deployment_targets: ["agent_engine", "cloud_run"]

{agent_starter_pack-0.3.7 → agent_starter_pack-0.3.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.3.7"
+version = "0.3.8"
 description = "CLI tool to create GCP-based AI agent projects from templates"
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },

{agent_starter_pack-0.3.7 → agent_starter_pack-0.3.8}/src/deployment_targets/agent_engine/app/agent_engine_app.py

@@ -102,20 +102,12 @@ from app.utils.typing import Feedback, InputChat, dumpd, ensure_valid_config
 class AgentEngineApp:
     """Class for managing agent engine functionality."""
 
-    def __init__(
-        self, project_id: str | None = None, env_vars: dict[str, str] | None = None
-    ) -> None:
+    def __init__(self, project_id: str | None = None) -> None:
         """Initialize the AgentEngineApp variables"""
         self.project_id = project_id
-        self.env_vars = env_vars if env_vars is not None else {}
 
     def set_up(self) -> None:
         """The set_up method is used to define application initialization logic"""
-        import os
-
-        for k, v in self.env_vars.items():
-            os.environ[k] = v
-
         # Lazy import agent at setup time to avoid deployment dependencies
         from app.agent import agent
 
@@ -214,7 +206,7 @@ def deploy_agent_engine_app(
     agent_name: str | None = None,
     requirements_file: str = ".requirements.txt",
     extra_packages: list[str] = ["./app"],
-    env_vars: dict[str, str] | None = None,
+    env_vars: dict[str, str] = {},
 ) -> agent_engines.AgentEngine:
     """Deploy the agent engine app to Vertex AI."""
 
@@ -229,19 +221,20 @@ def deploy_agent_engine_app(
     with open(requirements_file) as f:
         requirements = f.read().strip().split("\n")
 {% if "adk" in cookiecutter.tags %}
-    agent_engine = AgentEngineApp(
-        agent=root_agent,
-        env_vars=env_vars,
-    )
+    agent_engine = AgentEngineApp(agent=root_agent)
 {% else %}
-    agent_engine = AgentEngineApp(project_id=project, env_vars=env_vars)
+    agent_engine = AgentEngineApp(project_id=project)
 {% endif %}
+    # Set worker parallelism to 1
+    env_vars["NUM_WORKERS"] = "1"
+
     # Common configuration for both create and update operations
     agent_config = {
         "agent_engine": agent_engine,
         "display_name": agent_name,
         "description": "{{cookiecutter.agent_description}}",
         "extra_packages": extra_packages,
+        "env_vars": env_vars,
     }
     logging.info(f"Agent config: {agent_config}")
     agent_config["requirements"] = requirements
@@ -308,9 +301,8 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     # Parse environment variables if provided
-    env_vars = None
+    env_vars = {}
     if args.set_env_vars:
-        env_vars = {}
         for pair in args.set_env_vars.split(","):
             key, value = pair.split("=", 1)
             env_vars[key] = value

{agent_starter_pack-0.3.7 → agent_starter_pack-0.3.8}/uv.lock

@@ -11,7 +11,7 @@ resolution-markers = [
 
 [[package]]
 name = "agent-starter-pack"
-version = "0.3.7"
+version = "0.3.8"
 source = { editable = "." }
 dependencies = [
     { name = "backoff" },