everything working again after cleanup
This commit is contained in:
@@ -125,7 +125,9 @@ def test_custom_bot_command_success(
|
||||
|
||||
asyncio.run(
|
||||
main_module.custom_bot(
|
||||
mock_ctx, bot_name="alfred", personality="you are a british butler",
|
||||
mock_ctx,
|
||||
bot_name="alfred",
|
||||
personality="you are a british butler",
|
||||
),
|
||||
)
|
||||
|
||||
@@ -199,7 +201,9 @@ def test_custom_bot_command_create_fails(
|
||||
|
||||
asyncio.run(
|
||||
main_module.custom_bot(
|
||||
mock_ctx, bot_name="alfred", personality="you are a british butler",
|
||||
mock_ctx,
|
||||
bot_name="alfred",
|
||||
personality="you are a british butler",
|
||||
),
|
||||
)
|
||||
call_args = mock_ctx.send.call_args[0][0]
|
||||
@@ -347,7 +351,9 @@ def test_handle_chat_success(
|
||||
|
||||
import vibe_bot.main as main_module
|
||||
|
||||
mock_llama_wrapper.chat_completion_with_history.return_value = "This is a bot response" # noqa: E501
|
||||
mock_llama_wrapper.chat_completion_with_history.return_value = (
|
||||
"This is a bot response"
|
||||
)
|
||||
|
||||
asyncio.run(
|
||||
main_module.handle_chat(
|
||||
|
||||
Reference in New Issue
Block a user