agent-starter-pack 0.1.6__tar.gz → 0.1.7__tar.gz

{agent_starter_pack-0.1.6 → agent_starter_pack-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.1.6
+Version: 0.1.7
 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.1.6 → agent_starter_pack-0.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.1.6"
+version = "0.1.7"
 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.1.6 → agent_starter_pack-0.1.7}/src/frontends/live_api_react/frontend/package-lock.json

@@ -446,25 +446,25 @@
       }
     },
     "node_modules/@babel/helpers": {
-      "version": "7.26.0",
-      "resolved": "https://registry.npmjs.org/@babel/helpers/-/helpers-7.26.0.tgz",
-      "integrity": "sha512-tbhNuIxNcVb21pInl3ZSjksLCvgdZy9KwJ8brv993QtIVKJBBkYXz4q4ZbAv31GdnC+R90np23L5FbEBlthAEw==",
+      "version": "7.26.10",
+      "resolved": "https://registry.npmjs.org/@babel/helpers/-/helpers-7.26.10.tgz",
+      "integrity": "sha512-UPYc3SauzZ3JGgj87GgZ89JVdC5dj0AoetR5Bw6wj4niittNyFh6+eOGonYvJ1ao6B8lEa3Q3klS7ADZ53bc5g==",
       "license": "MIT",
       "dependencies": {
-        "@babel/template": "^7.25.9",
-        "@babel/types": "^7.26.0"
+        "@babel/template": "^7.26.9",
+        "@babel/types": "^7.26.10"
       },
       "engines": {
         "node": ">=6.9.0"
       }
     },
     "node_modules/@babel/parser": {
-      "version": "7.26.3",
-      "resolved": "https://registry.npmjs.org/@babel/parser/-/parser-7.26.3.tgz",
-      "integrity": "sha512-WJ/CvmY8Mea8iDXo6a7RK2wbmJITT5fN3BEkRuFlxVyNx8jOKIIhmC4fSkTcPcf8JyavbBwIe6OpiCOBXt/IcA==",
+      "version": "7.26.10",
+      "resolved": "https://registry.npmjs.org/@babel/parser/-/parser-7.26.10.tgz",
+      "integrity": "sha512-6aQR2zGE/QFi8JpDLjUZEPYOs7+mhKXm86VaKFiLP35JQwQb6bwUE+XbvkH0EptsYhbNBSUGaUBLKqxH1xSgsA==",
       "license": "MIT",
       "dependencies": {
-        "@babel/types": "^7.26.3"
+        "@babel/types": "^7.26.10"
       },
       "bin": {
         "parser": "bin/babel-parser.js"
@@ -2039,14 +2039,14 @@
       }
     },
     "node_modules/@babel/template": {
-      "version": "7.25.9",
-      "resolved": "https://registry.npmjs.org/@babel/template/-/template-7.25.9.tgz",
-      "integrity": "sha512-9DGttpmPvIxBb/2uwpVo3dqJ+O6RooAFOS+lB+xDqoE2PVCE8nfoHMdZLpfCQRLwvohzXISPZcgxt80xLfsuwg==",
+      "version": "7.26.9",
+      "resolved": "https://registry.npmjs.org/@babel/template/-/template-7.26.9.tgz",
+      "integrity": "sha512-qyRplbeIpNZhmzOysF/wFMuP9sctmh2cFzRAZOn1YapxBsE1i9bJIY586R/WBLfLcmcBlM8ROBiQURnnNy+zfA==",
       "license": "MIT",
       "dependencies": {
-        "@babel/code-frame": "^7.25.9",
-        "@babel/parser": "^7.25.9",
-        "@babel/types": "^7.25.9"
+        "@babel/code-frame": "^7.26.2",
+        "@babel/parser": "^7.26.9",
+        "@babel/types": "^7.26.9"
       },
       "engines": {
         "node": ">=6.9.0"
@@ -2071,9 +2071,9 @@
       }
     },
     "node_modules/@babel/types": {
-      "version": "7.26.3",
-      "resolved": "https://registry.npmjs.org/@babel/types/-/types-7.26.3.tgz",
-      "integrity": "sha512-vN5p+1kl59GVKMvTHt55NzzmYVxprfJD+ql7U9NFIfKCBkYE55LYtS+WtPlaYOyzydrKI8Nezd+aZextrd+FMA==",
+      "version": "7.26.10",
+      "resolved": "https://registry.npmjs.org/@babel/types/-/types-7.26.10.tgz",
+      "integrity": "sha512-emqcG3vHrpxUKTrxcblR36dcrcoRDvKmnL/dCL6ZsHaShW80qxCAcNhzQZrpeM765VzEos+xOi4s+r4IXzTwdQ==",
       "license": "MIT",
       "dependencies": {
         "@babel/helper-string-parser": "^7.25.9",

{agent_starter_pack-0.1.6 → agent_starter_pack-0.1.7}/src/frontends/streamlit/frontend/utils/stream_handler.py

@@ -159,7 +159,7 @@ class Client:
             if self.authenticate_request:
                 headers["Authorization"] = f"Bearer {self.id_token}"
             with requests.post(
-                self.url, json={"input": data}, headers=headers, stream=True, timeout=10
+                self.url, json=data, headers=headers, stream=True, timeout=10
             ) as response:
                 for line in response.iter_lines():
                     if line:
@@ -169,7 +169,7 @@ class Client:
                         except json.JSONDecodeError:
                             print(f"Failed to parse event: {line.decode('utf-8')}")
         elif self.agent is not None:
-            yield from self.agent.stream_query(input=data)
+            yield from self.agent.stream_query(**data)
 
 
 class StreamHandler:
@@ -207,17 +207,17 @@ class EventProcessor:
         self.current_run_id: str | None = None
         self.additional_kwargs: dict[str, Any] = {}
 
-    def process_events(self, run_id: str | None = None) -> None:
+    def process_events(self) -> None:
         """Process events from the stream, handling each event type appropriately."""
         messages = self.st.session_state.user_chats[
             self.st.session_state["session_id"]
         ]["messages"]
-        self.current_run_id = run_id or str(uuid.uuid4())
+        self.current_run_id = str(uuid.uuid4())
         # Set run_id in session state at start of processing
         self.st.session_state["run_id"] = self.current_run_id
         stream = self.client.stream_messages(
             data={
-                "messages": messages,
+                "input": {"messages": messages},
                 "config": {
                     "run_id": self.current_run_id,
                     "metadata": {

{agent_starter_pack-0.1.6 → agent_starter_pack-0.1.7}/tests/test_frontend/test_utils/test_stream_handler.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 from typing import Any
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, patch
 
 from frontend.utils.stream_handler import EventProcessor, StreamHandler
 
@@ -150,7 +150,8 @@ def test_event_processor_direct_response() -> None:
     stream_handler_mock.new_token.assert_any_call(" World")
 
 
-def test_event_processor_session_state_updates() -> None:
+@patch("uuid.uuid4", return_value="test_run")
+def test_event_processor_session_state_updates(mock_uuid: MagicMock) -> None:
     """Test EventProcessor updates to session state"""
     st_mock = MockStreamlit()
     client_mock = MagicMock()
@@ -165,22 +166,24 @@ def test_event_processor_session_state_updates() -> None:
     st_mock.session_state.user_chats["test_session"]["messages"] = []
 
     processor = EventProcessor(st_mock, client_mock, stream_handler_mock)
-    run_id = "test_run"
-    processor.process_events(run_id=run_id)
+    # Set run_id before processing events
+    st_mock.session_state["run_id"] = "test_run"
+    processor.process_events()
 
     # Verify stream events call
     client_mock.stream_messages.assert_called_once()
     call_args = client_mock.stream_messages.call_args[1]["data"]
 
-    # Verify messages passed correctly
-    assert "messages" in call_args
-    assert call_args["messages"] == []
+    # Verify input and config structure
+    assert "input" in call_args
+    assert "messages" in call_args["input"]
+    assert call_args["input"]["messages"] == []
 
     # Verify config passed correctly
     assert "config" in call_args
     config = call_args["config"]
     assert "run_id" in config
-    assert config["run_id"] == processor.current_run_id
+    assert config["run_id"] == "test_run"
     assert config["metadata"]["user_id"] == "test_user"
     assert config["metadata"]["session_id"] == "test_session"
 
@@ -189,10 +192,12 @@ def test_event_processor_session_state_updates() -> None:
     assert len(messages) == 1
     assert messages[0]["content"] == "test response"
     assert "id" in messages[0]
-    assert st_mock.session_state["run_id"] == run_id
+    assert messages[0]["id"] == "test_run"
+    assert st_mock.session_state["run_id"] == "test_run"
 
 
-def test_event_processor_session_state_updates_with_tools() -> None:
+@patch("uuid.uuid4", return_value="test_run")
+def test_event_processor_session_state_updates_with_tools(mock_uuid: MagicMock) -> None:
     """Test EventProcessor updates to session state when using tools"""
     st_mock = MockStreamlit()
     client_mock = MagicMock()
@@ -229,8 +234,9 @@ def test_event_processor_session_state_updates_with_tools() -> None:
     st_mock.session_state.user_chats["test_session"]["messages"] = []
 
     processor = EventProcessor(st_mock, client_mock, stream_handler_mock)
-    run_id = "test_run"
-    processor.process_events(run_id=run_id)
+    # Set run_id before processing events
+    st_mock.session_state["run_id"] = "test_run"
+    processor.process_events()
 
     # Verify session state updates
     messages = st_mock.session_state.user_chats["test_session"]["messages"]

{agent_starter_pack-0.1.6 → agent_starter_pack-0.1.7}/uv.lock

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