[Agents] Add zod import to example

[ashishajr]

Summary

Added the missing zod import to the tools example. The example in docs is a reference to this example which correctly imports zod .

Documentation checklist

• The change adheres to the documentation style guide.