diff --git a/packages/base-types/src/models/project/knowledgeBase.ts b/packages/base-types/src/models/project/knowledgeBase.ts
index e0688a17d..79b7dff7f 100644
--- a/packages/base-types/src/models/project/knowledgeBase.ts
+++ b/packages/base-types/src/models/project/knowledgeBase.ts
@@ -1,4 +1,4 @@
-import { AIContextParams, AIModelParams } from '@base-types/utils/ai';
+import { AIKnowledgeContextParams, AIModelParams } from '@base-types/utils/ai';
 
 export enum KnowledgeBaseDocumentType {
   PDF = 'pdf',
@@ -98,7 +98,7 @@ export interface RecursiveTextSplitter {
 
 export interface KnowledgeBaseSettings {
   chunkStrategy: RecursiveTextSplitter;
-  summarization: AIModelParams & AIContextParams;
+  summarization: AIModelParams & AIKnowledgeContextParams;
   search: {
     limit: number;
     metric: string;