给出全部代码,用Python实现

以下是一个完整的、可运行的 Python 实现的 RAG(Retrieval-Augmented Generation)程序,具备以下功能: --- ### ✅ 功能说明: - 从本地 Ollama 获取模型列表 - 使用本地 SentenceTransformer 模型进行文档嵌入 - 使用 FAISS 构建向量数据库进行语义检索 - 支持参数控制(Top-K、Temperature、Max Tokens) - 使用 PyQt5 构建图形界面 - 支持文档加载与处理(使用 `langchain` 和 `unstructured`) --- ### 📁 项目结构(简化) ``` rag_app/ ├── main.py ├── documents/ │ └── sample.txt └── requirements.txt ``` --- ### 📦 安装依赖(`requirements.txt`) ```txt requests sentence-transformers faiss-cpu langchain unstructured PyQt5 ``` --- ### 🧠 `main.py`(完整代码) ```python import sys import os import requests import numpy as np from PyQt5.QtWidgets import QApplication, QWidget, QVBoxLayout, QLabel, QLineEdit, QPushButton, QTextEdit, QComboBox from sentence_transformers import SentenceTransformer from langchain.document_loaders import TextLoader from langchain.text_splitter import RecursiveCharacterTextSplitter import faiss # ========== Ollama 模型获取 ========== def get_ollama_models(): try: response = requests.get("http://localhost:11434/api/tags") if response.status_code == 200: return [model["name"] for model in response.json()["models"]] else: return ["连接失败"] except Exception as e: print("Ollama连接失败:", e) return ["连接失败"] # ========== 文档加载与处理 ========== def load_documents(folder_path="documents"): documents = [] for filename in os.listdir(folder_path): if filename.endswith(".txt"): loader = TextLoader(os.path.join(folder_path, filename)) documents.extend(loader.load()) text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=100) return text_splitter.split_documents(documents) # ========== 嵌入模型加载 ========== embedding_model = SentenceTransformer("all-MiniLM-L6-v2") # ========== 构建 FAISS 索引 ========== texts = [doc.page_content for doc in load_documents()] embeddings = embedding_model.encode(texts) dimension = embeddings.shape[1] index = faiss.IndexFlatL2(dimension) index.add(np.array(embeddings)) # ========== 检索函数 ========== def retrieve_documents(query, top_k=5): query_embedding = embedding_model.encode([query]) distances, indices = index.search(np.array(query_embedding), top_k) return [texts[i] for i in indices[0]] # ========== 生成回答函数 ========== def generate_answer(prompt, model_name, temperature=0.7, max_tokens=200): url = "http://localhost:11434/api/generate" payload = { "model": model_name, "prompt": prompt, "stream": False, "temperature": temperature, "max_tokens": max_tokens, } response = requests.post(url, json=payload) if response.status_code == 200: return response.json()["response"] else: return "生成失败:" + response.text # ========== RAG 主流程 ========== def rag_query(query, model_name, top_k=5, temperature=0.7, max_tokens=200): retrieved_docs = retrieve_documents(query, top_k) context = "\n".join(retrieved_docs) full_prompt = f"Context: {context}\n\nQuestion: {query}\n\nAnswer:" answer = generate_answer(full_prompt, model_name, temperature, max_tokens) return answer # ========== PyQt5 GUI ========== class RAGApp(QWidget): def __init__(self): super().__init__() self.setWindowTitle("RAG 程序 - 本地部署") self.layout = QVBoxLayout() self.model_label = QLabel("选择模型:") self.layout.addWidget(self.model_label) self.model_combo = QComboBox() self.model_combo.addItems(get_ollama_models()) self.layout.addWidget(self.model_combo) self.query_input = QLineEdit() self.query_input.setPlaceholderText("输入你的问题...") self.layout.addWidget(self.query_input) self.top_k_input = QLineEdit() self.top_k_input.setPlaceholderText("Top-K 检索数量 (默认5)") self.layout.addWidget(self.top_k_input) self.temperature_input = QLineEdit() self.temperature_input.setPlaceholderText("Temperature (默认0.7)") self.layout.addWidget(self.temperature_input) self.max_tokens_input = QLineEdit() self.max_tokens_input.setPlaceholderText("Max Tokens (默认200)") self.layout.addWidget(self.max_tokens_input) self.submit_button = QPushButton("提交") self.submit_button.clicked.connect(self.on_submit) self.layout.addWidget(self.submit_button) self.result_output = QTextEdit() self.result_output.setReadOnly(True) self.layout.addWidget(self.result_output) self.setLayout(self.layout) def on_submit(self): query = self.query_input.text() model_name = self.model_combo.currentText() top_k = int(self.top_k_input.text()) if self.top_k_input.text() else 5 temperature = float(self.temperature_input.text()) if self.temperature_input.text() else 0.7 max_tokens = int(self.max_tokens_input.text()) if self.max_tokens_input.text() else 200 answer = rag_query(query, model_name, top_k=top_k, temperature=temperature, max_tokens=max_tokens) self.result_output.setText(answer) # ========== 启动应用 ========== if __name__ == "__main__": app = QApplication(sys.argv) window = RAGApp() window.resize(600, 500) window.show() sys.exit(app.exec_()) ``` --- ### 📌 注意事项: 1. **Ollama 必须运行在本地**,启动命令:`ollama serve` 2. 需要提前下载模型,如:`ollama pull llama3` 或 `ollama pull qwen` 3. 放入 `documents/` 文件夹中的 `.txt` 文件将被自动加载 4. 如果使用其他格式(如 PDF),可扩展 `langchain` 支持的加载器 --- ### ✅ 示例运行流程: 1. 安装依赖:`pip install -r requirements.txt` 2. 启动 Ollama:`ollama serve` 3. 下载模型:`ollama pull llama3` 4. 放入文档到 `documents/` 文件夹 5. 运行程序:`python main.py` 6. 在界面中选择模型、输入问题、调节参数、查看结果 --- ###

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

