feat: Add support for task type (CODE_RETRIEVAL_QUERY) through get_embeddings.

vertex-sdk-bot · copybara-github · commit f2ce1e4caea9 · 2024-08-05T16:03:31.000-07:00
PiperOrigin-RevId: 659711828
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.
     """
 
