add image to payload

Yuyz0112 · Dec 21, 2023 · 95bd9b8 · 95bd9b8
1 parent 4045b1a
commit 95bd9b8
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/prompts/ui-gen.ts b/prompts/ui-gen.ts
@@ -296,7 +296,21 @@ async function main() {
       },
       {
         role: "user",
-        content: prompt,
+        content: [
+          {
+            type: "text",
+            text: prompt,
+          },
+          ...images.map(
+            (image) =>
+              ({
+                type: "image_url",
+                image_url: {
+                  url: image,
+                },
+              } as const)
+          ),
+        ],
       },
     ],
     "gpt-4-vision-preview"