fix: ai capture

voiceflow · Dec 19, 2023 · 5eef908 · 5eef908
1 parent bfa70e5
commit 5eef908
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 0 deletions.
diff --git a/packages/base-types/src/node/aiCapture.ts b/packages/base-types/src/node/aiCapture.ts
@@ -0,0 +1,44 @@
+import { AIModelParams } from '../utils/ai';
+import { NodeType } from './constants';
+import {
+  BaseNode,
+  BaseNoMatchNodeData,
+  BaseNoMatchStepData,
+  BaseNoReplyNodeData,
+  BaseNoReplyStepData,
+  BaseStep,
+  BaseStepPorts,
+  BuiltInFailPort,
+  BuiltInNextPort,
+  BuiltInNoMatchNoReplyPorts,
+  NodeElseID,
+  NodeIntentScope,
+  NodeNextID,
+  StepIntentScope,
+} from './utils';
+
+export interface StepData extends AIModelParams {}
+
+export interface Step extends BaseStep<StepData, BaseStepPorts<BuiltInNextPort & BuiltInFailPort>> {
+  type: NodeType.AI_CAPTURE;
+
+  rules: string[];
+  entities: string[];
+  exitScenerios: string[];
+}
+
+export interface Node extends BaseNode, StepData, NodeNextID, NodeElseID {
+  type: NodeType.AI_CAPTURE;
+}
+
+export interface BaseCaptureData extends BaseNoReplyStepData, StepIntentScope, BaseNoMatchStepData {}
+
+export interface StepData extends AIModelParams {}
+
+export interface StepBuiltInPorts extends BuiltInNextPort, BuiltInNoMatchNoReplyPorts {}
+
+export interface StepPorts extends BaseStepPorts<StepBuiltInPorts, []> {}
+
+export interface Node extends BaseNode, StepData, NodeNextID, NodeIntentScope, BaseNoReplyNodeData, BaseNoMatchNodeData {
+  type: NodeType.AI_CAPTURE;
+}
diff --git a/packages/base-types/src/node/constants.ts b/packages/base-types/src/node/constants.ts
@@ -9,6 +9,7 @@ export enum NodeType {
   INTERACTION = 'interaction',
 
   // ai
+  AI_CAPTURE = 'ai_capture',
   AI_RESPONSE = 'generative',
   AI_SET = 'ai_set',
 

diff --git a/packages/base-types/src/node/index.ts b/packages/base-types/src/node/index.ts
@@ -1,4 +1,5 @@
 import * as _v1 from './_v1';
+import * as AICapture from './aiCapture';
 import * as AIResponse from './aiResponse';
 import * as AISet from './aiSet';
 import * as Api from './api';
@@ -28,6 +29,7 @@ import { BaseNode, NodeNextID } from './utils';
 import * as Zapier from './zapier';
 
 export * as _v1 from './_v1';
+export * as AICapture from './aiCapture';
 export * as AIResponse from './aiResponse';
 export * as AISet from './aiSet';
 export * as Api from './api';
@@ -107,6 +109,7 @@ export type AnyCommonStep =
   | CustomBlockPointer.Step
   | AIResponse.Step
   | AISet.Step
+  | AICapture.Step
   | _v1.Step;
 
 /**
@@ -136,6 +139,7 @@ export type AnyCommonNode =
   | ChannelAction.Node
   | AIResponse.Node
   | AISet.Node
+  | AICapture.Node
   | _v1.Node;
 
 export type AnyCommonCommand = Intent.Command | Command.Command;