heyong.fu
32 lines
813 B
Python
32 lines
813 B
Python
# 创建临时客户端
|
|
|
|
import chromadb
|
|
|
|
# 创建一个临时的内存客户端（不会保存到硬盘）
|
|
client = chromadb.EphemeralClient()
|
|
|
|
# 创建一个集合
|
|
collection = client.create_collection(name="test")
|
|
|
|
# 添加一条数据
|
|
collection.add(
|
|
documents=["今天天气有风", "很冷", "注意保暖", "加油学习"],
|
|
ids=["test_1", "test_2", "test_3", "test_4"],
|
|
)
|
|
|
|
# 查询数据
|
|
results = collection.query(query_texts=["天气"], n_results=2)
|
|
|
|
print(f"打印数据结果{results}")
|
|
|
|
# {
|
|
# 'ids': [['test_1', 'test_2']],
|
|
# 'embeddings': None,
|
|
# 'documents': [['今天天气有风', '很冷']],
|
|
# 'uris': None,
|
|
# 'included': ['metadatas', 'documents', 'distances'],
|
|
# 'data': None,
|
|
# 'metadatas': [[None, None]],
|
|
# 'distances': [[0.2988046705722809, 0.9478188753128052]]
|
|
# }
|