fix: add instructions to ai context

voiceflow · Nov 14, 2023 · 414d03a · 414d03a
1 parent 2c45395
commit 414d03a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/packages/base-types/src/node/aiResponse.ts b/packages/base-types/src/node/aiResponse.ts
@@ -1,8 +1,8 @@
-import { AIContextParams, AIKnowledgeParams, AIModelParams } from '../utils/ai';
+import { AIKnowledgeContextParams, AIKnowledgeParams, AIModelParams } from '../utils/ai';
 import { NodeType } from './constants';
 import { BaseNode, BaseStep, BaseStepPorts, BuiltInFailPort, BuiltInNextPort, NodeElseID, NodeNextID } from './utils';
 
-export interface StepData extends AIModelParams, AIContextParams, AIKnowledgeParams {
+export interface StepData extends AIModelParams, AIKnowledgeContextParams, AIKnowledgeParams {
   notFoundPath?: boolean;
 
   // the existance of this property is also a flag if the step is a legacy version or not

diff --git a/packages/base-types/src/node/aiSet.ts b/packages/base-types/src/node/aiSet.ts
@@ -1,8 +1,8 @@
-import { AIContextParams, AIKnowledgeParams, AIModelParams } from '../utils/ai';
+import { AIKnowledgeContextParams, AIKnowledgeParams, AIModelParams } from '../utils/ai';
 import { NodeType } from './constants';
 import { BaseNode, BaseStep, NodeNextID } from './utils';
 
-export interface Set extends AIContextParams {
+export interface Set extends AIKnowledgeContextParams {
   variable: string | null;
 }
 

diff --git a/packages/base-types/src/utils/ai.ts b/packages/base-types/src/utils/ai.ts
@@ -34,6 +34,12 @@ export interface AIContextParams {
   prompt: string;
 }
 
+export interface AIKnowledgeContextParams {
+  mode: PROMPT_MODE;
+  prompt: string;
+  instruction?: string;
+}
+
 export interface AIKnowledgeParams {
   // make it optional for backward compatibility
   source?: DATA_SOURCE;