Skip to content

Commit

Permalink
chore: Update output_dimensionality docstring to a more generic descr…
Browse files Browse the repository at this point in the history
…iption.

PiperOrigin-RevId: 721564654
  • Loading branch information
vertex-sdk-bot authored and copybara-github committed Jan 31, 2025
1 parent ef596f5 commit 58fbabd
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions vertexai/language_models/_language_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -2135,7 +2135,8 @@ def _prepare_text_embedding_request(
Args:
texts(str): A list of texts or `TextEmbeddingInput` objects to embed.
auto_truncate(bool): Whether to automatically truncate long texts. Default: True.
output_dimensionality: Optional dimensions of embeddings. Range: [1, 768]. Default: None.
output_dimensionality: Optional dimensions of embeddings. This must be a positive
integer, with a maximum value dependent upon the model version. Default: None.
Returns:
A `_MultiInstancePredictionRequest` object.
Expand Down Expand Up @@ -2175,7 +2176,8 @@ def get_embeddings(
Args:
texts: A list of texts or `TextEmbeddingInput` objects to embed.
auto_truncate: Whether to automatically truncate long texts. Default: True.
output_dimensionality: Optional dimensions of embeddings. Range: [1, 768]. Default: None.
output_dimensionality: Optional dimensions of embeddings. This must be a positive
integer, with a maximum value dependent upon the model version. Default: None.
Returns:
A list of `TextEmbedding` objects.
Expand Down Expand Up @@ -2210,7 +2212,8 @@ async def get_embeddings_async(
Args:
texts: A list of texts or `TextEmbeddingInput` objects to embed.
auto_truncate: Whether to automatically truncate long texts. Default: True.
output_dimensionality: Optional dimensions of embeddings. Range: [1, 768]. Default: None.
output_dimensionality: Optional dimensions of embeddings. This must be a positive
integer, with a maximum value dependent upon the model version. Default: None.
Returns:
A list of `TextEmbedding` objects.
Expand Down

0 comments on commit 58fbabd

Please sign in to comment.