Default to gpt-4o-mini instead of gpt-3.5-turbo in tests, func args

GPT-4o-mini is cheaper, smarter and can hold more context than
GPT-3.5-turbo. In production, we also default to gpt-4o-mini, so makes
sense to upgrade defaults and tests to work with it
This commit is contained in:
Debanjum Singh Solanky
2024-08-22 19:04:49 -07:00
parent 8a4c20d59a
commit 9986c183ea
5 changed files with 7 additions and 6 deletions

View File

@@ -6,7 +6,7 @@ from khoj.processor.conversation import utils
class TestTruncateMessage:
max_prompt_size = 10
model_name = "gpt-3.5-turbo"
model_name = "gpt-4o-mini"
encoder = tiktoken.encoding_for_model(model_name)
def test_truncate_message_all_small(self):