agent-starter-pack 0.3.3__py3-none-any.whl → 0.21.0__py3-none-any.whl

agent_starter_pack/agents/adk_base/tests/integration/test_agent.py

@@ -0,0 +1,58 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# mypy: disable-error-code="union-attr"
+from google.adk.agents.run_config import RunConfig, StreamingMode
+from google.adk.runners import Runner
+from google.adk.sessions import InMemorySessionService
+from google.genai import types
+
+from {{cookiecutter.agent_directory}}.agent import root_agent
+
+
+def test_agent_stream() -> None:
+    """
+    Integration test for the agent stream functionality.
+    Tests that the agent returns valid streaming responses.
+    """
+
+    session_service = InMemorySessionService()
+
+    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(
+        role="user", parts=[types.Part.from_text(text="Why is the sky blue?")]
+    )
+
+    events = list(
+        runner.run(
+            new_message=message,
+            user_id="test_user",
+            session_id=session.id,
+            run_config=RunConfig(streaming_mode=StreamingMode.SSE),
+        )
+    )
+    assert len(events) > 0, "Expected at least one message"
+
+    has_text_content = False
+    for event in events:
+        if (
+            event.content
+            and event.content.parts
+            and any(part.text for part in event.content.parts)
+        ):
+            has_text_content = True
+            break
+    assert has_text_content, "Expected at least one message with text content"

agents/crewai_coding_crew/template/.templateconfig.yaml → agent_starter_pack/agents/adk_live/.template/templateconfig.yaml

@@ -12,15 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-description: "A multi-agent system implemented with CrewAI created to support coding activities"
+description: "Real-time multimodal agent with ADK and Gemini Live API for low-latency voice and video interaction."
 settings:
   requires_data_ingestion: false
+  frontend_type: "adk_live_react"
   deployment_targets: ["agent_engine", "cloud_run"]
-  extra_dependencies: [
-    "langchain-google-vertexai~=2.0.7",
-    "langchain~=0.3.14",
-    "langchain-community~=0.3.17", 
-    "langchain-openai~=0.3.5",
-    "langgraph~=0.2.63",
-    "crewai~=0.102.0"
-  ]
+  extra_dependencies: ["google-adk>=1.16.0,<2.0.0", "click>=8.0.0,<9.0.0", "uvicorn>=0.18.0,<1.0.0", "fastapi>=0.75.0,<1.0.0", "backoff>=2.0.0,<3.0.0"]
+  tags: ["adk", "adk_live"]
+example_question: "What's the weather in San Francisco?"

agent_starter_pack/agents/adk_live/README.md

