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

Render references next to chat responses by khoj in chat modal

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-03-28 15:35:06 +07:00
parent 1d3d949962
commit 112f388ada
1 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/interface/obsidian/src/chat_modal.ts
View File

@@ -24,8 +24,8 @@ export class KhojChatModal extends Modal {
let chatUrl = `${this.setting.khojUrl}/api/chat?`;
let response = await request(chatUrl);
let chatLogs = JSON.parse(response).response;
chatLogs.forEach( (chatLog: any) => {
this.renderMessage(chatLog.message, chatLog.by);
chatLogs.forEach((chatLog: any) => {
this.renderMessageWithReferences(chatLog.message, chatLog.by, chatLog.context, new Date(chatLog.created));
});
// Add chat input field
@@ -40,7 +40,26 @@ export class KhojChatModal extends Modal {
.onClick(async () => { await this.getChatResponse(this.result) }));
}
renderMessage(message: string, sender: string) {
generateReference(messageEl: any, reference: string, index: number) {
// Generate HTML for Chat Reference
// `<sup><abbr title="${escaped_ref}" tabindex="0">${index}</abbr></sup>`;
let escaped_ref = reference.replace(/"/g, "\\\"")
return messageEl.createEl("sup").createEl("abbr", {
attr: {
title: escaped_ref,
tabindex: "0",
},
text: `[${index}] `,
});
}
renderMessageWithReferences(message: string, sender: string, context?: [string], dt?: Date) {
let messageEl = this.renderMessage(message, sender, dt);
if (context && !!messageEl) {
context.map((reference, index) => this.generateReference(messageEl, reference, index));
}
}
renderMessage(message: string, sender: string, dt?: Date): Element | null {
let { contentEl } = this;
// Append message to conversation history HTML element.
@@ -50,6 +69,8 @@ export class KhojChatModal extends Modal {
let emojified_sender = sender == "khoj" ? "🦅 Khoj" : "🤔 You";
chat_body_el.createDiv({ text: `${emojified_sender}: ${message}` })
}
return chat_body_el
}
async getChatResponse(query: string): Promise<void> {