see run-llama#12874: provide the exception to the StreamChatErrorEvent (

run-llama#12879)
chrisalexiuk-nvidia · Apr 25, 2024 · 8834e71 · 8834e71
1 parent b4de677
commit 8834e71
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/llama-index-core/llama_index/core/chat_engine/types.py b/llama-index-core/llama_index/core/chat_engine/types.py
@@ -175,7 +175,7 @@ def write_response_to_history(
                 chat.message.content = final_text.strip()  # final message
                 memory.put(chat.message)
         except Exception as e:
-            dispatch_event(StreamChatErrorEvent())
+            dispatch_event(StreamChatErrorEvent(exception=e))
             if not raise_error:
                 logger.warning(
                     f"Encountered exception writing response to history: {e}"
@@ -229,7 +229,7 @@ async def awrite_response_to_history(
                 chat.message.content = final_text.strip()  # final message
                 memory.put(chat.message)
         except Exception as e:
-            dispatch_event(StreamChatErrorEvent())
+            dispatch_event(StreamChatErrorEvent(exception=e))
             logger.warning(f"Encountered exception writing response to history: {e}")
         dispatch_event(StreamChatEndEvent())
         self._is_done = True

diff --git a/llama-index-core/llama_index/core/instrumentation/events/chat_engine.py b/llama-index-core/llama_index/core/instrumentation/events/chat_engine.py
@@ -16,6 +16,8 @@ def class_name(cls):
 
 
 class StreamChatErrorEvent(BaseEvent):
+    exception: Exception
+
     @classmethod
     def class_name(cls):
         """Class name."""