Vertex AI PaLM API is a service on Google Cloud exposing the embedding models.
Note: This integration is separate from the Google PaLM integration.
By default, Google Cloud does not use Customer Data to train its foundation models as part of Google Cloud`s AI/ML Privacy Commitment. More details about how Google processes data can also be found in Google’s Customer Data Processing Addendum (CDPA).
To use Vertex AI PaLM you must have the
package installed and either: - Have credentials configured for your
environment (gcloud, workload identity, etc…) - Store the path to a
service account JSON file as the GOOGLE_APPLICATION_CREDENTIALS
This codebase uses the
google.auth library which first looks for the
application credentials variable mentioned above, and then looks for
For more information, see: - https://cloud.google.com/docs/authentication/application-default-credentials#GAC - https://googleapis.dev/python/google-auth/latest/reference/google.auth.html#module-google.auth
#!pip install google-cloud-aiplatform
from langchain.embeddings import VertexAIEmbeddings
embeddings = VertexAIEmbeddings()
text = "This is a test document."
query_result = embeddings.embed_query(text)
doc_result = embeddings.embed_documents([text])