diff --git a/tests/bot/test_querybot.py b/tests/bot/test_querybot.py index 13d3cc0ce..7d0228d46 100644 --- a/tests/bot/test_querybot.py +++ b/tests/bot/test_querybot.py @@ -8,21 +8,24 @@ collection_name=st.text( alphabet="abcdefghijklmnopqrstuvwxyz0123456789", min_size=4, max_size=63 ), - dummy_text=st.text(), - mock_response=st.text(), - human_message=st.text(), + dummy_text=st.text(min_size=400), + mock_response=st.text(min_size=4), + stream_target=st.one_of(st.just("panel"), st.just("stdout")), ) @settings(suppress_health_check=[HealthCheck.function_scoped_fixture], deadline=None) -def test_querybot_init( - tmp_path, system_prompt, collection_name, dummy_text, mock_response, human_message +def test_querybot( + tmp_path, system_prompt, collection_name, dummy_text, mock_response, stream_target ): """Test initialization of QueryBot.""" tempfile = tmp_path / "test.txt" tempfile.write_text(dummy_text) - QueryBot( + bot = QueryBot( system_prompt=system_prompt, collection_name=collection_name, document_paths=tempfile, mock_response=mock_response, + stream_target=stream_target, ) + + bot("How are you doing?")