fix dataset mapping

app/api/chat/route.ts

@@ -8,6 +8,7 @@ import {
 	ChatCompletionContentPartImage,
 	ChatCompletionMessageParam,
 } from 'openai/resources';
+import { DatasetImageEntity } from '../../../lib/types';
 
 export const runtime = 'edge';
 
@@ -19,7 +20,7 @@ export async function POST(req: Request) {
 	const json = await req.json();
 	const { messages, dataset } = json as {
 		messages: ChatCompletionMessageParam[];
-		dataset: string[];
+		dataset: DatasetImageEntity[];
 	};
 
 	const userId = (await auth())?.user.id;
@@ -39,10 +40,10 @@ export async function POST(req: Request) {
 					text: message.content as string,
 				},
 				...dataset.map(
-					image =>
+					entity =>
 						({
 							type: 'image_url',
-							image_url: { url: image },
+							image_url: { url: entity.url },
 						}) satisfies ChatCompletionContentPartImage,
 				),
 			];

lib/hooks/useImageUpload.ts

@@ -1,7 +1,8 @@
 import { useAtom } from 'jotai';
 import { DropzoneOptions, useDropzone } from 'react-dropzone';
-import { DatasetImageEntity, datasetAtom } from '../../state';
+import { datasetAtom } from '../../state';
 import { toast } from 'react-hot-toast';
+import { DatasetImageEntity } from '../types';
 
 const useImageUpload = (options?: Partial<DropzoneOptions>) => {
 	const [, setTarget] = useAtom(datasetAtom);

lib/types.ts

@@ -1,18 +1,24 @@
-import { type Message } from 'ai'
+import { type Message } from 'ai';
 
 export interface Chat extends Record<string, any> {
-  id: string
-  title: string
-  createdAt: Date
-  userId: string
-  path: string
-  messages: Message[]
-  sharePath?: string
+	id: string;
+	title: string;
+	createdAt: Date;
+	userId: string;
+	path: string;
+	messages: Message[];
+	sharePath?: string;
 }
 
 export type ServerActionResult<Result> = Promise<
-  | Result
-  | {
-      error: string
-    }
->
+	| Result
+	| {
+			error: string;
+	  }
+>;
+
+export type DatasetImageEntity = {
+	url: string;
+	selected: boolean;
+	name: string;
+};

state/index.ts

@@ -1,10 +1,6 @@
 import { atom } from 'jotai';
+import { DatasetImageEntity } from '../lib/types';
 
-export type DatasetImageEntity = {
-	url: string;
-	selected: boolean;
-	name: string;
-};
 // list of image urls or base64 strings
 export const datasetAtom = atom<DatasetImageEntity[]>([]);
 // export const selectedImagesAtom = atom<number[]>([]);