为你推荐:

Python内容推荐

基于python实现的微信机器人程序源代码.zip

基于python实现的微信机器人程序源代码.zip

基于python实现的微信机器人程序源代码基于python实现的微信机器人程序源代码基于python实现的微信机器人程序源代码基于python实现的微信机器人程序源代码基于python实现的微信机器人程序源代码基于python实现的微信...

统一社会信用代码校验python实现

统一社会信用代码校验python实现

统一社会信用代码校验python实现,统一社会信用代码校验python实现

间调度问题代码(python实现).zip

间调度问题代码(python实现).zip

python间调度问题代码(python实现).zippython间调度问题代码(python实现).zippython间调度问题代码(python实现).zippython间调度问题代码(python实现).zippython间调度问题代码(python实现).zippython间调度问题代码...

基于python实现的今天吃什么程序源代码.zip

基于python实现的今天吃什么程序源代码.zip

基于python实现的今天吃什么程序源代码 基于python实现的今天吃什么程序源代码 基于python实现的今天吃什么程序源代码 基于python实现的今天吃什么程序源代码 基于python实现的今天吃什么程序源代码 基于python实现...

基于python实现的甜橙音乐网程序源代码.zip

基于python实现的甜橙音乐网程序源代码.zip

基于python实现的甜橙音乐网程序源代码 基于python实现的甜橙音乐网程序源代码 基于python实现的甜橙音乐网程序源代码基于python实现的甜橙音乐网程序源代码 基于python实现的甜橙音乐网程序源代码 基于python实现的...

冒泡.py 使用python代码实现

冒泡.py 使用python代码实现

冒泡.py 使用python代码实现冒泡.py 使用python代码实现冒泡.py 使用python代码实现冒泡.py 使用python代码实现冒泡.py 使用python代码实现冒泡.py 使用python代码实现冒泡.py 使用python代码实现冒泡.py 使用python...

基于Python实现的桥梁易损性分析源码+代码注释.zip

基于Python实现的桥梁易损性分析源码+代码注释.zip

基于Python实现的桥梁易损性分析源码+代码注释.zip 基于Python实现的桥梁易损性分析源码+代码注释.zip 基于Python实现的桥梁易损性分析源码+代码注释.zip 基于Python实现的桥梁易损性分析源码+代码注释.zip 基于...

基于Python实现stm32 ISP串口烧录hex文件源代码.zip

基于Python实现stm32 ISP串口烧录hex文件源代码.zip

基于Python实现stm32 ISP串口烧录hex文件源代码.zip基于Python实现stm32 ISP串口烧录hex文件源代码.zip基于Python实现stm32 ISP串口烧录hex文件源代码.zip基于Python实现stm32 ISP串口烧录hex文件源代码.zip基于...

希尔排序.py 使用python实现的代码

希尔排序.py 使用python实现的代码

希尔排序.py 使用python实现的代码希尔排序.py 使用python实现的代码希尔排序.py 使用python实现的代码希尔排序.py 使用python实现的代码希尔排序.py 使用python实现的代码希尔排序.py 使用python实现的代码希尔排序...

选择排序.py 使用python实现的代码

选择排序.py 使用python实现的代码

选择排序.py 使用python实现的代码选择排序.py 使用python实现的代码选择排序.py 使用python实现的代码选择排序.py 使用python实现的代码选择排序.py 使用python实现的代码选择排序.py 使用python实现的代码选择排序...

基于python实现的智慧校园考试系统程序源代码.zip

基于python实现的智慧校园考试系统程序源代码.zip

基于python实现的智慧校园考试系统程序源代码 基于python实现的智慧校园考试系统程序源代码 基于python实现的智慧校园考试系统程序源代码 基于python实现的智慧校园考试系统程序源代码 基于python实现的智慧校园考试...

包络分析代码实现python,包络数据分析,Python

包络分析代码实现python,包络数据分析,Python

通过Python实现DEA,不仅可以提高分析效率,还能灵活地调整模型参数,以适应不同的研究需求。 总之,Python提供了强大的工具和库来实现DEA,通过阅读提供的代码和分析数据,我们可以深入理解DEA的实现细节,以及...

基于python的动漫推荐系统的设计与实现代码

基于python的动漫推荐系统的设计与实现代码

基于python的动漫推荐系统的设计与实现代码 基于python的动漫推荐系统的设计与实现代码 基于python的动漫推荐系统的设计与实现代码 基于python的动漫推荐系统的设计与实现代码 基于python的动漫推荐系统的设计与实现...

智能问答系统python实现

智能问答系统python实现

基于知识图谱的智能问答系统python实现(复旦大学论文基于qa语料和知识库的问答系统)_python 智能问答,python 智能问答系统-机器学习代码类资源本代码实现是基于python实现的基于复旦大学崔万云博士的learing ...

归并排序.py 使用python代码实现

归并排序.py 使用python代码实现

归并排序.py 使用python代码实现归并排序.py 使用python代码实现归并排序.py 使用python代码实现归并排序.py 使用python代码实现归并排序.py 使用python代码实现归并排序.py 使用python代码实现归并排序.py 使用...

桶排序.py 使用python代码实现

桶排序.py 使用python代码实现

桶排序.py 使用python代码实现桶排序.py 使用python代码实现桶排序.py 使用python代码实现桶排序.py 使用python代码实现桶排序.py 使用python代码实现桶排序.py 使用python代码实现桶排序.py 使用python代码实现桶...

基于python实现的看图猜成语程序源代码.zip

基于python实现的看图猜成语程序源代码.zip

基于python实现的看图猜成语程序源代码基于python实现的看图猜成语程序源代码基于python实现的看图猜成语程序源代码基于python实现的看图猜成语程序源代码基于python实现的看图猜成语程序源代码基于python实现的看图...

基于python实现的51商城的程序源代码.zip

基于python实现的51商城的程序源代码.zip

基于python实现的51商城的程序源代码 基于python实现的51商城的程序源代码 基于python实现的51商城的程序源代码 基于python实现的51商城的程序源代码 基于python实现的51商城的程序源代码 基于python实现的51商城的...

基于python实现的BBS问答社区程序源代码.zip

基于python实现的BBS问答社区程序源代码.zip

基于python实现的BBS问答社区程序源代码 基于python实现的BBS问答社区程序源代码 基于python实现的BBS问答社区程序源代码 基于python实现的BBS问答社区程序源代码 基于python实现的BBS问答社区程序源代码 基于python...

Python批量出图,实现CAD批量出图,完整代码 Python

Python批量出图,实现CAD批量出图,完整代码 Python

Python批量出图,实现CAD批量出图,完整代码。Python3.0以上可用。

最新推荐最新推荐

recommend-type

答题辅助python代码实现

本题主要涉及的是使用Python编程语言来实现一个答题辅助工具,该工具能够自动识别屏幕上的问题和答案选项。以下是对实现这个功能的关键技术点的详细解释: 1. **屏幕截图**:首先,代码中使用了`screenshot`模块来...
recommend-type

详解python实现交叉验证法与留出法

