package com.trafficaudit.llmintegration.controller;
|
|
import com.trafficaudit.common.Result;
|
import com.trafficaudit.llmintegration.service.AuditExplanationService;
|
import com.trafficaudit.llmintegration.service.LLMService;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@RestController
|
@RequestMapping("/api/llm")
|
public class LLMController {
|
|
@Resource
|
private AuditExplanationService explanationService;
|
@Resource
|
private LLMService llmService;
|
|
@PostMapping("/analyze")
|
public Result<String> analyzeExplanation(@RequestBody Map<String, Object> params) {
|
String explanation = (String) params.get("explanation");
|
String ruleDesc = (String) params.get("ruleDesc");
|
Long reportId = Long.valueOf(params.get("reportId").toString());
|
|
String analysis = explanationService.analyze(explanation, ruleDesc, reportId);
|
return Result.ok(analysis);
|
}
|
|
/** 多轮对话：AI 可调用 query_database 工具查库后回答 */
|
@PostMapping("/chat")
|
public Result<String> chat(@RequestBody Map<String, Object> params) {
|
Object msgsObj = params.get("messages");
|
List<Map<String, Object>> messages = new ArrayList<>();
|
if (msgsObj instanceof List) {
|
for (Object o : (List<?>) msgsObj) {
|
if (o instanceof Map) {
|
messages.add((Map<String, Object>) o);
|
}
|
}
|
}
|
Object ctxObj = params.get("context");
|
Map<String, Object> context = ctxObj instanceof Map ? (Map<String, Object>) ctxObj : new HashMap<>();
|
String reply = llmService.chatWithTools(messages, context);
|
return Result.ok(reply);
|
}
|
}
|
