agent-starter-pack 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

{agent_starter_pack-0.1.2.dist-info → agent_starter_pack-0.1.3.dist-info}/METADATA

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

@@ -128,13 +128,13 @@ src/frontends/live_api_react/frontend/src/utils/utils.ts,sha256=qQIhLzfyCBLecU0k
 src/frontends/live_api_react/frontend/src/utils/worklets/audio-processing.ts,sha256=ULgnXphZUfbHkRhGoPT_670WHjzaXJwWgYU0ISQRSXI,1979
 src/frontends/live_api_react/frontend/src/utils/worklets/vol-meter.ts,sha256=DEXn9ywn7N_tqKBhQ9eUxvWCWqteyfT-Q0dwjbvjjpY,1780
 src/frontends/streamlit/frontend/side_bar.py,sha256=HWMu8TRz7GZUMIoppUdAGO57IvhzTbO-P8HJLFIhTtc,8827
-src/frontends/streamlit/frontend/streamlit_app.py,sha256=xHvoj4QtpicMuueg6xNd693m5JNRWjGW32DjCBRg5fc,9747
+src/frontends/streamlit/frontend/streamlit_app.py,sha256=O-mF9-0nSL220mZSohKK6QrdSiaRsUJL1o_XeKdj8zo,9868
 src/frontends/streamlit/frontend/style/app_markdown.py,sha256=4H9GGlaOIbG_4T4QvHyTMfoeumJze7cZ2VuGIv39wFo,1050
 src/frontends/streamlit/frontend/utils/chat_utils.py,sha256=Z0OYQu-14_d9tDmH9Z4Vvi8xpSUEm9vuPZ8wmFgl4sQ,2262
 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=Gl_HqRQ4sFiHG-kPGFbr2u4WLus676L2dkMz46Bq0fc,11574
+src/frontends/streamlit/frontend/utils/stream_handler.py,sha256=PZxffzmJqj_u_5DDx8TFuXFl9C756-oK_m6VAhlo4rg,11971
 src/frontends/streamlit/frontend/utils/title_summary.py,sha256=B0cadS_KPW-tsbABauI4J681aqjEtuKFDa25e9R1WKc,3030
 src/resources/containers/data_processing/Dockerfile,sha256=uXjqDiSrr7YpoX-vtzaIJIVXZ2qeUpu1Nzju9qWZ9oI,859
 src/resources/locks/uv-agentic_rag_vertexai_search-agent_engine.lock,sha256=6muWlgt4dduqGDiGUvTg-d2zzY5rgW9zP9DbDnt3krc,474353
@@ -150,8 +150,8 @@ src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5
 src/utils/generate_locks.py,sha256=xu5IAhGGBPkVQGSJX4kk7_JNDwWJUEaXAHbmaQIohbg,4386
 src/utils/lock_utils.py,sha256=_QdzQtgIbCmJ87s046_i1g966slVNmvr3bJDeHbRQSM,2419
 src/utils/watch_and_rebuild.py,sha256=IdbTLReYknTXXu5KWPz3sc7Zg5HTEMS5UVlMSnuhs5E,6067
-agent_starter_pack-0.1.2.dist-info/METADATA,sha256=UiAHSpbBqjztaTdeDjwgLR-kNbh2Nb7QjPa20o4kaxk,7576
-agent_starter_pack-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.1.2.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.1.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.1.2.dist-info/RECORD,,
+agent_starter_pack-0.1.3.dist-info/METADATA,sha256=P9ZI9yGUBDyMMhvXQfilFSdjhxCmgZ3E24EAY5kn2QY,7576
+agent_starter_pack-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.1.3.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.1.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.1.3.dist-info/RECORD,,

src/frontends/streamlit/frontend/streamlit_app.py

@@ -239,8 +239,10 @@ def display_feedback(side_bar: SideBar) -> None:
         )
         if feedback is not None:
             client = Client(
-                agent_callable_path=side_bar.agent_callable_path,
                 remote_agent_engine_id=side_bar.remote_agent_engine_id,
+                agent_callable_path=side_bar.agent_callable_path,
+                url=side_bar.url_input_field,
+                authenticate_request=side_bar.should_authenticate_request,
             )
             client.log_feedback(
                 feedback_dict=feedback,

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

@@ -124,9 +124,20 @@ class Client:
         headers = {
             "Content-Type": "application/json",
         }
-        if self.authenticate_request:
-            headers["Authorization"] = f"Bearer {self.id_token}"
-        requests.post(url, data=json.dumps(feedback_dict), headers=headers, timeout=10)
+        if self.url:
+            url = urljoin(self.url, "feedback")
+            headers = {
+                "Content-Type": "application/json",
+            }
+            if self.authenticate_request:
+                headers["Authorization"] = f"Bearer {self.id_token}"
+            requests.post(
+                url, data=json.dumps(feedback_dict), headers=headers, timeout=10
+            )
+        elif self.agent is not None:
+            self.agent.register_feedback(feedback=feedback_dict)
+        else:
+            raise ValueError("No agent or URL configured for feedback logging")
 
     def stream_messages(
         self, data: dict[str, Any]