From e298dda60cfc9afdee36af71a4a68f243aaa996b Mon Sep 17 00:00:00 2001
From: Rahmani mohamed rami <123117552+Ramyrahmeni@users.noreply.github.com>
Date: Fri, 21 Jun 2024 14:52:02 +0100
Subject: [PATCH] improving vector search

---
 app.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/app.py b/app.py
index 06923d6..8e3097c 100644
--- a/app.py
+++ b/app.py
@@ -462,7 +462,7 @@ def main():
                 display_chat_message(message.role, question)
             else:
                 with st.chat_message(translate_role_for_streamlit(message.role)):
-                    display_chat_message(message.role, message['text'])
+                    display_chat_message(message.role, message.parts[0].text)
 
 if __name__ == "__main__":
     main()