Spaces:
Sleeping
Sleeping
Shageenderan Sapai commited on
Commit ·
6bf62ed
1
Parent(s): 319dccb
delete_hidden_messages updated logic
Browse files- 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 |
-
#
|
| 284 |
-
|
| 285 |
-
|
| 286 |
-
|
| 287 |
-
|
| 288 |
-
|
| 289 |
-
|
| 290 |
-
|
| 291 |
-
|
| 292 |
-
|
| 293 |
-
|
| 294 |
-
|
| 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):
|