diff --git a/test/agentchat/contrib/test_web_surfer.py b/test/agentchat/contrib/test_web_surfer.py index 855afc6b499..01d2cfd13e2 100644 --- a/test/agentchat/contrib/test_web_surfer.py +++ b/test/agentchat/contrib/test_web_surfer.py @@ -99,10 +99,10 @@ def test_web_surfer(): def test_web_surfer_oai(): llm_config = {"config_list": config_list, "timeout": 180, "cache_seed": None} + model = ["gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-16k"] + model = model + [m.replace(".", "") for m in model] summarizer_llm_config = { - "config_list": filter_config( - config_list, {"model": ["gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-16k"]} - ), + "config_list": filter_config(config_list, {"model": model}), "timeout": 180, "cache_seed": None, } diff --git a/test/test_browser_utils.py b/test/test_browser_utils.py index c650c6b9ba4..dcc15d1b83a 100644 --- a/test/test_browser_utils.py +++ b/test/test_browser_utils.py @@ -78,7 +78,9 @@ def test_simple_text_browser(): assert BLOG_POST_STRING in browser.page_content # Check if page splitting works - approx_pages = int(len(browser.page_content) / viewport_size + 0.5) # May be fewer, since it aligns to word breaks + approx_pages = ( + int(len(browser.page_content) / viewport_size + 0.5) + 1 + ) # May be fewer, since it aligns to word breaks assert len(browser.viewport_pages) <= approx_pages assert abs(len(browser.viewport_pages) - approx_pages) <= 1 # allow only a small deviation assert browser.viewport_pages[0][0] == 0