feat(lightrag): 添加环境变量配置以支持动态选择 LLM 提供者和名称,优化 LLM 函数获取逻辑

This commit is contained in:
Wenjie Zhang 2025-07-23 12:55:02 +08:00
parent 346b85836b
commit b434c39182

View File

@ -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(