CTI-RAG/config.yaml at main · itranbong/CTI-RAG · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# ThreatRAG 统一配置文件

# 功能开关
enable_reranker: false
enable_knowledge_base: true
enable_knowledge_graph: true
enable_web_search: true
enable_hybrid_retrieval: true

# 查询配置
use_rewrite_query: "on"  # 是否使用重写查询

# 模型配置
model_provider: "deepseek"
model_name: "deepseek-chat"

# 嵌入模型
embed_model: "local/BAAI/bge-m3"

# Reranker 模型 - 使用相同的bge-m3模型
reranker: "local/BAAI/bge-m3"
model_local_paths:
  local/BAAI/bge-m3: "/app/models/embedding_model/bge-m3"

# 设备配置
device: "cuda"

# FastAPI服务器配置
fastapi_server:
  host: 0.0.0.0  # 修改为0.0.0.0以允许外部访问
  port: 8000

# 向量数据库配置
vector_database:
  path: /app/data/vector_db

# 文件上传配置
file_upload:
  path: /app/data/file_uploads

# 模型路径配置
model:
  embedding_model:
    path: /app/data/embedding_model

# Milvus配置
milvus:
  auto_start: false
  data_dir: ./milvus_lite
  host: milvus-standalone
  port: 19530

# Neo4j配置
neo4j:
  auto_start: false
  data_dir: ./neo4j_data
  host: neo4j
  port: 7688
  http_port: 7474
  # 添加索引器配置
  index_interval: 600  # 索引间隔，单位为秒，默认1小时
  index_batch_size: 200  # 每次处理的实体数量
  auto_index: true  # 是否自动启动索引器