diff --git a/libs/vertexai/langchain_google_vertexai/embeddings.py b/libs/vertexai/langchain_google_vertexai/embeddings.py index c5563597..29edab23 100644 --- a/libs/vertexai/langchain_google_vertexai/embeddings.py +++ b/libs/vertexai/langchain_google_vertexai/embeddings.py @@ -83,9 +83,9 @@ def __init__( self.instance["task_executor"] = ThreadPoolExecutor( max_workers=request_parallelism ) - self.instance["embeddings_task_type_supported"] = ( - not self.client._endpoint_name.endswith("/textembedding-gecko@001") - ) + self.instance[ + "embeddings_task_type_supported" + ] = not self.client._endpoint_name.endswith("/textembedding-gecko@001") @staticmethod def _split_by_punctuation(text: str) -> List[str]: