Google Memorystore for Redis
Google Memorystore for Redis 是一項完全託管的服務，由 Redis 記憶體內資料儲存提供支援，用於建置應用程式快取，以提供次毫秒級的資料存取。擴展您的資料庫應用程式，以建置由 AI 驅動的體驗，並利用 Memorystore for Redis 的 Langchain 整合。
本筆記本說明如何使用 Memorystore for Redis，透過 MemorystoreDocumentLoader 和 MemorystoreDocumentSaver 儲存、載入和刪除 langchain 文件。
在 GitHub 上瞭解有關此套件的更多資訊。
開始之前
若要執行此筆記本，您需要執行下列操作
確認在此筆記本的執行階段環境中可存取資料庫後，填寫下列值並執行儲存格，然後再執行範例指令碼。
# @markdown Please specify an endpoint associated with the instance and a key prefix for demo purpose.
ENDPOINT = "redis://127.0.0.1:6379" # @param {type:"string"}
KEY_PREFIX = "doc:" # @param {type:"string"}
🦜🔗 程式庫安裝
此整合存在於其自身的 langchain-google-memorystore-redis 套件中，因此我們需要安裝它。
%pip install -upgrade --quiet langchain-google-memorystore-redis
僅限 Colab：取消註解下列儲存格以重新啟動核心，或使用按鈕重新啟動核心。對於 Vertex AI Workbench，您可以使用頂端的按鈕重新啟動終端機。
# # Automatically restart kernel after installs so that your environment can access the new packages
# import IPython
# app = IPython.Application.instance()
# app.kernel.do_shutdown(True)
☁ 設定您的 Google Cloud 專案
設定您的 Google Cloud 專案，以便您可以在此筆記本中利用 Google Cloud 資源。
如果您不知道您的專案 ID，請嘗試下列操作
- 執行
gcloud config list。 - 執行
gcloud projects list。 - 請參閱支援頁面：找出專案 ID。
# @markdown Please fill in the value below with your Google Cloud project ID and then run the cell.
PROJECT_ID = "my-project-id" # @param {type:"string"}
# Set the project id
!gcloud config set project {PROJECT_ID}
🔐 驗證
以登入此筆記本的 IAM 使用者身分驗證 Google Cloud，以便存取您的 Google Cloud 專案。
- 如果您使用 Colab 執行此筆記本，請使用下列儲存格並繼續。
- 如果您使用 Vertex AI Workbench，請查看 此處 的設定指示。
from google.colab import auth
auth.authenticate_user()
基本用法
儲存文件
使用 MemorystoreDocumentSaver.add_documents(<documents>) 儲存 langchain 文件。若要初始化 MemorystoreDocumentSaver 類別，您需要提供 2 個項目
client-redis.Redis用戶端物件。key_prefix- 用於在 Redis 中儲存文件的金鑰前置字串。
文件將儲存到隨機產生的金鑰中，並具有指定的 key_prefix 前置字串。或者，您可以在 add_documents 方法中指定 ids，以指定金鑰的後置字串。
import redis
from langchain_core.documents import Document
from langchain_google_memorystore_redis import MemorystoreDocumentSaver
test_docs = [
Document(
page_content="Apple Granny Smith 150 0.99 1",
metadata={"fruit_id": 1},
),
Document(
page_content="Banana Cavendish 200 0.59 0",
metadata={"fruit_id": 2},
),
Document(
page_content="Orange Navel 80 1.29 1",
metadata={"fruit_id": 3},
),
]
doc_ids = [f"{i}" for i in range(len(test_docs))]
redis_client = redis.from_url(ENDPOINT)
saver = MemorystoreDocumentSaver(
client=redis_client,
key_prefix=KEY_PREFIX,
content_field="page_content",
)
saver.add_documents(test_docs, ids=doc_ids)
載入文件
初始化載入器，以載入儲存在 Memorystore for Redis 執行個體中具有特定前置字串的所有文件。
使用 MemorystoreDocumentLoader.load() 或 MemorystoreDocumentLoader.lazy_load() 載入 langchain 文件。lazy_load 會傳回產生器，該產生器只會在迭代期間查詢資料庫。若要初始化 MemorystoreDocumentLoader 類別，您需要提供
client-redis.Redis用戶端物件。key_prefix- 用於在 Redis 中儲存文件的金鑰前置字串。
import redis
from langchain_google_memorystore_redis import MemorystoreDocumentLoader
redis_client = redis.from_url(ENDPOINT)
loader = MemorystoreDocumentLoader(
client=redis_client,
key_prefix=KEY_PREFIX,
content_fields=set(["page_content"]),
)
for doc in loader.lazy_load():
print("Loaded documents:", doc)
刪除文件
使用 MemorystoreDocumentSaver.delete() 刪除 Memorystore for Redis 執行個體中具有指定前置字串的所有金鑰。如果您知道，也可以指定金鑰的後置字串。
docs = loader.load()
print("Documents before delete:", docs)
saver.delete(ids=[0])
print("Documents after delete:", loader.load())
saver.delete()
print("Documents after delete all:", loader.load())
進階用法
自訂文件頁面內容 & Metadata
使用超過 1 個內容欄位初始化載入器時，載入之文件的 page_content 將包含 JSON 編碼的字串，其中最上層欄位等於 content_fields 中指定的欄位。
如果指定了 metadata_fields，則載入之文件的 metadata 欄位將只具有最上層欄位，且等於指定的 metadata_fields。如果任何中繼資料欄位的值儲存為 JSON 編碼的字串，則在載入至中繼資料欄位之前，將會先解碼。
loader = MemorystoreDocumentLoader(
client=redis_client,
key_prefix=KEY_PREFIX,
content_fields=set(["content_field_1", "content_field_2"]),
metadata_fields=set(["title", "author"]),
)