feat: rag

2026-05-06 11:35:10 +08:00
commit a17c65c4bc
75 changed files with 5196 additions and 0 deletions
@@ -0,0 +1,31 @@
+# 创建临时客户端
+
+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]]
+# }