baohuynhbk14 commited on
Commit
4170f81
·
1 Parent(s): 2285967

Remove user message logging and adjust message history retrieval in Conversation class

Browse files
Files changed (2) hide show
  1. app.py +0 -2
  2. conversation.py +1 -1
app.py CHANGED
@@ -278,8 +278,6 @@ def ai_bot(
278
 
279
  try:
280
  # Stream output
281
- _, message = state.get_user_message(source=state.USER, position='last')
282
- logger.info(f"==== User message ====\n{message}")
283
  logger.info(f"==== Image paths ====\n{all_image_paths}")
284
 
285
  response, _ = predict(state,
 
278
 
279
  try:
280
  # Stream output
 
 
281
  logger.info(f"==== Image paths ====\n{all_image_paths}")
282
 
283
  response, _ = predict(state,
conversation.py CHANGED
@@ -139,7 +139,7 @@ class Conversation:
139
  def get_history(self):
140
  results = []
141
  system_message = self.system_message
142
- messages = self.messages[:-1]
143
 
144
  logger.info(f"=== Raw messages ===\n{messages}")
145
 
 
139
  def get_history(self):
140
  results = []
141
  system_message = self.system_message
142
+ messages = self.messages[:-2]
143
 
144
  logger.info(f"=== Raw messages ===\n{messages}")
145