From fe8da2cd83c7a734b67f890af751fc5ea0b5348c Mon Sep 17 00:00:00 2001 From: fcogidi <41602287+fcogidi@users.noreply.github.com> Date: Tue, 9 Dec 2025 11:28:39 -0500 Subject: [PATCH] Update model version in verbose.py and test_integration.py to "gemini-2.5-flash-lite" --- src/2_frameworks/2_multi_agent/verbose.py | 2 +- tests/tool_tests/test_integration.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/2_frameworks/2_multi_agent/verbose.py b/src/2_frameworks/2_multi_agent/verbose.py index a59128c..4edc409 100644 --- a/src/2_frameworks/2_multi_agent/verbose.py +++ b/src/2_frameworks/2_multi_agent/verbose.py @@ -151,7 +151,7 @@ async def _main(question: str, gr_messages: list[ChatMessage]): instructions=RESEARCHER_INSTRUCTIONS, tools=[agents.function_tool(async_knowledgebase.search_knowledgebase)], model=agents.OpenAIChatCompletionsModel( - model="gemini-2.5-flash-lite-preview-06-17", + model="gemini-2.5-flash-lite", openai_client=async_openai_client, ), model_settings=agents.ModelSettings(tool_choice="required"), diff --git a/tests/tool_tests/test_integration.py b/tests/tool_tests/test_integration.py index 576ef89..326f69d 100644 --- a/tests/tool_tests/test_integration.py +++ b/tests/tool_tests/test_integration.py @@ -90,7 +90,7 @@ async def test_weaviate_kb_tool_and_llm( }, ] response = await client.chat.completions.create( - model="gemini-2.5-flash-lite-preview-06-17", messages=messages + model="gemini-2.5-flash-lite", messages=messages ) message = response.choices[0].message assert message.role == "assistant"