agent-starter-pack 0.7.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

src/deployment_targets/cloud_run/tests/load_test/load_test.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import json
 import os
 import time
 {%- if "adk" in cookiecutter.tags %}
@@ -44,15 +43,19 @@ class ChatStreamUser(HttpUser):
 {%- if "adk" in cookiecutter.tags %}
         # Create session first
         user_id = f"user_{uuid.uuid4()}"
-        session_id = f"session_{uuid.uuid4()}"
-        session_data = {"state": {"preferred_language": "English", "visit_count": 5}}
-        requests.post(
-            f"{self.client.base_url}/apps/app/users/{user_id}/sessions/{session_id}",
+        session_data = {"state": {"preferred_language": "English", "visit_count": 1}}
+
+        session_url = f"{self.client.base_url}/apps/app/users/{user_id}/sessions"
+        session_response = requests.post(
+            session_url,
             headers=headers,
             json=session_data,
             timeout=10,
         )
 
+        # Get session_id from response
+        session_id = session_response.json()["id"]
+
         # Send chat message
         data = {
             "app_name": "app",
@@ -60,7 +63,7 @@ class ChatStreamUser(HttpUser):
             "session_id": session_id,
             "new_message": {
                 "role": "user",
-                "parts": [{"text": "What's the weather in San Francisco?"}],
+                "parts": [{"text": "Hello! Weather in New york?"}],
             },
             "streaming": True,
         }
@@ -93,24 +96,25 @@ class ChatStreamUser(HttpUser):
                 events = []
                 for line in response.iter_lines():
                     if line:
-{%- if "adk" in cookiecutter.tags %}
-                        # SSE format is "data: {json}"
                         line_str = line.decode("utf-8")
-                        if line_str.startswith("data: "):
-                            event_json = line_str[6:]  # Remove "data: " prefix
-                            event = json.loads(event_json)
-                            events.append(event)
-{%- else %}
-                        event = json.loads(line)
-                        events.append(event)
-{%- endif %}
+                        events.append(line_str)
+
+                        if "429 Too Many Requests" in line_str:
+                            self.environment.events.request.fire(
+                                request_type="POST",
+                                name=f"{ENDPOINT} rate_limited 429s",
+                                response_time=0,
+                                response_length=len(line),
+                                response=response,
+                                context={},
+                            )
                 end_time = time.time()
                 total_time = end_time - start_time
                 self.environment.events.request.fire(
                     request_type="POST",
                     name=f"{ENDPOINT} end",
                     response_time=total_time * 1000,  # Convert to milliseconds
-                    response_length=len(json.dumps(events)),
+                    response_length=len(events),
                     response=response,
                     context={},
                 )

src/frontends/adk_gemini_fullstack/frontend/src/App.tsx

@@ -81,12 +81,11 @@ export default function App() {
   };
 
   const createSession = async (): Promise<{userId: string, sessionId: string, appName: string}> => {
-    const generatedSessionId = uuidv4();
-    const response = await fetch(`/api/apps/app/users/u_999/sessions/${generatedSessionId}`, {
+    const response = await fetch(`/api/apps/app/users/u_999/sessions`, {
       method: "POST",
       headers: {
         "Content-Type": "application/json"
-      }
+      },
     });
     
     if (!response.ok) {

src/resources/docs/adk-cheatsheet.md

@@ -994,7 +994,7 @@ Serverless container platform for custom web applications.
     # Ensure your agent_folder (e.g., 'my_first_agent') is in the same directory as main.py
     app: FastAPI = get_fast_api_app(
         agents_dir=os.path.dirname(os.path.abspath(__file__)),
-        session_db_url="sqlite:///./sessions.db", # In-container SQLite, for simple cases
+        session_service_uri="sqlite:///./sessions.db", # In-container SQLite, for simple cases
         # For production: use a persistent DB (Cloud SQL) or VertexAiSessionService
         allow_origins=["*"],
         web=True # Serve ADK UI