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

Update file summarization unit tests now that multiple files are allowed

Browse Source
This commit is contained in:
sabaimran
2024-11-04 17:45:54 -08:00
parent 7543360210
commit 1f372bf2b1
2 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_openai_chat_director.py
View File

@@ -312,7 +312,6 @@ def test_summarize_one_file(chat_client, default_user2: KhojUser):
# Assert
assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True)
@@ -344,7 +343,6 @@ def test_summarize_extra_text(chat_client, default_user2: KhojUser):
# Assert
assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True)
@@ -371,7 +369,7 @@ def test_summarize_multiple_files(chat_client, default_user2: KhojUser):
response_message = response.json()["response"]
# Assert
assert response_message == "Only one file can be selected for summarization."
assert response_message is not None
@pytest.mark.django_db(transaction=True)
@@ -435,7 +433,6 @@ def test_summarize_different_conversation(chat_client, default_user2: KhojUser):
assert (
response_message_conv1 != "No files selected for summarization. Please add files using the section on the left."
)
assert response_message_conv1 != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True)