[improvement][headless]Restructure LLMReq and LLMSchema.

tencentmusic · Sep 12, 2024 · ed78e29 · ed78e29
1 parent 2b3acc8
commit ed78e29
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/LLMRequestService.java b/...chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/LLMRequestService.java
@@ -75,7 +75,7 @@ public LLMReq getLlmReq(ChatQueryContext queryCtx, Long dataSetId) {
         }
 
         llmReq.setCurrentDate(DateUtils.getBeforeDate(0));
-        llmReq.setTerms(getTerms(queryCtx, dataSetId));
+        llmReq.setTerms(getMappedTerms(queryCtx, dataSetId));
         llmReq.setSqlGenType(
                 LLMReq.SqlGenType.valueOf(parserConfig.getParameterValue(PARSER_STRATEGY_TYPE)));
         llmReq.setModelConfig(queryCtx.getModelConfig());
@@ -94,7 +94,7 @@ public LLMResp runText2SQL(LLMReq llmReq) {
         return result;
     }
 
-    protected List<LLMReq.Term> getTerms(ChatQueryContext queryCtx, Long dataSetId) {
+    protected List<LLMReq.Term> getMappedTerms(ChatQueryContext queryCtx, Long dataSetId) {
         List<SchemaElementMatch> matchedElements =
                 queryCtx.getMapInfo().getMatchedElements(dataSetId);
         if (CollectionUtils.isEmpty(matchedElements)) {