Skip to content

Commit d60e1f3

Browse files
authored
fix image url parameter (#1721)
1 parent 0c6f619 commit d60e1f3

File tree

3 files changed

+11
-5
lines changed

3 files changed

+11
-5
lines changed

app/api/chat/anthropic/route.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,14 @@ export async function POST(request: NextRequest) {
3737
return { type: "text", text: content }
3838
} else if (
3939
content?.type === "image_url" &&
40-
content?.image_url?.length
40+
content?.image_url?.url?.length
4141
) {
4242
return {
4343
type: "image",
4444
source: {
4545
type: "base64",
46-
media_type: getMediaTypeFromDataURL(content.image_url),
47-
data: getBase64FromDataURL(content.image_url)
46+
media_type: getMediaTypeFromDataURL(content.image_url.url),
47+
data: getBase64FromDataURL(content.image_url.url)
4848
}
4949
}
5050
} else {

app/api/chat/openai/route.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,11 @@ export async function POST(request: Request) {
2828
model: chatSettings.model as ChatCompletionCreateParamsBase["model"],
2929
messages: messages as ChatCompletionCreateParamsBase["messages"],
3030
temperature: chatSettings.temperature,
31-
max_tokens: chatSettings.model === "gpt-4-vision-preview" ? 4096 : null, // TODO: Fix
31+
max_tokens:
32+
chatSettings.model === "gpt-4-vision-preview" ||
33+
chatSettings.model === "gpt-4o"
34+
? 4096
35+
: null, // TODO: Fix
3236
stream: true
3337
})
3438

lib/build-prompt.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,9 @@ export async function buildFinalMessages(
144144

145145
return {
146146
type: "image_url",
147-
image_url: formedUrl
147+
image_url: {
148+
url: formedUrl
149+
}
148150
}
149151
})
150152
]

0 commit comments

Comments
 (0)