feat: 调整文案

This commit is contained in:
李如威 2025-07-28 23:34:49 +08:00
parent 19926f5706
commit 34e78d0ce0
1 changed files with 9 additions and 8 deletions

View File

@ -1,6 +1,7 @@
import sys import sys
import os 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 from base_rag import BaseRAG
@ -9,7 +10,7 @@ class SimpleRAG(BaseRAG):
def ingest(self, documents): def ingest(self, documents):
for doc in documents: for doc in documents:
self.vector_store.add_texts([doc]) self.vector_store.add_texts([doc])
def query(self, question, k=3): def query(self, question, k=3):
docs = self.vector_store.similarity_search(question, k=k) docs = self.vector_store.similarity_search(question, k=k)
return docs return docs
@ -18,23 +19,23 @@ class SimpleRAG(BaseRAG):
def main(): def main():
config = { config = {
"model_name": "sentence-transformers/all-MiniLM-L6-v2", "model_name": "sentence-transformers/all-MiniLM-L6-v2",
"embedding_type": "local" "embedding_type": "local",
} }
rag = SimpleRAG(embedding_config=config) rag = SimpleRAG(embedding_config=config)
print("RAG初始化完成!") print("RAG初始化完成!")
# 添加一些文档 # 添加一些文档
documents = [ documents = [
"苹果是一种水果,味道甜美,营养丰富。", "苹果是一种水果,味道甜美,营养丰富。",
"苹果公司是一家科技公司生产iPhone和Mac等产品。", "苹果公司是一家科技公司生产iPhone和Mac等产品。",
"Python是一种编程语言简单易学功能强大。" "Python是一种编程语言简单易学功能强大。",
] ]
print("正在添加文档...") print("正在添加文档...")
rag.ingest(documents) rag.ingest(documents)
print("文档添加完成!") print("文档添加完成!")
# 测试查询 # 测试查询
print("\n正在查询: '什么是苹果?'") print("\n正在查询: '什么是苹果?'")
result = rag.query("什么是苹果?") result = rag.query("什么是苹果?")