feat: Add support for task type (CODE_RETRIEVAL_QUERY) through get_em…

…beddings. PiperOrigin-RevId: 659711828
googleapis · Aug 5, 2024 · f2ce1e4 · f2ce1e4
1 parent b73ef3e
commit f2ce1e4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -2092,6 +2092,8 @@ class TextEmbeddingInput:
                 Specifies that the embeddings will be used for question answering.
             FACT_VERIFICATION
                 Specifies that the embeddings will be used for fact verification.
+            CODE_RETRIEVAL_QUERY
+                Specifies that the embeddings will be used for code retrieval.
         title: Optional identifier of the text content.
     """