From 952026da6c23553eeb27e2b490c4965b313df0e4 Mon Sep 17 00:00:00 2001
From: Varun Sharma <contactvarun27@gmail.com>
Date: Fri, 14 Feb 2025 15:33:35 +0100
Subject: [PATCH] fix: ValueError breaks Ui #629 (#630) #none

---
 libs/ktem/ktem/pages/chat/__init__.py | 55 ++++++++++++++-------------
 1 file changed, 29 insertions(+), 26 deletions(-)

diff --git a/libs/ktem/ktem/pages/chat/__init__.py b/libs/ktem/ktem/pages/chat/__init__.py
index 451684d..ba19749 100644
--- a/libs/ktem/ktem/pages/chat/__init__.py
+++ b/libs/ktem/ktem/pages/chat/__init__.py
@@ -1317,39 +1317,42 @@ class ChatPage(BasePage):
             chat_state,
         )
 
-        for response in pipeline.stream(chat_input, conversation_id, chat_history):
+        try:
+            for response in pipeline.stream(chat_input, conversation_id, chat_history):
 
-            if not isinstance(response, Document):
-                continue
+                if not isinstance(response, Document):
+                    continue
 
-            if response.channel is None:
-                continue
+                if response.channel is None:
+                    continue
 
-            if response.channel == "chat":
-                if response.content is None:
-                    text = ""
-                else:
-                    text += response.content
+                if response.channel == "chat":
+                    if response.content is None:
+                        text = ""
+                    else:
+                        text += response.content
 
-            if response.channel == "info":
-                if response.content is None:
-                    refs = ""
-                else:
-                    refs += response.content
+                if response.channel == "info":
+                    if response.content is None:
+                        refs = ""
+                    else:
+                        refs += response.content
 
-            if response.channel == "plot":
-                plot = response.content
-                plot_gr = self._json_to_plot(plot)
+                if response.channel == "plot":
+                    plot = response.content
+                    plot_gr = self._json_to_plot(plot)
 
-            chat_state[pipeline.get_info()["id"]] = reasoning_state["pipeline"]
+                chat_state[pipeline.get_info()["id"]] = reasoning_state["pipeline"]
 
-            yield (
-                chat_history + [(chat_input, text or msg_placeholder)],
-                refs,
-                plot_gr,
-                plot,
-                chat_state,
-            )
+                yield (
+                    chat_history + [(chat_input, text or msg_placeholder)],
+                    refs,
+                    plot_gr,
+                    plot,
+                    chat_state,
+                )
+        except ValueError as e:
+            print(e)
 
         if not text:
             empty_msg = getattr(