Mimi commited on
Commit
32e3a9a
·
1 Parent(s): 3b7c309

Fixed typos

Browse files
Files changed (2) hide show
  1. agent.py +1 -1
  2. app.py +1 -1
agent.py CHANGED
@@ -128,7 +128,7 @@ class Naomi:
128
  )
129
 
130
  def respond(self, user_input: dict, **kwargs):
131
- """ Invoked during stream. """
132
  # user msg handling
133
  format_user_input = self.model.tokenizer_.hf_tokenizer.apply_chat_template([user_input], tokenize=False, add_generation_prompt=False)
134
  self.chat_history += format_user_input
 
128
  )
129
 
130
  def respond(self, user_input: dict, **kwargs):
131
+ """ Called during stream. """
132
  # user msg handling
133
  format_user_input = self.model.tokenizer_.hf_tokenizer.apply_chat_template([user_input], tokenize=False, add_generation_prompt=False)
134
  self.chat_history += format_user_input
app.py CHANGED
@@ -76,7 +76,7 @@ def open_chat_window(**kwargs):
76
  st.session_state.messages.add_message(role='user', content=user_input)
77
  # Display user message in chat message container
78
  msgbox.chat_message("user").write(user_input)
79
- response = st.session_state.naomi.invoke(st.session_state.messages[-1])
80
  msgbox.chat_message('assistant').write_stream(st.session_state.naomi.gen(response))
81
  st.session_state.messages.add_message(role='assistant', content=response)
82
 
 
76
  st.session_state.messages.add_message(role='user', content=user_input)
77
  # Display user message in chat message container
78
  msgbox.chat_message("user").write(user_input)
79
+ response = st.session_state.naomi.respond(st.session_state.messages[-1])
80
  msgbox.chat_message('assistant').write_stream(st.session_state.naomi.gen(response))
81
  st.session_state.messages.add_message(role='assistant', content=response)
82