Merge pull request #20 from TencentCloudBase/feat--support-db-query

feat: 支持数据模型检索

Author: Mrjing

apps/miniprogram-agent-ui/miniprogram/components/agent-ui/index.js

@@ -762,6 +762,7 @@ Component({
               knowledge_meta,
               knowledge_base,
               finish_reason,
+              search_results
             } = dataJson;
             const newValue = [...this.data.chatRecords];
             // 取最后一条消息更新
@@ -831,6 +832,14 @@ Component({
                 chatStatus: 2,
               });
             }
+            // 数据库，只更新一次
+            if(type === "db" && !lastValue.db_len) {
+              lastValue.db_len = search_results.relateTables || 0
+              this.setData({
+                [`chatRecords[${lastValueIndex}].db_len`]: lastValue.db_len,
+                chatStatus: 2,
+              })
+            }
           } catch (e) {
             // console.log('err', event, e)
             break;

apps/miniprogram-agent-ui/miniprogram/components/agent-ui/index.wxml

@@ -30,6 +30,10 @@
             </view>
           </block>
           <block wx:else>
+            <!-- 数据库检索 -->
+            <view wx:if="{{item.db_len}}" style="border-radius: 8px;margin-bottom: 12px;background-color: #f5f5f5;padding: 18rpx 26rpx;display: inline-block;opacity: 0.7;font-size: 14px;">
+              已匹配 {{item.db_len}} 张数据表
+            </view>
             <!-- 联网搜索 -->
             <FoldedCard wx:if="{{item.search_info}}" initStatus="{{false}}" showBgColor="{{true}}">
               <view slot="title" style="opacity: 0.7;font-size: 14px;display: flex; align-items: center; gap: 8px;">