4.8.10 test (#2433)

* perf: node template ui

* perf: select tool path load error

* pay i18n

* i18n

* perf: oneapi code

* perf: doc

* node templates

* perf: usage table role

* feat: count vector total by datasetId

* perf: user select tip and tempalte market ui

* i18n

* perf: sso config tip

packages/service/common/vectorStore/controller.ts

@@ -20,6 +20,7 @@ export const deleteDatasetDataVector = Vector.delete;
 export const recallFromVectorStore = Vector.embRecall;
 export const getVectorDataByTime = Vector.getVectorDataByTime;
 export const getVectorCountByTeamId = Vector.getVectorCountByTeamId;
+export const getVectorCountByDatasetId = Vector.getVectorCountByDatasetId;
 
 export const insertDatasetDataVector = async ({
   model,

packages/service/common/vectorStore/milvus/class.ts

@@ -297,6 +297,20 @@ export class MilvusCtrl {
 
     return total;
   };
+  getVectorCountByDatasetId = async (teamId: string, datasetId: string) => {
+    const client = await this.getClient();
+
+    const result = await client.query({
+      collection_name: DatasetVectorTableName,
+      output_fields: ['count(*)'],
+      filter: `(teamId == "${String(teamId)}") and (dataset == "${String(datasetId)}")`
+    });
+
+    const total = result.data?.[0]?.['count(*)'] as number;
+
+    return total;
+  };
+
   getVectorDataByTime = async (start: Date, end: Date) => {
     const client = await this.getClient();
     const startTimestamp = new Date(start).getTime();

packages/service/common/vectorStore/pg/class.ts

@@ -205,13 +205,6 @@ export class PgVectorCtrl {
       });
     }
   };
-  getVectorCountByTeamId = async (teamId: string) => {
-    const total = await PgClient.count(DatasetVectorTableName, {
-      where: [['team_id', String(teamId)]]
-    });
-
-    return total;
-  };
   getVectorDataByTime = async (start: Date, end: Date) => {
     const { rows } = await PgClient.query<{
       id: string;
@@ -230,4 +223,18 @@ export class PgVectorCtrl {
       datasetId: item.dataset_id
     }));
   };
+  getVectorCountByTeamId = async (teamId: string) => {
+    const total = await PgClient.count(DatasetVectorTableName, {
+      where: [['team_id', String(teamId)]]
+    });
+
+    return total;
+  };
+  getVectorCountByDatasetId = async (teamId: string, datasetId: string) => {
+    const total = await PgClient.count(DatasetVectorTableName, {
+      where: [['team_id', String(teamId)], 'and', ['dataset_id', String(datasetId)]]
+    });
+
+    return total;
+  };
 }