Fix followup issues in ZeroShot Agent

What does this MR do and why?

Addresses followup issues https://gitlab.com/gitlab-org/gitlab/-/issues/412741 & https://gitlab.com/gitlab-org/gitlab/-/issues/412641

Screenshots or screen recordings

Screenshots are required for UI changes, and strongly recommended for all other merge requests.

Before	After

How to set up and validate locally

Numbered steps to set up and validate the change are strongly suggested.

MR acceptance checklist

This checklist encourages us to confirm any changes have been analyzed to reduce risks in quality, performance, reliability, security, and maintainability.

• I have evaluated the MR acceptance checklist for this MR.