Google PaLM Embeddings¶

如果您在colab上打开这个笔记本，您可能需要安装LlamaIndex 🦙。

In [ ]:

%pip install llama-index-embeddings-google

In [ ]:

!pip install llama-index

In [ ]:

# 导入
from llama_index.embeddings.google import GooglePaLMEmbedding

In [ ]:

# 获取API密钥并创建嵌入

model_name = "models/embedding-gecko-001"
api_key = "YOUR API KEY"

embed_model = GooglePaLMEmbedding(model_name=model_name, api_key=api_key)

embeddings = embed_model.get_text_embedding("Google PaLM Embeddings.")

/opt/homebrew/lib/python3.11/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html
  from .autonotebook import tqdm as notebook_tqdm

In [ ]:

print(f"Dimension of embeddings: {len(embeddings)}")

Dimension of embeddings: 768

In [ ]:

embeddings[:5]

Out[ ]:

[0.028517298, -0.0028859433, -0.035110522, 0.021982985, -0.0039763353]