feat: kb type models add kbtags (NLU-864)

voiceflow · Sep 27, 2023 · 149b7ef · 149b7ef
1 parent 3ce6585
commit 149b7ef
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/base-types/src/models/project/knowledgeBase.ts b/packages/base-types/src/models/project/knowledgeBase.ts
@@ -40,6 +40,11 @@ export enum KnowledgeBaseDocumentStatus {
   INITIALIZED = 'INITIALIZED',
 }
 
+export interface KBTag {
+  tagID: string;
+  label: string;
+}
+
 export interface KnowledgeBaseDocument {
   data: KnowledgeBasePDF | KnowledgeBaseText | KnowledgeBaseURL | KnowledgeBaseDocx;
   status: {
@@ -51,6 +56,7 @@ export interface KnowledgeBaseDocument {
   documentID: string;
   s3ObjectRef: string;
   version?: number;
+  tags?: Record<string, KBTag>;
 }
 
 export enum ChunkStrategyType {
@@ -75,4 +81,5 @@ export interface KnowledgeBaseSettings {
 export interface KnowledgeBase {
   settings?: KnowledgeBaseSettings;
   documents: Record<string, KnowledgeBaseDocument>;
+  tags?: Record<string, KBTag>;
 }