test: Adds test for streaming + function calls

Browse Source

This commit includes a number of new tests for live streaming with function calls. These tests cover various scenarios:
- Single function calls
- Multiple function calls
- Parallel function calls
- Function calls with errors
- Synchronous function calls
- Simple streaming tools
- Video streaming tools
- Stopping a streaming tool
- Multiple streaming tools simultaneously

The tests use mock models and custom runners to simulate the interaction between the agent, model, and tools. They verify that function calls are correctly generated, executed, and that the expected data is returned.

PiperOrigin-RevId: 781318483

...

This commit is contained in:

Hangfei Lin

2025-07-09 19:41:27 -07:00

committed by Copybara-Service

parent da34692231

commit d792018c09

1 changed files with 828 additions and 329 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

tests/unittests/streaming/test_streaming.py

+828 -329

File diff suppressed because it is too large Load Diff