Skip to main content
Open In Colab在 GitHub 上打开

分析型数据库

云原生数据仓库 AnalyticDB PostgreSQL 版是一种大规模并行处理 （MPP） 数据仓库服务，旨在在线分析大量数据。

AnalyticDB for PostgreSQL是基于开源的Greenplum Database项目，并通过Alibaba Cloud.AnalyticDB PostgreSQL 版兼容 ANSI SQL 2003 语法，兼容 PostgreSQL 和 Oracle 数据库生态。AnalyticDB PostgreSQL 版还支持行存储和列存储。云原生数据仓库 AnalyticDB PostgreSQL 版支持高性能的 PB 级离线数据处理，并支持高并发在线查询。

您需要安装langchain-community跟pip install -qU langchain-community使用此集成

此笔记本展示了如何使用与AnalyticDBvector 数据库。 要运行，您应该启动并运行一个 AnalyticDB 实例：

from langchain_community.vectorstores import AnalyticDB
from langchain_openai import OpenAIEmbeddings
from langchain_text_splitters import CharacterTextSplitter

通过调用 OpenAI API 拆分文档并获取嵌入

from langchain_community.document_loaders import TextLoader

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()
API 参考：TextLoader

通过设置相关 ENVIRONMENTS 连接到 AnalyticDB。

export PG_HOST={your_analyticdb_hostname}
export PG_PORT={your_analyticdb_port} # Optional, default is 5432
export PG_DATABASE={your_database} # Optional, default is postgres
export PG_USER={database_username}
export PG_PASSWORD={database_password}

然后将您的嵌入和文档存储到 AnalyticDB 中

import os

connection_string = AnalyticDB.connection_string_from_db_params(
    driver=os.environ.get("PG_DRIVER", "psycopg2cffi"),
    host=os.environ.get("PG_HOST", "localhost"),
    port=int(os.environ.get("PG_PORT", "5432")),
    database=os.environ.get("PG_DATABASE", "postgres"),
    user=os.environ.get("PG_USER", "postgres"),
    password=os.environ.get("PG_PASSWORD", "postgres"),
)

vector_db = AnalyticDB.from_documents(
    docs,
    embeddings,
    connection_string=connection_string,
)

查询和检索数据

query = "What did the president say about Ketanji Brown Jackson"
docs = vector_db.similarity_search(query)
print(docs[0].page_content)
Tonight. I call on the Senate to: Pass the Freedom to Vote Act. Pass the John Lewis Voting Rights Act. And while you’re at it, pass the Disclose Act so Americans can know who is funding our elections. 

Tonight, I’d like to honor someone who has dedicated his life to serve this country: Justice Stephen Breyer—an Army veteran, Constitutional scholar, and retiring Justice of the United States Supreme Court. Justice Breyer, thank you for your service. 

One of the most serious constitutional responsibilities a President has is nominating someone to serve on the United States Supreme Court. 

And I did that 4 days ago, when I nominated Circuit Court of Appeals Judge Ketanji Brown Jackson. One of our nation’s top legal minds, who will continue Justice Breyer’s legacy of excellence.