马一丁

Allows changing GRAPHRAG_ENABLED and GRAPHRAG_MAX_QUERIES on the front end

@@ -76,6 +76,10 @@ class Settings(BaseSettings): @@ -76,6 +76,10 @@ class Settings(BaseSettings):
76 KEYWORD_OPTIMIZER_BASE_URL: Optional[str] = Field(None, description="Keyword Optimizer BaseUrl,可按所选服务配置") 76 KEYWORD_OPTIMIZER_BASE_URL: Optional[str] = Field(None, description="Keyword Optimizer BaseUrl,可按所选服务配置")
77 KEYWORD_OPTIMIZER_MODEL_NAME: Optional[str] = Field(None, description="Keyword Optimizer LLM 模型名称,例如 qwen-plus") 77 KEYWORD_OPTIMIZER_MODEL_NAME: Optional[str] = Field(None, description="Keyword Optimizer LLM 模型名称,例如 qwen-plus")
78 78
  79 + # ================== GraphRAG 配置 ====================
  80 + GRAPHRAG_ENABLED: bool = Field(False, description="是否启用GraphRAG知识图谱功能(true/false)")
  81 + GRAPHRAG_MAX_QUERIES: int = Field(3, description="GraphRAG每个章节生成前的最大查询次数")
  82 +
79 # ================== 网络工具配置 ==================== 83 # ================== 网络工具配置 ====================
80 # Tavily API(申请地址:https://www.tavily.com/) 84 # Tavily API(申请地址:https://www.tavily.com/)
81 TAVILY_API_KEY: Optional[str] = Field(None, description="Tavily API(申请地址:https://www.tavily.com/)API密钥,用于Tavily网络搜索") 85 TAVILY_API_KEY: Optional[str] = Field(None, description="Tavily API(申请地址:https://www.tavily.com/)API密钥,用于Tavily网络搜索")
@@ -2120,6 +2120,14 @@ @@ -2120,6 +2120,14 @@
2120 { key: 'BOCHA_WEB_SEARCH_API_KEY', label: 'Bocha API Key', type: 'password', condition: { key: 'SEARCH_TOOL_TYPE', value: 'BochaAPI' } }, 2120 { key: 'BOCHA_WEB_SEARCH_API_KEY', label: 'Bocha API Key', type: 'password', condition: { key: 'SEARCH_TOOL_TYPE', value: 'BochaAPI' } },
2121 { key: 'ANSPIRE_API_KEY', label: 'Anspire API Key', type: 'password', condition: { key: 'SEARCH_TOOL_TYPE', value: 'AnspireAPI' } } 2121 { key: 'ANSPIRE_API_KEY', label: 'Anspire API Key', type: 'password', condition: { key: 'SEARCH_TOOL_TYPE', value: 'AnspireAPI' } }
2122 ] 2122 ]
  2123 + },
  2124 + {
  2125 + title: 'GraphRAG 图谱增强',
  2126 + subtitle: '章节生成前的知识图谱查询,默认关闭',
  2127 + fields: [
  2128 + { key: 'GRAPHRAG_ENABLED', label: '是否启用', type: 'select', options: ['False', 'True'] },
  2129 + { key: 'GRAPHRAG_MAX_QUERIES', label: '每章节查询次数上限', type: 'number', condition: { key: 'GRAPHRAG_ENABLED', value: 'True' } }
  2130 + ]
2123 } 2131 }
2124 ]; 2132 ];
2125 2133