[Update] Default LLM (#3587)
This commit is contained in:
+3
-3
@@ -68,14 +68,14 @@ def test_completions_create(mock_memory_client, mock_litellm):
|
||||
mock_litellm.completion.return_value = {"choices": [{"message": {"content": "I'm doing well, thank you!"}}]}
|
||||
mock_litellm.supports_function_calling.return_value = True
|
||||
|
||||
response = completions.create(model="gpt-4o-mini", messages=messages, user_id="test_user", temperature=0.7)
|
||||
response = completions.create(model="gpt-4.1-nano-2025-04-14", messages=messages, user_id="test_user", temperature=0.7)
|
||||
|
||||
mock_memory_client.add.assert_called_once()
|
||||
mock_memory_client.search.assert_called_once()
|
||||
|
||||
mock_litellm.completion.assert_called_once()
|
||||
call_args = mock_litellm.completion.call_args[1]
|
||||
assert call_args["model"] == "gpt-4o-mini"
|
||||
assert call_args["model"] == "gpt-4.1-nano-2025-04-14"
|
||||
assert len(call_args["messages"]) == 2
|
||||
assert call_args["temperature"] == 0.7
|
||||
|
||||
@@ -93,7 +93,7 @@ def test_completions_create_with_system_message(mock_memory_client, mock_litellm
|
||||
mock_litellm.completion.return_value = {"choices": [{"message": {"content": "I'm doing well, thank you!"}}]}
|
||||
mock_litellm.supports_function_calling.return_value = True
|
||||
|
||||
completions.create(model="gpt-4o-mini", messages=messages, user_id="test_user")
|
||||
completions.create(model="gpt-4.1-nano-2025-04-14", messages=messages, user_id="test_user")
|
||||
|
||||
call_args = mock_litellm.completion.call_args[1]
|
||||
assert call_args["messages"][0]["role"] == "system"
|
||||
|
||||
Reference in New Issue
Block a user