Update app.py

app.py

@@ -22,9 +22,13 @@ def generate_response(prompt, chat_history):
     # Preprocess chat history to include thinking tags
     processed_chat_history = []
     for message in chat_history:
-        # Skipping Thought Process in history 
-        if message["role"] == "assistant" and "metadata" in message and message["metadata"].get("title", "").startswith("Thought"):
-            pass
+        # Skipping Thought Process in history
+        if message["role"] == "assistant":
+            metadata = message.get("metadata", {})
+            if isinstance(metadata, dict) and metadata.get("title", "").startswith("Thought"):
+                pass
+            else:
+                processed_chat_history.append(message)
         else:
             processed_chat_history.append(message)