6大核心模块(Modules)
示例(Examples)
Tair

LangChain

Tair#

本笔记展示如何使用与Tair向量数据库相关的功能。 要运行,请确保已经启动了Tair (opens in a new tab)实例。

from langchain.embeddings.fake import FakeEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from langchain.vectorstores import Tair
 
from langchain.document_loaders import TextLoader
loader = TextLoader('../../../state_of_the_union.txt')
documents = loader.load()
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
docs = text_splitter.split_documents(documents)
 
embeddings = FakeEmbeddings(size=128)
 

使用TAIR_URL环境变量或关键字参数tair_url连接到Tair。

export TAIR_URL="redis://{username}:{password}@{tair_address}:{tair_port}"
 

然后将文档和嵌入存储到Tair中。

查询相似的文档。

tair_url = "redis://localhost:6379"
 
# drop first if index already exists
Tair.drop_index(tair_url=tair_url)
 
vector_store = Tair.from_documents(
    docs,
    embeddings,
    tair_url=tair_url
)
 

查询相似的文档。

query = "What did the president say about Ketanji Brown Jackson"
docs = vector_store.similarity_search(query)
docs[0]
 
Document(page_content='We’re going after the criminals who stole billions in relief money meant for small businesses and millions of Americans.    And tonight, I’m announcing that the Justice Department will name a chief prosecutor for pandemic fraud.   By the end of this year, the deficit will be down to less than half what it was before I took office.    The only president ever to cut the deficit by more than one trillion dollars in a single year.   Lowering your costs also means demanding more competition.   I’m a capitalist, but capitalism without competition isn’t capitalism.   It’s exploitation—and it drives up prices.   When corporations don’t have to compete, their profits go up, your prices go up, and small businesses and family farmers and ranchers go under.   We see it happening with ocean carriers moving goods in and out of America.   During the pandemic, these foreign-owned companies raised prices by as much as 1,000% and made record profits.', metadata={'source': '../../../state_of_the_union.txt'})