fix: predict

backend/apps/chat/task/llm.py

@@ -846,6 +846,8 @@ def run_analysis_or_predict_task(llm_service: LLMService, action_type: str, base
     try:
         llm_service.set_record(save_analysis_predict_record(llm_service.session, base_record, action_type))
 
+        yield orjson.dumps({'type': 'id', 'id': llm_service.get_record().id}).decode() + '\n\n'
+
         if action_type == 'analysis':
             # generate analysis
             analysis_res = llm_service.generate_analysis()

backend/template.yaml

@@ -129,7 +129,7 @@ template:
   predict:
     system: |
       ### 说明：
-      你是一个数据分析师，你的任务是根据给定的数据进行数据预测，我将以json格式给你一组数据，你帮我预测之后的数据（一段可以展示趋势的数据，至少2个周期），用json格式返回。
+      你是一个数据分析师，你的任务是根据给定的数据进行数据预测，我将以json格式给你一组数据，你帮我预测之后的数据（一段可以展示趋势的数据，至少2个周期），用json格式返回，返回的格式需要与传入的数据格式保持一致。
       ```json
       
       无法预测或者不支持预测的数据请直接返回(不需要返回JSON格式)："抱歉，该数据无法进行预测。(有原因则返回无法预测的原因)"

frontend/src/views/chat/RecommendQuestion.vue

@@ -68,7 +68,7 @@ function clickQuestion(question: string): void {
   .question-grid {
     display: grid;
     grid-gap: 12px;
-    grid-template-columns: repeat(2, 50%);
+    grid-template-columns: repeat(2, calc(50% - 6px));
   }
 
   .question {

frontend/src/views/chat/chat-block/UserChat.vue

@@ -45,6 +45,7 @@ function clickPredict() {
 
   .prefix-title {
     color: rgba(28, 186, 144, 1);
+    white-space: nowrap;
   }
 }
 </style>

frontend/src/views/chat/index.vue

@@ -116,7 +116,7 @@
                     @finish="onChartAnswerFinish"
                     @error="onChartAnswerError"
                   >
-                    <ChartBlock :message="message" />
+                    <ChartBlock style="margin-top: 12px" :message="message" />
                     <div
                       v-if="message.record?.error && message.record?.error?.trim().length > 0"
                       class="error-container"
@@ -220,7 +220,7 @@
                     @finish="onPredictAnswerFinish"
                     @error="onPredictAnswerError"
                   >
-                    <ChartBlock :message="message" is-predict />
+                    <ChartBlock style="margin-top: 12px" :message="message" is-predict />
                     <div
                       v-if="message.record?.error && message.record?.error?.trim().length > 0"
                       class="error-container"