Skip to content

Change unit primitive for text embeddings

Madelein van Niekerk requested to merge 457724-unit-primitive-for-embeddings into master

What does this MR do and why?

Updates the unit primitive for embeddings to accommodate semantic_search_issue added in Unit primitive for issue embeddings (gitlab-org/modelops/applied-ml/code-suggestions/ai-assist!918 - merged)

Currently embeddings all use the documentation_search primitive but since embeddings are now being generated for other cases, we want that to be passed in and set by the calling class.

MR acceptance checklist

Please evaluate this MR against the MR acceptance checklist. It helps you analyze changes to reduce risks in quality, performance, reliability, security, and maintainability.

Related to #457724 (closed)

Edited by Madelein van Niekerk

Merge request reports