ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2026-05-02 01:02:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: vector ts (#6166)

Browse Source 
* stop design doc

* remove invalid doc

* rename

* fix: ts

* perf: auto fit

* perf: comment menu

* remove invalid checker

* save button colorschema

* fix: icon

* perf: color schema

* fix: icon

* perf: icon

* perf: icon

* perf: icon

* perf: icon
This commit is contained in:
Archer
2025-12-31 00:54:04 +08:00
committed by GitHub
parent f8bedc1209
commit 005fbf8cdb
285 changed files with 441 additions and 400 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/service/common/vectorDB/oceanbase/index.ts
+6 -16
View File
@@ -2,19 +2,13 @@
import { DatasetVectorTableName } from '../constants';
import { ObClient } from './controller';
import { type RowDataPacket } from 'mysql2/promise';
import {
type DelDatasetVectorCtrlProps,
type EmbeddingRecallCtrlProps,
type EmbeddingRecallResponse,
type InsertVectorControllerProps
} from '../controller.d';
import type { VectorControllerType } from '../type';
import dayjs from 'dayjs';
import { addLog } from '../../system/log';
export class ObVectorCtrl implements VectorControllerType {
constructor() {}
init = async () => {
init: VectorControllerType['init'] = async () => {
try {
await ObClient.query(`
CREATE TABLE IF NOT EXISTS ${DatasetVectorTableName} (
@@ -42,7 +36,7 @@ export class ObVectorCtrl implements VectorControllerType {
}
};
insert = async (props: InsertVectorControllerProps): Promise<{ insertIds: string[] }> => {
insert: VectorControllerType['insert'] = async (props) => {
const { teamId, datasetId, collectionId, vectors } = props;
const values = vectors.map((vector) => [
@@ -64,7 +58,7 @@ export class ObVectorCtrl implements VectorControllerType {
insertIds
};
};
delete = async (props: DelDatasetVectorCtrlProps): Promise<any> => {
delete: VectorControllerType['delete'] = async (props) => {
const { teamId } = props;
const teamIdWhere = `team_id='${String(teamId)}' AND`;
@@ -99,7 +93,7 @@ export class ObVectorCtrl implements VectorControllerType {
where: [where]
});
};
embRecall = async (props: EmbeddingRecallCtrlProps): Promise<EmbeddingRecallResponse> => {
embRecall: VectorControllerType['embRecall'] = async (props) => {
const { teamId, datasetIds, vector, limit, forbidCollectionIdList, filterCollectionIdList } =
props;
@@ -159,7 +153,7 @@ export class ObVectorCtrl implements VectorControllerType {
}))
};
};
getVectorDataByTime = async (start: Date, end: Date) => {
getVectorDataByTime: VectorControllerType['getVectorDataByTime'] = async (start, end) => {
const rows = await ObClient.query<
({
id: string;
@@ -182,11 +176,7 @@ export class ObVectorCtrl implements VectorControllerType {
}));
};
getVectorCount = async (props: {
teamId?: string;
datasetId?: string;
collectionId?: string;
}) => {
getVectorCount: VectorControllerType['getVectorCount'] = async (props) => {
const { teamId, datasetId, collectionId } = props;
// Build where conditions dynamically