Update src/chat_display.py

src/chat_display.py

@@ -112,8 +112,8 @@ def display_chat_history():
     """
     user_avatar, assistant_avatar = get_avatars()
     
-    # Get full history from database
-    history = get_full_history()
+    # Get history directly from session state (local cache)
+    history = st.session_state.get('history', [])
     
     # Display all messages instead of limiting to a fixed number
     display_history = history