From b434c39182fddfcd508a708587565d57ff1c2514 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Wed, 23 Jul 2025 12:55:02 +0800 Subject: [PATCH] =?UTF-8?q?feat(lightrag):=20=E6=B7=BB=E5=8A=A0=E7=8E=AF?= =?UTF-8?q?=E5=A2=83=E5=8F=98=E9=87=8F=E9=85=8D=E7=BD=AE=E4=BB=A5=E6=94=AF?= =?UTF-8?q?=E6=8C=81=E5=8A=A8=E6=80=81=E9=80=89=E6=8B=A9=20LLM=20=E6=8F=90?= =?UTF-8?q?=E4=BE=9B=E8=80=85=E5=92=8C=E5=90=8D=E7=A7=B0=EF=BC=8C=E4=BC=98?= =?UTF-8?q?=E5=8C=96=20LLM=20=E5=87=BD=E6=95=B0=E8=8E=B7=E5=8F=96=E9=80=BB?= =?UTF-8?q?=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/core/lightrag_kb.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/core/lightrag_kb.py b/src/core/lightrag_kb.py index 0a4b563d..e8efcdfd 100644 --- a/src/core/lightrag_kb.py +++ b/src/core/lightrag_kb.py @@ -15,6 +15,10 @@ from src import config from src.utils import logger, hashstr, get_docker_safe_url +LIGHTRAG_LLM_PROVIDER = os.getenv("LIGHTRAG_LLM_PROVIDER", "openai") +LIGHTRAG_LLM_NAME = os.getenv("LIGHTRAG_LLM_NAME", "gpt-4.1-mini") + + class LightRagKB(KnowledgeBase): """基于 LightRAG 的知识库实现""" @@ -105,7 +109,7 @@ class LightRagKB(KnowledgeBase): def _get_llm_func(self, llm_info: Dict): """获取 LLM 函数""" from src.models import select_model - model = select_model("dashscope", "qwen-max-latest") + model = select_model(LIGHTRAG_LLM_PROVIDER, LIGHTRAG_LLM_NAME) async def llm_model_func(prompt, system_prompt=None, history_messages=[], **kwargs): return await openai_complete_if_cache(