improving vector search

Ramyrahmeni · Jun 21, 2024 · e298dda · e298dda
1 parent ce3e68a
commit e298dda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()