test: add complex gemini tests (#1853)

Co-authored-by: Matt Zhou <mattzhou@Matts-MacBook-Pro.local>
2024-10-09 10:00:21 -07:00
parent fa0b0a5b9f
commit c0501a743d
5 changed files with 50 additions and 18 deletions
--- a/tests/helpers/endpoints_helper.py
+++ b/tests/helpers/endpoints_helper.py
@@ -132,7 +132,6 @@ def check_first_response_is_valid_for_llm_endpoint(filename: str) -> ChatComplet

    # Get inner_thoughts_in_kwargs
    inner_thoughts_in_kwargs = derive_inner_thoughts_in_kwargs(OptionState.DEFAULT, agent_state.llm_config.model)
-
    # Assert that the message has an inner monologue
    assert_contains_correct_inner_monologue(choice, inner_thoughts_in_kwargs)