Xata
Xata 是一个基于 PostgreSQL 的无服务器数据平台。它提供了一个 Python SDK 用于与您的数据库交互，以及一个用于管理数据的 UI。Xata 具有本机向量类型，可以添加到任何表中，并支持相似性搜索。LangChain 将向量直接插入 Xata，并查询它以查找给定向量的最近邻居，因此您可以将所有 LangChain 嵌入集成与 Xata 一起使用。
此笔记本向您介绍如何使用 Xata 作为 VectorStore。
设置
创建一个用作向量存储的数据库
在 Xata UI 中创建一个新数据库。您可以随意命名它，在此笔记本中，我们将使用 langchain。创建一个表，同样，您可以随意命名它，但我们将使用 vectors。通过 UI 添加以下列
content类型为 "Text"。它用于存储Document.pageContent值。embedding类型为 "Vector"。使用您计划使用的模型所使用的维度。在此笔记本中，我们使用 OpenAI 嵌入，它具有 1536 个维度。source类型为 "Text"。此示例将此用作元数据列。- 您想用作元数据的任何其他列。它们从
Document.metadata对象中填充。例如，如果在Document.metadata对象中您有一个title属性，您可以在表中创建一个title列，它将被填充。
让我们首先安装我们的依赖项
%pip install --upgrade --quiet xata langchain-openai langchain-community tiktoken langchain
让我们将 OpenAI 密钥加载到环境中。如果您没有，您可以创建一个 OpenAI 帐户并在此 页面 上创建一个密钥。
import getpass
import os
os.environ["OPENAI_API_KEY"] = getpass.getpass("OpenAI API Key:")
同样，我们需要获取 Xata 的环境变量。您可以通过访问您的 帐户设置 来创建一个新的 API 密钥。要查找数据库 URL，请转到您创建的数据库的“设置”页面。数据库 URL 应该类似于：https://demo-uni3q8.eu-west-1.xata.sh/db/langchain。
api_key = getpass.getpass("Xata API key: ")
db_url = input("Xata database URL (copy it from your DB settings):")
from langchain_community.document_loaders import TextLoader
from langchain_community.vectorstores.xata import XataVectorStore
from langchain_openai import OpenAIEmbeddings
from langchain_text_splitters import CharacterTextSplitter
创建 Xata 向量存储
让我们导入我们的测试数据集
loader = TextLoader("../../how_to/state_of_the_union.txt")
documents = loader.load()
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
docs = text_splitter.split_documents(documents)
embeddings = OpenAIEmbeddings()
现在创建实际的向量存储，它由 Xata 表支持。
vector_store = XataVectorStore.from_documents(
docs, embeddings, api_key=api_key, db_url=db_url, table_name="vectors"
)
运行上述命令后，如果您转到 Xata UI，您应该看到文档及其嵌入一起加载。要使用已包含向量内容的现有 Xata 表，请初始化 XataVectorStore 构造函数
vector_store = XataVectorStore(
api_key=api_key, db_url=db_url, embedding=embeddings, table_name="vectors"
)
相似性搜索
query = "What did the president say about Ketanji Brown Jackson"
found_docs = vector_store.similarity_search(query)
print(found_docs)
带有分数（向量距离）的相似性搜索
query = "What did the president say about Ketanji Brown Jackson"
result = vector_store.similarity_search_with_score(query)
for doc, score in result:
print(f"document={doc}, score={score}")