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"