fix: import statements for ZaiClient in batches and files

[wellenzheng]

… modules

Description

Thank you for opening a Pull Request!
Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

• Follow the CONTRIBUTING Guide.
• Make your Pull Request title in the https://www.conventionalcommits.org/ specification.
• Ensure the tests pass (Run mvn clean test from the repository root)
• Appropriate docs were updated (if necessary)

Fixes #<issue_number_goes_here>