agent-starter-pack 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

{agent_starter_pack-0.5.0.dist-info → agent_starter_pack-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.5.0
+Version: 0.5.1
 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.5.0.dist-info → agent_starter_pack-0.5.1.dist-info}/RECORD

@@ -138,7 +138,7 @@ src/frontends/streamlit/frontend/utils/chat_utils.py,sha256=Z0OYQu-14_d9tDmH9Z4V
 src/frontends/streamlit/frontend/utils/local_chat_history.py,sha256=9wc8L8j4tk10DBPQdV64kdZvqE1fHxC2esK8szid0l8,4741
 src/frontends/streamlit/frontend/utils/message_editing.py,sha256=YWoPe2KeWMuL3YVTm0am6MK3kzjEIYVmdkdwTQpmGdQ,2263
 src/frontends/streamlit/frontend/utils/multimodal_utils.py,sha256=v6YbCkz_YcnEo-9YvRjwBNt0SzU4M39bYxJGmKk69vE,7313
-src/frontends/streamlit/frontend/utils/stream_handler.py,sha256=kRKv7gQDEDsVtJxUsxY5uqCCtR3nQmT1nkZyrQ6Hpsw,12030
+src/frontends/streamlit/frontend/utils/stream_handler.py,sha256=-XVRfLH6ck1KP6NS4vrhcvPyZe6z3NLHliEg17s9jjo,12551
 src/frontends/streamlit/frontend/utils/title_summary.py,sha256=B0cadS_KPW-tsbABauI4J681aqjEtuKFDa25e9R1WKc,3030
 src/frontends/streamlit_adk/frontend/side_bar.py,sha256=VsrIVJkjZPg8AAkVn8Nrn9XN4KasrPVYEKhsZXFqGkU,9053
 src/frontends/streamlit_adk/frontend/streamlit_app.py,sha256=0KtzcvVzc8Ge3XJp0nmeZ691RlI22wsU32WUhKoX9LM,12243
@@ -169,8 +169,8 @@ src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5
 src/utils/generate_locks.py,sha256=6V1B8V2BEuevWnXUsxZVTrLjXwFRII8UfsIGrQqZxVs,4320
 src/utils/lock_utils.py,sha256=RSE6n3hBkH64xNm3Q5wrR0Pqbeo-oc7xaTOhA9yzHjk,2275
 src/utils/watch_and_rebuild.py,sha256=vP4yIiA7E_lj5sfQdJUl8TXas6V7msDg8XWUutAC05Q,6679
-agent_starter_pack-0.5.0.dist-info/METADATA,sha256=NWdmJCeRy_Z3AIw_3UUGJEwIk4MfqV8dkPygrT-c1QY,10454
-agent_starter_pack-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.5.0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.5.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.5.0.dist-info/RECORD,,
+agent_starter_pack-0.5.1.dist-info/METADATA,sha256=2GCZDXrJusKMveFpA8rFwssXfaekQfwPsRZzM5G587o,10454
+agent_starter_pack-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.5.1.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.5.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.5.1.dist-info/RECORD,,

src/frontends/streamlit/frontend/utils/stream_handler.py

@@ -253,10 +253,19 @@ class EventProcessor:
                         msg = f"\n\nTool response: `{content}`"
                         self.stream_handler.new_status(msg)
 
-                    # Handle AI responses
-                    elif content := message.get("content"):
-                        self.final_content += content
-                        self.stream_handler.new_token(content)
+                    # Handle incremental AI response chunks
+                    # These are partial content pieces that need to be accumulated
+                    elif (
+                        message.get("content")
+                        and message.get("type") == "AIMessageChunk"
+                    ):
+                        self.final_content += message.get("content")
+                        self.stream_handler.new_token(message.get("content"))
+
+                    # Handle complete AI responses
+                    # This is used when receiving a full message rather than chunks
+                    elif message.get("content") and message.get("type") == "ai":
+                        self.final_content = message.get("content")
 
         # Handle end of stream
         if self.final_content: