coder
ab6469b9dd
新增配置: - USE_LLM_INDEX = False (默认关闭LLM增强索引) 简单索引模式: - 不调用LLM，速度快 - 使用词频统计提取关键词 - 适合快速建立索引 LLM增强模式(USE_LLM_INDEX=True): - 使用LLM分析文档，提取摘要、关键词、实体 - 索引质量更高，但速度慢、需要LLM服务
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
"""
|
||
LLM Index RAG 配置文件
|
||
基于索引和搜索的RAG系统（不使用向量模型）
|
||
"""
|
||
|
||
# ==================== 应用配置 ====================
|
||
APP_NAME = "LLM Index RAG"
|
||
APP_VERSION = "1.0.0"
|
||
SECRET_KEY = "llm-index-rag-secret-key"
|
||
|
||
# ==================== LLM配置 ====================
|
||
LLM_CONFIG = {
|
||
"api_base": "http://192.168.2.5:1234/v1",
|
||
"api_key": "sk-lm-fuP5tGU8:Hi7YU87jHyDP6Ay8Tl2j",
|
||
"model": "qwen/qwen3.5-35b-a3b",
|
||
"max_tokens": 4000,
|
||
"temperature": 0.3,
|
||
"timeout": 120,
|
||
}
|
||
|
||
# ==================== 文档配置 ====================
|
||
DOCUMENT_DIR = "documents" # 文档存储目录
|
||
INDEX_DIR = "indexes" # 索引存储目录
|
||
LOG_DIR = "logs" # 日志目录
|
||
|
||
SUPPORTED_FORMATS = ['.txt', '.md', '.pdf', '.docx', '.html', '.json']
|
||
|
||
# 文档处理配置
|
||
DOC_CONFIG = {
|
||
"chunk_size": 2000, # 文档分块大小
|
||
"chunk_overlap": 200, # 分块重叠
|
||
"max_keywords": 20, # 每个文档块最大关键词数
|
||
"max_summary_length": 500, # 摘要最大长度
|
||
}
|
||
|
||
# ==================== 索引模式 ====================
|
||
# use_llm_index: 是否使用LLM增强索引
|
||
# True - 使用LLM分析文档，提取关键词、摘要等（需要LLM服务，速度慢）
|
||
# False - 使用简单分词和词频统计（速度快，无需LLM）
|
||
USE_LLM_INDEX = False # 默认关闭LLM索引
|
||
|
||
# ==================== 索引配置 ====================
|
||
INDEX_CONFIG = {
|
||
# BM25参数
|
||
"bm25_k1": 1.5, # 词频饱和参数
|
||
"bm25_b": 0.75, # 文档长度归一化参数
|
||
|
||
# 检索配置
|
||
"max_results": 20, # 最大返回结果数
|
||
"min_score": 0.1, # 最低相关性分数
|
||
|
||
# 关键词权重
|
||
"title_weight": 3.0, # 标题关键词权重
|
||
"keyword_weight": 2.0, # 显式关键词权重
|
||
"content_weight": 1.0, # 内容关键词权重
|
||
"summary_weight": 1.5, # 摘要关键词权重
|
||
}
|
||
|
||
# ==================== 查询配置 ====================
|
||
QUERY_CONFIG = {
|
||
"max_expansion_terms": 5, # 查询扩展最大词数
|
||
"use_query_expansion": True, # 是否启用查询扩展
|
||
"use_rerank": True, # 是否使用重排
|
||
}
|
||
|
||
# ==================== 数据库配置 ====================
|
||
DATABASE_URL = "sqlite:///llm_index_rag.db"
|
||
|
||
# ==================== API配置 ====================
|
||
API_HOST = "0.0.0.0"
|
||
API_PORT = 19001 |