From 34e78d0ce0d0f5566222466aa2a1da864e967a14 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=A6=82=E5=A8=81?= Date: Mon, 28 Jul 2025 23:34:49 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E8=B0=83=E6=95=B4=E6=96=87=E6=A1=88?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- examples/quick_start.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/examples/quick_start.py b/examples/quick_start.py index afceb0a..4a65706 100644 --- a/examples/quick_start.py +++ b/examples/quick_start.py @@ -1,6 +1,7 @@ import sys import os -sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'src')) + +sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src")) from base_rag import BaseRAG @@ -9,7 +10,7 @@ class SimpleRAG(BaseRAG): def ingest(self, documents): for doc in documents: self.vector_store.add_texts([doc]) - + def query(self, question, k=3): docs = self.vector_store.similarity_search(question, k=k) return docs @@ -18,23 +19,23 @@ class SimpleRAG(BaseRAG): def main(): config = { "model_name": "sentence-transformers/all-MiniLM-L6-v2", - "embedding_type": "local" + "embedding_type": "local", } - + rag = SimpleRAG(embedding_config=config) print("RAG初始化完成!") - + # 添加一些文档 documents = [ "苹果是一种水果,味道甜美,营养丰富。", "苹果公司是一家科技公司,生产iPhone和Mac等产品。", - "Python是一种编程语言,简单易学,功能强大。" + "Python是一种编程语言,简单易学,功能强大。", ] - + print("正在添加文档...") rag.ingest(documents) print("文档添加完成!") - + # 测试查询 print("\n正在查询: '什么是苹果?'") result = rag.query("什么是苹果?")