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

Remove output_modes function reference in chat tests

Browse Source
This commit is contained in:
sabaimran
2024-11-15 14:03:07 -08:00
parent e3f1ea9dee
commit c77dc84a68
2 changed files with 0 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
tests/test_offline_chat_actors.py
View File

@@ -18,7 +18,6 @@ from khoj.processor.conversation.offline.chat_model import (
)
from khoj.processor.conversation.offline.utils import download_model
from khoj.processor.conversation.utils import message_to_log
from khoj.routers.helpers import aget_relevant_output_modes
from khoj.utils.constants import default_offline_chat_models
@@ -549,34 +548,6 @@ def test_filter_questions():
assert filtered_questions[0] == "Who is on the basketball team?"
# ----------------------------------------------------------------------------------------------------
@pytest.mark.anyio
@pytest.mark.django_db(transaction=True)
async def test_use_text_response_mode(client_offline_chat):
# Arrange
user_query = "What's the latest in the Israel/Palestine conflict?"
# Act
mode = await aget_relevant_output_modes(user_query, {})
# Assert
assert mode.value == "text"
# ----------------------------------------------------------------------------------------------------
@pytest.mark.anyio
@pytest.mark.django_db(transaction=True)
async def test_use_image_response_mode(client_offline_chat):
# Arrange
user_query = "Paint a picture of the scenery in Timbuktu in the winter"
# Act
mode = await aget_relevant_output_modes(user_query, {})
# Assert
assert mode.value == "image"
# Helpers
# ----------------------------------------------------------------------------------------------------
def populate_chat_history(message_list):