# 创建临时客户端

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]]
# }