@stoker-platform/types 0.2.4 → 0.2.5

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@stoker-platform/types",
-    "version": "0.2.4",
+    "version": "0.2.5",
     "type": "module",
     "license": "SEE LICENSE IN LICENSE.md",
     "main": "src/main.ts",

package/src/types/schema.ts

@@ -336,6 +336,8 @@ export type PostFileUpdateHookArgs = [
           },
 ]
 
+export type SetEmbeddingHookArgs = [record: StokerRecord]
+
 export type HookArgs =
     | PreOperationHookArgs
     | PreReadHookArgs
@@ -350,6 +352,7 @@ export type HookArgs =
     | PreFileUpdateHookArgs
     | PostFileAddHookArgs
     | PostFileUpdateHookArgs
+    | SetEmbeddingHookArgs
 
 export type PreOperationHook = (...args: PreOperationHookArgs) => boolean | void | Promise<boolean | void>
 export type PreReadHook = (...args: PreReadHookArgs) => void | Promise<void>
@@ -368,6 +371,8 @@ export type PreFileUpdateHook = (...args: PreFileUpdateHookArgs) => boolean | vo
 export type PostFileAddHook = (...args: PostFileAddHookArgs) => boolean | void | Promise<boolean | void>
 export type PostFileUpdateHook = (...args: PostFileUpdateHookArgs) => boolean | void | Promise<boolean | void>
 
+export type SetEmbeddingHook = (...args: SetEmbeddingHookArgs) => string | Promise<string>
+
 export type Hook =
     | PreOperationHook
     | PreReadHook
@@ -382,6 +387,7 @@ export type Hook =
     | PreFileUpdateHook
     | PostFileAddHook
     | PostFileUpdateHook
+    | SetEmbeddingHook
 
 export type Hooks = {
     preOperation?: PreOperationHook
@@ -397,6 +403,7 @@ export type Hooks = {
     preFileUpdate?: PreFileUpdateHook
     postFileAdd?: PostFileAddHook
     postFileUpdate?: PostFileUpdateHook
+    setEmbedding?: SetEmbeddingHook
 }
 
 export interface PreloadCacheRange {
@@ -1005,13 +1012,10 @@ export interface CollectionSchema {
     roleSystemFields?: RoleSystemField[]
     skipRulesValidation?: boolean
     ai?: {
-        embedding?: {
-            inputField: string
-            outputField: string
-            defaultQueryLimit?: number
-        }
+        embedding?: boolean
         chat?: {
             name: string
+            defaultQueryLimit?: number
             roles: StokerRole[]
         }
     }