klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle add/delete file filter operation on non-existent conversation

Browse Source
This commit is contained in:
Debanjum
2024-10-30 02:23:34 -07:00
parent b3a63017b5
commit f64f5b3b6e
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/khoj/database/adapters/__init__.py
View File

@@ -1320,6 +1320,8 @@ class ConversationAdapters:
def add_files_to_filter(user: KhojUser, conversation_id: str, files: List[str]): def add_files_to_filter(user: KhojUser, conversation_id: str, files: List[str]):
conversation = ConversationAdapters.get_conversation_by_user(user, conversation_id=conversation_id) conversation = ConversationAdapters.get_conversation_by_user(user, conversation_id=conversation_id)
file_list = EntryAdapters.get_all_filenames_by_source(user, "computer") file_list = EntryAdapters.get_all_filenames_by_source(user, "computer")
if not conversation:
return []
for filename in files: for filename in files:
if filename in file_list and filename not in conversation.file_filters: if filename in file_list and filename not in conversation.file_filters:
conversation.file_filters.append(filename) conversation.file_filters.append(filename)
@@ -1333,6 +1335,8 @@ class ConversationAdapters:
@staticmethod @staticmethod
def remove_files_from_filter(user: KhojUser, conversation_id: str, files: List[str]): def remove_files_from_filter(user: KhojUser, conversation_id: str, files: List[str]):
conversation = ConversationAdapters.get_conversation_by_user(user, conversation_id=conversation_id) conversation = ConversationAdapters.get_conversation_by_user(user, conversation_id=conversation_id)
if not conversation:
return []
for filename in files: for filename in files:
if filename in conversation.file_filters: if filename in conversation.file_filters:
conversation.file_filters.remove(filename) conversation.file_filters.remove(filename)