[improvement][Headless] Unit tests do not use the LLM parsing tool by default. (#699)

This commit is contained in:
lexluo09
2024-01-30 17:40:28 +08:00
committed by GitHub
parent 954c67c947
commit 26aefceb04
3 changed files with 36 additions and 23 deletions

View File

@@ -100,6 +100,9 @@ public class LLMRequestService {
public NL2SQLTool getParserTool(QueryContext queryCtx, Set<Long> modelIdSet) {
Agent agent = queryCtx.getAgent();
if (Objects.isNull(agent)) {
return null;
}
List<NL2SQLTool> commonAgentTools = agent.getParserTools(AgentToolType.NL2SQL_LLM);
Optional<NL2SQLTool> llmParserTool = commonAgentTools.stream()
.filter(tool -> {