From 9380323a577f1fb99c26753923a9c08577db44f7 Mon Sep 17 00:00:00 2001
From: philschmid <schmidphilipp1995@gmail.com>
Date: Wed, 18 Jun 2025 14:35:12 +0200
Subject: [PATCH] fixed state and app

---
 backend/src/agent/graph.py                   |  4 ++--
 backend/src/agent/state.py                   |  2 +-
 frontend/src/App.tsx                         | 13 ++-----------
 frontend/src/components/ChatMessagesView.tsx |  3 ---
 4 files changed, 5 insertions(+), 17 deletions(-)

diff --git a/backend/src/agent/graph.py b/backend/src/agent/graph.py
index 0340d72..2d4f9d9 100644
--- a/backend/src/agent/graph.py
+++ b/backend/src/agent/graph.py
@@ -78,7 +78,7 @@ def generate_query(state: OverallState, config: RunnableConfig) -> QueryGenerati
     )
     # Generate the search queries
     result = structured_llm.invoke(formatted_prompt)
-    return {"query_list": result.query}
+    return {"search_query": result.query}
 
 
 def continue_to_web_research(state: QueryGenerationState):
@@ -88,7 +88,7 @@ def continue_to_web_research(state: QueryGenerationState):
     """
     return [
         Send("web_research", {"search_query": search_query, "id": int(idx)})
-        for idx, search_query in enumerate(state["query_list"])
+        for idx, search_query in enumerate(state["search_query"])
     ]
 
 
diff --git a/backend/src/agent/state.py b/backend/src/agent/state.py
index 5a45fb4..6644c19 100644
--- a/backend/src/agent/state.py
+++ b/backend/src/agent/state.py
@@ -37,7 +37,7 @@ class Query(TypedDict):
 
 
 class QueryGenerationState(TypedDict):
-    query_list: list[Query]
+    search_query: list[Query]
 
 
 class WebSearchState(TypedDict):
diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx
index 5e186f4..484eefa 100644
--- a/frontend/src/App.tsx
+++ b/frontend/src/App.tsx
@@ -16,7 +16,6 @@ export default function App() {
   const scrollAreaRef = useRef<HTMLDivElement>(null);
   const hasFinalizeEventOccurredRef = useRef(false);
   const [error, setError] = useState<string | null>(null);
-  console.log(import.meta.env.DEV);
   const thread = useStream<{
     messages: Message[];
     initial_search_query_count: number;
@@ -28,15 +27,12 @@ export default function App() {
       : "http://localhost:8123",
     assistantId: "agent",
     messagesKey: "messages",
-    onFinish: (event: any) => {
-      console.log(event);
-    },
     onUpdateEvent: (event: any) => {
       let processedEvent: ProcessedEvent | null = null;
       if (event.generate_query) {
         processedEvent = {
           title: "Generating Search Queries",
-          data: event.generate_query.query_list.join(", "),
+          data: event.generate_query?.search_query?.join(", ") || "",
         };
       } else if (event.web_research) {
         const sources = event.web_research.sources_gathered || [];
@@ -54,12 +50,7 @@ export default function App() {
       } else if (event.reflection) {
         processedEvent = {
           title: "Reflection",
-          data: event.reflection.is_sufficient
-            ? "Search successful, generating final answer."
-            : `Need more information, searching for ${
-                event.reflection.follow_up_queries?.join(", ") ||
-                "additional information"
-              }`,
+          data: "Analysing Web Research Results",
         };
       } else if (event.finalize_answer) {
         processedEvent = {
diff --git a/frontend/src/components/ChatMessagesView.tsx b/frontend/src/components/ChatMessagesView.tsx
index a323648..cc0e2b0 100644
--- a/frontend/src/components/ChatMessagesView.tsx
+++ b/frontend/src/components/ChatMessagesView.tsx
@@ -252,9 +252,6 @@ export function ChatMessagesView({
       console.error("Failed to copy text: ", err);
     }
   };
-
-  console.log("liveActivityEvents", liveActivityEvents);
-
   return (
     <div className="flex flex-col h-full overflow-hidden">
       <ScrollArea className="flex-1 min-h-0" ref={scrollAreaRef}>