Skip to content

Add workers to create, update and cleanup vertex embeddings

Alexandru Croitor requested to merge workers_to_create_embedings_on_vertex_gitlab_docs into master

What does this MR do and why?

Add model and DB tables to support vertex embeddings for docs

This MR adds database table vertex_gitlab_docs to store vertex embeddigns for gitlab documentation.

This also adds the text_embeddings method to the VertexAI::Client class to call vertex ai endpoint for building text embeddings.

re https://gitlab.com/gitlab-org/gitlab/-/issues/420939

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.

Merge request reports