Python中,我们可以使用`sklearn.model_selection.KFold`实现交叉验证,例如: ```python from sklearn.model_selection import KFold data = pd.read_excel('') # 导入数据 kf = KFold(n_splits=4, shuffle=False,...
recommend-type

Python实现结构体代码实例

下面将详细介绍如何使用Python实现结构体,并通过具体的代码实例进行讲解。 在Python中,类是一种自定义的数据类型,可以包含数据成员(属性)和成员函数(方法)。我们可以创建一个基类(base class),然后定义...
recommend-type

Python实现代码块儿折叠

一种常见的实现代码块折叠的方法是使用`#描述">`和`#</editor-fold>`这对注释。这里的`desc`参数用于提供折叠后的描述文本,例如"折叠后要显示的内容"。下面是一个简单的例子: ```python #折叠后要显示的内容"> ...
recommend-type

python学生信息管理系统实现代码

本文将详细介绍如何使用Python实现一个简单的学生信息管理系统。这个系统能够完成学生信息的创建、查看、查询、删除和修改等基本功能。通过阅读和理解以下内容,你可以了解到如何利用Python的文件操作、JSON序列化...
recommend-type

随机算法详解:概念、分类、性能分析与实例应用

资源摘要信息:"算法设计与分析ch8随机算法" ### 算法设计与分析课程介绍 课程中的第八章专注于随机算法的概念和分析方法。随机算法在计算机科学中占有重要地位,它们在解决各种问题时具有独特的优势。 ### 随机算法的基本概念 随机算法是那些在执行过程中使用概率和统计方法对计算步骤进行随机选择的算法。这类算法的性质通常通过其执行过程中的随机行为来定义。 ### 随机算法的优点 随机算法具有几个显著的优点: 1. 简单性:相比确定性算法,随机算法在设计上往往更为简洁。 2. 时间复杂度低:在许多情况下,随机算法能够在较短的时间内完成计算任务。 3. 具有简短和时间复杂度低的双重优势:随机算法能够在保证较低时间复杂度的同时,算法结构也相对简单。 ### 随机算法的随机性 随机算法的特点是每次执行同一个实例时,结果可能完全不同。算法的效果可能会有很大的差异,这种差异依赖于算法中使用的随机变量。随机算法的正确性和准确性也是随机的。 ### 随机算法的分类 随机算法可以根据其应用和行为特点进行分类: 1. 随机数值算法:主要用于数值问题求解,输出往往是近似解,近似解的精度与算法执行时间成正比。 2. Monte Carlo算法:适用于需要准确解的问题,算法可能给出错误答案,但获得准确解的概率与执行时间成正比。 3. Las Vegas算法:一旦找到解,该解一定是正确的,找到解的概率与执行时间成正比。通过增加对问题的反复求解次数,可以减少求解无效的概率。 ### 分析随机算法的方法 分析随机算法时,需要考虑算法的期望性能以及最坏情况下的性能。这通常涉及到概率论和统计学的知识,以确保算法分析的正确性和准确性。 ### 总结 随机算法为计算机科学提供了一种高效且简洁的问题求解方式。它们在处理具有不确定性的复杂问题时尤为有用,并且能够以较小的时间和资源成本提供有效的解决方案。正确理解和应用随机算法的原理,对于算法设计师和分析员来说至关重要。
recommend-type

Qt实战:用ListWidget和TableWidget快速搞定一个简易文件管理器界面

# Qt实战:用ListWidget和TableWidget构建高效文件管理器界面 在桌面应用开发中,文件管理器是最基础也最考验UI设计能力的组件之一。作为Qt开发者,我们常需要快速实现一个既美观又实用的文件浏览界面。不同于教科书式的控件API讲解,本文将带您从实际项目角度,用**ListWidget**和**TableWidget**这两个核心控件,构建一个支持多视图切换、右键菜单和智能排序的完整解决方案。 ## 1. 界面架构设计与基础布局 我们先从整体框架入手。一个标准的文件管理器通常包含以下元素: - 左侧目录树(本文暂用QListWidget简化实现) - 右侧主视图区域(支持
recommend-type

Spring Boot项目一启动就自动退出,可能是什么原因导致的?

### Spring Boot 应用程序启动并立即停止的原因分析 应用程序启动后立刻关闭通常由多种因素引起。当Spring Boot应用未能保持运行状态,可能是因为入口类缺少必要的配置或存在异常未被捕获处理。 #### 主要原因及解决方案 如果 `main` 方法所在的类没有标注 `@SpringBootApplication` 或者该注解的位置不正确,则可能导致容器无法正常初始化[^1]。确保此注解位于引导类上,并且其包路径能够扫描到其他组件和服务。 另一个常见问题是端口冲突。默认情况下,Spring Boot会尝试监听8080端口;如果有其他服务正在占用这个端口,那么新启动的服务将
recommend-type

