fix: code generation display issue

Log: Change-Id: I23c64bf94e4a2546fb9c1f1ef381500e677725eb
linuxdeepin · Dec 12, 2023 · 587c01a · 587c01a
1 parent 4a66b66
commit 587c01a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/plugins/codegeex/copilot.cpp b/src/plugins/codegeex/copilot.cpp
@@ -36,7 +36,7 @@ Copilot::Copilot(QObject *parent)
         case CopilotApi::multilingual_code_generate:
             mutexResponse.lock();
             generateResponse = response;
-            if (editorService->showTips && !response.isEmpty()) {
+            if (editorService->showTips && responseValid(response)) {
                 editorService->showTips(generateResponse);
             }
             mutexResponse.unlock();
@@ -76,6 +76,11 @@ QString Copilot::apiKey() const
     return {};
 }
 
+bool Copilot::responseValid(const QString &response)
+{
+    return !(response.isEmpty() || response.startsWith("\n\n\n") || response.startsWith("\n    "));
+}
+
 Copilot *Copilot::instance()
 {
     static Copilot ins;

diff --git a/src/plugins/codegeex/copilot.h b/src/plugins/codegeex/copilot.h
@@ -43,6 +43,7 @@ public slots:
     explicit Copilot(QObject *parent = nullptr);
     QString selectedText() const;
     QString apiKey() const;
+    bool responseValid(const QString &response);
 
     CodeGeeX::CopilotApi copilotApi;
     dpfservice::EditorService *editorService = nullptr;