diff --git a/src/utils/type.ts b/src/utils/type.ts
index fa4510b..f548a03 100644
--- a/src/utils/type.ts
+++ b/src/utils/type.ts
@@ -3,6 +3,7 @@ import type { TASK_TYPE } from "./const"
 export interface PromptMessage {
   id: number
   role: string
+  source?: string
   content: string
   code_html: string | null
   code_css: string | null