feat: Add support for task types (TASK_QUESTION_ANSWERING, TASK_FACT_VERIFICATION) through get_embeddings.

vertex-sdk-bot · copybara-github · commit 9ec459014612 · 2024-03-19T12:48:33.000-07:00
PiperOrigin-RevId: 617262924
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -1969,6 +1969,10 @@ class TextEmbeddingInput:
                 Specifies that the given text will be classified.
             CLUSTERING
                 Specifies that the embeddings will be used for clustering.
+            QUESTION_ANSWERING
+                Specifies that the embeddings will be used for question answering.
+            FACT_VERIFICATION
+                Specifies that the embeddings will be used for fact verification.
         title: Optional identifier of the text content.
     """
 
