klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-03 05:29:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Render inline any image files output by code in message

Browse Source 
Update regex to also include any links to code generated images that
aren't explicitly meant to be displayed inline. This allows folks to
download the image (unlike the fake link that doesn't work created by
model)
This commit is contained in:
Debanjum Singh Solanky
2024-10-11 01:11:34 -07:00
parent 20d495c43a
commit 9daaae0fdb
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/interface/web/app/common/chatFunctions.ts
View File

@@ -116,7 +116,7 @@ export function processMessageChunk(
if (context) currentMessage.context = context;
// Replace file links with base64 data
currentMessage.rawResponse = replaceFileLinksWithBase64(
currentMessage.rawResponse = renderCodeGenImageInline(
currentMessage.rawResponse,
codeContext,
);
@@ -186,12 +186,12 @@ export function handleImageResponse(imageJson: any, liveStream: boolean): Respon
return reference;
}
export function replaceFileLinksWithBase64(message: string, codeContext: CodeContext) {
export function renderCodeGenImageInline(message: string, codeContext: CodeContext) {
if (!codeContext) return message;
Object.values(codeContext).forEach((contextData) => {
contextData.results.output_files?.forEach((file) => {
const regex = new RegExp(`!\\[.*?\\]\\(.*${file.filename}\\)`, "g");
const regex = new RegExp(`!?\\[.*?\\]\\(.*${file.filename}\\)`, "g");
if (file.filename.match(/\.(png|jpg|jpeg|gif|webp)$/i)) {
const replacement = `![${file.filename}](data:image/${file.filename.split(".").pop()};base64,${file.b64_data})`;
message = message.replace(regex, replacement);