PLC控制下的液体混合装置设计与实现

资源摘要信息:"本文旨在设计一种用于液体混合装置的PLC控制系统。PLC(可编程序逻辑控制器)是基于计算机技术的自动控制装置,它通过用户编写的程序来实现控制逻辑的改变。随着电子、计算机和通信技术的进步,PLC已经广泛应用于工业控制领域,尤其是在需要精确控制和监测的搅拌和混合应用中。 该系统主要由几个核心模块组成:CPU模块负责处理逻辑控制和数据运算;输入模块用于接收来自传感器和其他设备的信号;输出模块控制执行器,如电机和阀门;编程装置用于创建和修改控制程序。在液体混合装置中,PLC不仅使搅拌过程自动化,而且还能提高设备运行的稳定性和可靠性。 本文详细描述了液体自动混合系统的方案设计,包括设计原则、系统整体设计要求以及控制方式。方案设计强调了系统对搅拌精度和重复性的要求,同时也要考虑到系统的可扩展性和维护性。 在硬件设计章节中,详细讨论了硬件选型,特别是PLC机型的选择。选择合适的PLC机型对于确保系统的高性能和稳定性至关重要。文中还将探讨如何根据应用需求来选择合适的传感器和其他输入输出设备。 该系统的一个关键特点是其单周期或连续工作的能力,以及断电记忆功能,这意味着即便在电力中断的情况下,系统也能够保留其工作状态,并在电力恢复后继续运行,无需重新启动整个过程。此外,PLC的通信联网功能使得可以远程监控现场设备,这大大提高了工作和管理的便利性。 关键词:PLC,液位传感器,定时器" 知识点详细说明: 1. PLC控制系统概述 - PLC作为通用自动控制装置,其核心为计算机技术。 - PLC的组成:CPU模块、输入模块、输出模块和编程装置。 - PLC在工业混合搅拌设备中的应用,实现搅拌过程自动化,提升工作稳定性。 - PLC的编程可以实现控制功能的改变,适应不同的控制需求。 2. 工业自动控制中的PLC应用 - PLC作为工业控制系统的关键组成部分,正逐渐取代传统继电器控制系统。 - 微处理器和通信技术的发展对PLC性能的提升起到了推动作用。 - PLC的高可靠性和灵活性使其成为工业自动化领域的首选技术。 3. 液体自动混合系统的设计原则和要求 - 设计原则需考虑系统的精确度、可靠性和可维护性。 - 系统整体设计要求包括对搅拌工艺的理解,以及安全性和环境适应性。 - 控制方式系统要求设计应包括控制策略、反馈机制和用户界面。 4. 液体自动混合系统方案的设计思想 - 方案设计应具备灵活性和扩展性,以适应未来可能的工艺变化。 - 系统设计需要平衡成本和性能,确保经济效益。 5. 系统硬件设计 - 硬件选型的重要性,特别是在PLC机型选择方面。 - 输入输出设备的选择,包括传感器、执行器等。 - 需要确保硬件组件的兼容性和整合性,以保证系统的整体性能。 6. PLC程序设计 - 程序设计需根据实际的控制需求和逻辑来编写。 - 断电记忆功能对于保证生产连续性和减少损失至关重要。 - 程序应包含容错机制,以应对可能出现的异常情况。 7. PLC的通信联网功能和远程监控 - PLC可通过通信接口实现与其他系统的数据交换。 - 组态软件的使用提高了监控和管理的便利性。 - 远程监控功能实现了现场设备的实时监控和数据采集。 通过以上知识点,我们可以全面了解液体混合装置的PLC控制系统设计的关键要素和应用范围,以及如何选择合适的技术和组件来构建一个高效、可靠的自动化控制系统。
recommend-type

Parallels Desktop虚拟机USB设备无法识别?这个隐藏设置帮你搞定

# Parallels Desktop虚拟机USB设备无法识别?这个隐藏设置帮你搞定 当你在Mac上使用Parallels Desktop运行Windows虚拟机时,突然发现USB设备无法识别,这种体验确实令人沮丧。无论是外接硬盘、U盘还是其他USB设备,在主机和虚拟机之间无法正常切换使用,会严重影响工作效率。本文将深入分析这一常见问题的根源,并提供一个鲜为人知的解决方案。 ## 1. 理解Parallels Desktop的USB工作机制 Parallels Desktop作为Mac上最流行的虚拟机软件之一,其USB设备管理机制相对复杂但设计精妙。默认情况下,Parallels Des