Skip to content

Commit 3f03cfc

Browse files
committed
feat: 添加数据引擎执行请求和向量查找请求的模式定义,支持原始命令和相似性搜索
1 parent 4c12c8c commit 3f03cfc

1 file changed

Lines changed: 37 additions & 2 deletions

File tree

packages/spec/src/data/data-engine.zod.ts

Lines changed: 37 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,37 @@ export const DataEngineAggregateRequestSchema = z.object({
254254
query: DataEngineAggregateOptionsSchema
255255
});
256256

257+
/**
258+
* Data Engine Execute Request (Raw Command)
259+
* Execute a raw command/query native to the driver (e.g. SQL, Shell, Remote API).
260+
*/
261+
export const DataEngineExecuteRequestSchema = z.object({
262+
method: z.literal('execute'),
263+
/** The abstract command (string SQL, or JSON object) */
264+
command: z.any(),
265+
/** Optional options */
266+
options: z.record(z.any()).optional()
267+
});
268+
269+
/**
270+
* Data Engine Vector Find Request (AI/RAG)
271+
* Perform a similarity search using vector embeddings.
272+
*/
273+
export const DataEngineVectorFindRequestSchema = z.object({
274+
method: z.literal('vectorFind'),
275+
object: z.string(),
276+
/** The vector embedding to search for */
277+
vector: z.array(z.number()),
278+
/** Optional pre-filter (Metadata filtering) */
279+
filter: DataEngineFilterSchema.optional(),
280+
/** Fields to select */
281+
select: z.array(z.string()).optional(),
282+
/** Number of results */
283+
limit: z.number().int().default(5).optional(),
284+
/** Minimum similarity score (0-1) or distance threshold */
285+
threshold: z.number().optional()
286+
});
287+
257288
/**
258289
* Data Engine Batch Request
259290
* Execute multiple operations in a single transaction/request efficiently.
@@ -267,7 +298,9 @@ export const DataEngineBatchRequestSchema = z.object({
267298
DataEngineUpdateRequestSchema,
268299
DataEngineDeleteRequestSchema,
269300
DataEngineCountRequestSchema,
270-
DataEngineAggregateRequestSchema
301+
DataEngineAggregateRequestSchema,
302+
DataEngineExecuteRequestSchema,
303+
DataEngineVectorFindRequestSchema
271304
])),
272305
/**
273306
* Transaction Mode
@@ -289,7 +322,9 @@ export const DataEngineRequestSchema = z.discriminatedUnion('method', [
289322
DataEngineDeleteRequestSchema,
290323
DataEngineCountRequestSchema,
291324
DataEngineAggregateRequestSchema,
292-
DataEngineBatchRequestSchema
325+
DataEngineBatchRequestSchema,
326+
DataEngineExecuteRequestSchema,
327+
DataEngineVectorFindRequestSchema
293328
]).describe('Virtual ObjectQL Request Protocol');
294329

295330
// ==========================================================================

0 commit comments

Comments
 (0)