topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 02:46:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(chat)多轮对话增加rewrittenQuery的MapInfo (#1599)

Browse Source
This commit is contained in:
yudong
2024-08-26 10:20:52 +08:00
committed by GitHub
parent 3d7fac7ca9
commit 3915d90eec
3 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
benchmark/benchmark.py
View File

@@ -38,9 +38,10 @@ class BatchTest:
response = requests.post(url, headers=headers, data=json.dumps(data)) response = requests.post(url, headers=headers, data=json.dumps(data))
return response.json() return response.json()
def execute(self, query_text, queryId): def execute(self, agentId, query_text, queryId):
url = self.base_url + 'execute' url = self.base_url + 'execute'
data = { data = {
'agentId': agentId,
'queryText': query_text, 'queryText': query_text,
'parseId': 1, 'parseId': 1,
'chatId': self.chatId, 'chatId': self.chatId,
@@ -75,7 +76,7 @@ def benchmark(url:str, agentId:str, chatId:str, filePath:str, userName:str):
# 捕获异常，防止程序中断 # 捕获异常，防止程序中断
try: try:
parse_resp = batch_test.parse(question) parse_resp = batch_test.parse(question)
batch_test.execute(question, parse_resp['data']['queryId']) batch_test.execute(agentId, question, parse_resp['data']['queryId'])
except Exception as e: except Exception as e:
print('error:', e) print('error:', e)
traceback.print_exc() traceback.print_exc()

2
benchmark/requirements.txt
View File

@@ -1,4 +1,4 @@
pandas==2.0.3 pandas==2.2.2
PyJWT==2.8.0 PyJWT==2.8.0
requests==2.28.2 requests==2.28.2

4
chat/server/src/main/java/com/tencent/supersonic/chat/server/parser/NL2SQLParser.java
View File

@@ -203,8 +203,10 @@ public class NL2SQLParser implements ChatQueryParser {
Response<AiMessage> response = chatLanguageModel.generate(prompt.toUserMessage()); Response<AiMessage> response = chatLanguageModel.generate(prompt.toUserMessage());
String rewrittenQuery = response.content().text(); String rewrittenQuery = response.content().text();
keyPipelineLog.info("NL2SQLParser modelResp:{}", rewrittenQuery); keyPipelineLog.info("NL2SQLParser modelResp:{}", rewrittenQuery);
parseContext.setQueryText(rewrittenQuery); parseContext.setQueryText(rewrittenQuery);
QueryNLReq rewrittenQueryNLReq = QueryReqConverter.buildText2SqlQueryReq(parseContext);
MapResp rewrittenQueryMapResult = chatLayerService.performMapping(rewrittenQueryNLReq);
parseContext.setMapInfo(rewrittenQueryMapResult.getMapInfo());
log.info("Last Query: {} Current Query: {}, Rewritten Query: {}", log.info("Last Query: {} Current Query: {}, Rewritten Query: {}",
lastQuery.getQueryText(), currentMapResult.getQueryText(), rewrittenQuery); lastQuery.getQueryText(), currentMapResult.getQueryText(), rewrittenQuery);
} }