agent-starter-pack 0.4.2__tar.gz → 0.4.3__tar.gz

{agent_starter_pack-0.4.2 → agent_starter_pack-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.4.2
+Version: 0.4.3
 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 → agent_starter_pack-0.4.3}/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

{agent_starter_pack-0.4.2 → agent_starter_pack-0.4.3}/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()

{agent_starter_pack-0.4.2 → agent_starter_pack-0.4.3}/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:

{agent_starter_pack-0.4.2 → agent_starter_pack-0.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.4.2"
+version = "0.4.3"
 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.2 → agent_starter_pack-0.4.3}/uv.lock

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