Add vertex search tool with integration tests

[amrit110]

Summary

Clickup Ticket(s): Link(s) if applicable.

Type of Change

• 🐛 Bug fix (non-breaking change that fixes an issue)
• ✨ New feature (non-breaking change that adds functionality)
• 💥 Breaking change (fix or feature that would cause existing functionality to not work as expected)
• 📝 Documentation update
• 🔧 Refactoring (no functional changes)
• ⚡ Performance improvement
• 🧪 Test improvements
• 🔒 Security fix

Changes Made

• Add a new tool that allows for searching over custom knowledge base using vertex AI
• Add tests including a complete integration test using a dummy database
• Add a notebook showing how the custom knowledge base can be setup and integrated

Testing

• Tests pass locally (uv run pytest tests/)
• Type checking passes (uv run mypy <src_dir>)
• Linting passes (uv run ruff check src_dir/)
• Manual testing performed (describe below)

Manual testing details:

Screenshots/Recordings

Related Issues

Deployment Notes

Checklist

• Code follows the project's style guidelines
• Self-review of code completed
• Documentation updated (if applicable)
• No sensitive information (API keys, credentials) exposed

[lotif]

Approved with minor comments.

[lotif]

When is the user supposed to run this script? I don't see it on any README or notebook.

[amrit110]

Hmm, user won't run this. We will run it for the participants. I'll add another README or something later for internal documentation.