feat: add IoD search

src/db/index.ts

@@ -29,7 +29,8 @@ type Message = {
   role: string
   content: string
   images?: string[]
-  sources?: string[]
+  webSources?: string[]
+  iodSources?: string[]
   search?: WebSearch
   createdAt: number
   reasoning_time_taken?: number
@@ -254,7 +255,8 @@ export const saveMessage = async (
   role: string,
   content: string,
   images: string[],
-  source?: any[],
+  webSources?: any[],
+  iodSources?: any[],
   time?: number,
   message_type?: string,
   generationInfo?: any,
@@ -273,7 +275,8 @@ export const saveMessage = async (
     content,
     images,
     createdAt,
-    sources: source,
+    webSources,
+    iodSources,
     messageType: message_type,
     generationInfo: generationInfo,
     reasoning_time_taken
@@ -303,7 +306,8 @@ export const formatToMessage = (messages: MessageHistory): MessageType[] => {
       isBot: message.role === "assistant",
       message: message.content,
       name: message.name,
-      sources: message?.sources || [],
+      webSources: message?.webSources || [],
+      iodSources: message?.iodSources || [],
       images: message.images || [],
       generationInfo: message?.generationInfo,
       reasoning_time_taken: message?.reasoning_time_taken