asdF

app.py

@@ -70,11 +70,7 @@ def open_chat_window(**kwargs):
         msgbox.chat_message(message["role"]).write(message['content'])
 
     if user_input := st.chat_input('Enter your message'):
-        # Add user message to chat history
-        print(f'State: {st.session_state}\nUser inserted message: {user_input}')
-
         st.session_state.messages.add_message(role='user', content=user_input)
-        # Display user message in chat message container
         msgbox.chat_message("user").write(user_input)
         response = st.session_state.naomi.respond(st.session_state.messages[-1])
         msgbox.chat_message('assistant').write_stream(st.session_state.naomi.gen(response))

data_utils.py

@@ -125,7 +125,7 @@ class ChatSession:
 
     def __getitem__(self, index):
         """ Only returns the role and content for the requested index."""
-        if 0 <= index < len(self._messages):
+        if -len(self._messages) <= index < len(self._messages):
             msg = self._messages[index]
             return {
                 'role': msg.role,

tests/test_agent.py

@@ -0,0 +1 @@
+from agent import Naomi