feat: Add support for task types (TASK_QUESTION_ANSWERING, TASK_FACT_…

…VERIFICATION) through get_embeddings. PiperOrigin-RevId: 617262924
googleapis · Mar 19, 2024 · 9ec4590 · 9ec4590
1 parent b1cab3f
commit 9ec4590
Showing 1 changed file with 4 additions and 0 deletions.
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.
     """