From f7d2053b9d89e3d6ddaa55660aaf91ceda15e361 Mon Sep 17 00:00:00 2001
From: Henry He <henryhyn@163.com>
Date: Mon, 10 Jul 2023 18:47:18 +0800
Subject: [PATCH] fix: stop chars

---
 .../java/org/enthusa/askdata/controller/BiVenusController.java   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/main/java/org/enthusa/askdata/controller/BiVenusController.java b/src/main/java/org/enthusa/askdata/controller/BiVenusController.java
index 3293a3c..628745b 100644
--- a/src/main/java/org/enthusa/askdata/controller/BiVenusController.java
+++ b/src/main/java/org/enthusa/askdata/controller/BiVenusController.java
@@ -135,6 +135,7 @@ public class BiVenusController {
         completion.addUserMsg(userMsg);
         completion.setApiKey(System.getenv("INSCODE_API_KEY"));
         String reply = gptClient.chatCompletion(completion);
+        reply = StringUtils.split(reply, ';')[0];
         String sql = String.format("SELECT %s", reply);
         String res = sqlFormatFromDsAndSql(dsId, sql);
         long end = System.currentTimeMillis();
-- 
GitLab