()
| 320 | logger.info(f"| --------------------------------------------------") |
| 321 | |
| 322 | async def test_search(): |
| 323 | logger.info(f"| --------------------------------------------------") |
| 324 | logger.info(f"| Testing search with different models") |
| 325 | models = [ |
| 326 | "openrouter/gemini-3-flash-preview-plugins" |
| 327 | ] |
| 328 | |
| 329 | messages = [ |
| 330 | SystemMessage(content="You are a helpful assistant."), |
| 331 | HumanMessage(content=[ |
| 332 | ContentPartText(text="Please search the web for the latest news about the AAPL stock."), |
| 333 | ]), |
| 334 | ] |
| 335 | |
| 336 | for model in models: |
| 337 | logger.info(f"| Testing {model}") |
| 338 | response = await model_manager(model=model, messages=messages) |
| 339 | logger.info(f"| {model} Response: {json.dumps(response.model_dump(), indent=4)}") |
| 340 | logger.info(f"| --------------------------------------------------") |
| 341 | |
| 342 | def parse_args(): |
| 343 | parser = argparse.ArgumentParser(description='main') |
nothing calls this directly
no test coverage detected