Add pydantic import

[SHARANTANGEDA]

What does this PR address?

Add Pydantic import

Thanks for sending a pull request!

Fixes #(issue)

Before submitting:

• Does the Pull Request follow Conventional Commits specification naming? Here are GitHub's
  guide on how to create a pull request.
• Does the code follow mlmonitor's code style, pre-commit run -a script has passed ?
• Did your changes require updates to the documentation? Have you updated
  those accordingly?
• Did you write tests to cover your changes?