Shageenderan Sapai commited on
Commit
6bf62ed
·
1 Parent(s): 319dccb

delete_hidden_messages updated logic

Browse files
Files changed (1) hide show
  1. app/conversation_manager.py +12 -13
app/conversation_manager.py CHANGED
@@ -280,19 +280,18 @@ class ConversationManager:
280
  if old_thread is None:
281
  old_thread = self.current_thread
282
 
283
- # create a new thread
284
- messages = [msg for msg in self._get_current_thread_history(remove_system_message=False) if not msg['content'].startswith("[hidden]")]
285
- if len(messages) >= 29:
286
- messages = messages[-29:]
287
- logger.info(f"Current Thread Messages: {messages}", extra={"user_id": self.user.user_id, "endpoint": "delete_hidden_messages"})
288
-
289
- new_thread = self.client.beta.threads.create(messages=messages)
290
-
291
- # delete old thread
292
- self.client.beta.threads.delete(old_thread.id)
293
-
294
- # set current thread
295
- self.current_thread = new_thread
296
 
297
  @catch_error
298
  def cancel_run(self, run, thread = None):
 
280
  if old_thread is None:
281
  old_thread = self.current_thread
282
 
283
+ # find all hidden messages to delete
284
+ to_delete = [{'id': msg.id,'content': msg.content[0].text.value, 'role': msg.role} for msg in self.client.beta.threads.messages.list(self.current_thread.id, order="desc") if msg.content[0].text.value.startswith("[hidden]")]
285
+ logger.info(f"Marked for deletion= {to_delete}", extra={"user_id": self.user.user_id, "endpoint": "delete_hidden_messaged"})
286
+
287
+ for msg in to_delete:
288
+ # delete msg
289
+ deleted_message = self.client.beta.threads.messages.delete(
290
+ message_id=msg['id'],
291
+ thread_id=self.current_thread.id,
292
+ )
293
+ logger.info(f"Deleted Message = {deleted_message}", extra={"user_id": self.user.user_id, "endpoint": "delete_hidden_messaged"})
294
+ return True
 
295
 
296
  @catch_error
297
  def cancel_run(self, run, thread = None):