54 lines
1.4 KiB
TypeScript
54 lines
1.4 KiB
TypeScript
import ky, { HTTPError } from 'ky'
|
|
import OpenAI from 'openai'
|
|
|
|
export const splitUserTextMessageIntoOpenAIBlocks = async (
|
|
input: string
|
|
): Promise<string | OpenAI.Chat.ChatCompletionContentPart[]> => {
|
|
const urlRegex = /(^|\n\n)(https?:\/\/[^\s]+)(\n\n|$)/g
|
|
const match = input.match(urlRegex)
|
|
if (!match) return input
|
|
let parts: OpenAI.Chat.ChatCompletionContentPart[] = []
|
|
let processedInput = input
|
|
|
|
for (const url of match) {
|
|
const textBeforeUrl = processedInput.slice(0, processedInput.indexOf(url))
|
|
if (textBeforeUrl.trim().length > 0) {
|
|
parts.push({ type: 'text', text: textBeforeUrl })
|
|
}
|
|
const cleanUrl = url.trim()
|
|
|
|
try {
|
|
const response = await ky.get(cleanUrl)
|
|
if (
|
|
!response.ok ||
|
|
!response.headers.get('content-type')?.startsWith('image/')
|
|
) {
|
|
parts.push({ type: 'text', text: cleanUrl })
|
|
} else {
|
|
parts.push({
|
|
type: 'image_url',
|
|
image_url: {
|
|
url: url.trim(),
|
|
detail: 'auto',
|
|
},
|
|
})
|
|
}
|
|
} catch (err) {
|
|
if (err instanceof HTTPError) {
|
|
console.log(err.response.status, await err.response.text())
|
|
} else {
|
|
console.error(err)
|
|
}
|
|
}
|
|
processedInput = processedInput.slice(
|
|
processedInput.indexOf(url) + url.length
|
|
)
|
|
}
|
|
|
|
if (processedInput.trim().length > 0) {
|
|
parts.push({ type: 'text', text: processedInput })
|
|
}
|
|
|
|
return parts
|
|
}
|