agent-starter-pack 0.1.6__py3-none-any.whl → 0.2.0__py3-none-any.whl

src/frontends/live_api_react/frontend/package.json

@@ -12,7 +12,7 @@
     "react-scripts": "5.0.1",
     "react-select": "^5.8.3",
     "sass": "^1.80.6",
-    "vega": "^5.30.0",
+    "vega": "^5.33.0",
     "vega-embed": "^6.29.0",
     "vega-lite": "^5.22.0",
     "web-vitals": "^2.1.4",

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": {

src/resources/containers/data_processing/Dockerfile

@@ -22,4 +22,6 @@ RUN pip install --no-cache-dir \
     "google-cloud-aiplatform>=1.80.0" \
     "kfp>=1.4.0" \
     "google-cloud-discoveryengine==0.13.6" \
-    "backoff==2.2.1"
+    "backoff==2.2.1" \
+    "google-cloud-pipeline-components==2.19.0" \
+    "langchain-google-vertexai==2.0.13"