@@ -0,0 +1,32 @@
+# ADK Live Agent
+
+Real-time conversational agent built with Google ADK and Gemini's live audio model. Supports audio, video, and text interactions with native tool calling.
+
+![live_api_diagram](https://storage.googleapis.com/github-repo/generative-ai/sample-apps/e2e-gen-ai-app-starter-pack/live_api_diagram.png)
+
+**Key components:**
+
+- **Python Backend** (in `app/` folder): ADK-powered agent using Gemini's live audio model with native tool calling and deployment support for Cloud Run and Agent Engine
+
+- **React Frontend** (in `frontend/` folder): Web console for interacting with the live agent via audio, video, and text
+
+![live api demo](https://storage.googleapis.com/github-repo/generative-ai/sample-apps/e2e-gen-ai-app-starter-pack/adk_live_pattern_demo.gif)
+
+Once running, click the play button to connect and interact with the agent. Try asking "What's the weather like in San Francisco?" to see tool calling in action.
+
+## Additional Resources for Multimodal Live API
+
+Explore these resources to learn more about the Multimodal Live API and see examples of its usage:
+
+- [Project Pastra](https://github.com/heiko-hotz/gemini-multimodal-live-dev-guide/tree/main): a comprehensive developer guide for the Gemini Multimodal Live API.
+- [ADK Samples: Realtime Conversational Agent](https://github.com/google/adk-samples/tree/main/python/agents/realtime-conversational-agent): Full-stack, reusable template using Agent Development Kit (ADK) with the Gemini Live API.
+- [Google Cloud Multimodal Live API demos and samples](https://github.com/GoogleCloudPlatform/generative-ai/tree/main/gemini/multimodal-live-api): Collection of code samples and demo applications leveraging multimodal live API in Vertex AI
+- [Gemini 2 Cookbook](https://github.com/google-gemini/cookbook/tree/main/gemini-2): Practical examples and tutorials for working with Gemini 2
+- [Multimodal Live API Web Console](https://github.com/google-gemini/multimodal-live-api-web-console): Interactive React-based web interface for testing and experimenting with Gemini Multimodal Live API.
+
+## Current Status & Future Work
+
+This pattern is under active development. Key areas planned for future enhancement include:
+
+*   **Observability:** Implementing comprehensive monitoring and tracing features.
+*   **Load Testing:** Integrating load testing capabilities.

agent_starter_pack/agents/adk_live/app/__init__.py

@@ -0,0 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .agent import app
+
+__all__ = ["app"]

agent_starter_pack/agents/adk_live/app/agent.py

@@ -0,0 +1,51 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+
+import google.auth
+import vertexai
+from google.adk.agents import Agent
+from google.adk.apps.app import App
+
+_, project_id = google.auth.default()
+os.environ.setdefault("GOOGLE_CLOUD_PROJECT", project_id)
+os.environ.setdefault("GOOGLE_CLOUD_LOCATION", "us-central1")
+os.environ.setdefault("GOOGLE_GENAI_USE_VERTEXAI", "True")
+
+vertexai.init(project=project_id, location="us-central1")
+
+
+def get_weather(query: str) -> str:
+    """Simulates a web search. Use it get information on weather.
+
+    Args:
+        query: A string containing the location to get weather information for.
+
+    Returns:
+        A string with the simulated weather information for the queried location.
+    """
+    if "sf" in query.lower() or "san francisco" in query.lower():
+        return "It's 60 degrees and foggy."
+    return "It's 90 degrees and sunny."
+
+
+root_agent = Agent(
+    name="root_agent",
+    model="gemini-live-2.5-flash-preview-native-audio-09-2025",
+    instruction="You are a helpful AI assistant designed to provide accurate and useful information.",
+    tools=[get_weather],
+)
+
+app = App(root_agent=root_agent, name="app")

agent_starter_pack/agents/adk_live/tests/unit/test_dummy.py

@@ -0,0 +1,38 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+You can add your unit tests here.
+This is where you test your business logic, including agent functionality,
+data processing, and other core components of your application.
+"""
+
+from app.agent import get_weather
+
+
+def test_get_weather_san_francisco() -> None:
+    """Test get_weather function returns correct weather for San Francisco."""
+    result = get_weather("What's the weather in San Francisco?")
+    assert result == "It's 60 degrees and foggy."
+
+
+def test_get_weather_san_francisco_abbreviation() -> None:
+    """Test get_weather function returns correct weather for SF abbreviation."""
+    result = get_weather("weather in sf")
+    assert result == "It's 60 degrees and foggy."
+
+
+def test_get_weather_other_location() -> None:
+    """Test get_weather function returns default weather for other locations."""
+    result = get_weather("What's the weather in New York?")
+    assert result == "It's 90 degrees and sunny."

agents/agentic_rag/template/.templateconfig.yaml → agent_starter_pack/agents/agentic_rag/.template/templateconfig.yaml

@@ -12,17 +12,21 @@
 # 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."
+example_question: "How to save a pandas dataframe to CSV?"
 settings:
   requires_data_ingestion: true
+  requires_session: true
   deployment_targets: ["agent_engine", "cloud_run"]
   extra_dependencies: [
-    "google-adk~=0.1.0",
+    "google-adk>=1.15.0,<2.0.0",
     "langchain-google-vertexai~=2.0.7",
-    "langchain~=0.3.14",
+    "langchain~=0.3.24",
+    "langchain-core~=0.3.55",
     "langchain-community~=0.3.17",
     "langchain-openai~=0.3.5",
     "langchain-google-community[vertexaisearch]~=2.0.7",
+    "Jinja2~=3.1.6",
   ]
   tags: ["adk"]
   frontend_type: "None"

{agents → agent_starter_pack/agents}/agentic_rag/README.md

@@ -19,5 +19,5 @@ The agent implements the following architecture:
 - **Automated Data Ingestion Pipeline:** Automates the process of ingesting data from input sources.
 - **Custom Embeddings:** Generates embeddings using Vertex AI Embeddings and incorporates them into your data for enhanced semantic search.
 - **Terraform Deployment:** Ingestion pipeline is instantiated with Terraform alongside the rest of the infrastructure of the starter pack.
-- **Cloud Build Integration:** Deployment of ingestion pipelines is added to the CD pipelines of the starter pack.
+- **CI/CD Integration:** Deployment of ingestion pipelines is added to the CD pipelines of the starter pack.
 - **Customizable Code:** Easily adapt and customize the code to fit your specific application needs and data sources.

agent_starter_pack/agents/agentic_rag/app/__init__.py

@@ -0,0 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .agent import app
+
+__all__ = ["app"]

{agents → agent_starter_pack/agents}/agentic_rag/app/agent.py

@@ -18,18 +18,20 @@ import os
 import google
 import vertexai
 from google.adk.agents import Agent
+from google.adk.apps.app import App
 from langchain_google_vertexai import VertexAIEmbeddings
 
-from app.retrievers import get_compressor, get_retriever
-from app.templates import format_docs
+from {{cookiecutter.agent_directory}}.retrievers import get_compressor, get_retriever
+from {{cookiecutter.agent_directory}}.templates import format_docs
 
 EMBEDDING_MODEL = "text-embedding-005"
+LLM_LOCATION = "global"
 LOCATION = "us-central1"
-LLM = "gemini-2.0-flash-001"
+LLM = "gemini-2.5-flash"
 
 credentials, project_id = google.auth.default()
 os.environ.setdefault("GOOGLE_CLOUD_PROJECT", project_id)
-os.environ.setdefault("GOOGLE_CLOUD_LOCATION", LOCATION)
+os.environ.setdefault("GOOGLE_CLOUD_LOCATION", LLM_LOCATION)
 os.environ.setdefault("GOOGLE_GENAI_USE_VERTEXAI", "True")
 
 vertexai.init(project=project_id, location=LOCATION)
@@ -114,3 +116,5 @@ root_agent = Agent(
     instruction=instruction,
     tools=[retrieve_docs],
 )
+
+app = App(root_agent=root_agent, name="app")