完成RAG测试

This commit is contained in:
2026-01-27 01:41:45 +08:00
parent 155974572c
commit f78efc7125
16 changed files with 1434 additions and 66 deletions

View File

@@ -93,5 +93,28 @@ class LLMService:
logger.error(f"Rerank Exception: {e}")
# 降级策略
return documents[:top_n]
def chat(self, prompt: str, system_prompt: str = None, model: str = "qwen-turbo") -> str:
"""
[新增] 通用对话生成接口用于RAG的最终回答或作为测试裁判(Judge)
"""
messages = []
if system_prompt:
messages.append({'role': 'system', 'content': system_prompt})
messages.append({'role': 'user', 'content': prompt})
try:
resp = dashscope.Generation.call(
model=model,
messages=messages,
result_format='message'
)
if resp.status_code == HTTPStatus.OK:
return resp.output.choices[0].message.content
else:
logger.error(f"Chat API Error: {resp}")
return "Error generating response."
except Exception as e:
logger.error(f"Chat Exception: {e}")
return "Error generating response."
llm_service = LLMService()