Skip to content

Commit 525a46d

Browse files
committed
Add basic referenced sources support
1 parent 9185fa0 commit 525a46d

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

src/app/LightspeedChatbot/hooks/useChatbot.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -257,6 +257,8 @@ export const useChatbot = () => {
257257
setFileError(undefined);
258258
};
259259

260+
261+
260262
const handleSend = async (message: string | number) => {
261263
setIsSendButtonDisabled(true);
262264
const messageContent = String(message);
@@ -393,6 +395,12 @@ export const useChatbot = () => {
393395
share: { onClick: () => {} },
394396
listen: { onClick: () => {} },
395397
},
398+
sources: endData.referenced_documents && endData.referenced_documents.length > 0 ? {
399+
sources: endData.referenced_documents.map(doc => ({
400+
title: doc.doc_title,
401+
link: doc.doc_url,
402+
})),
403+
} : undefined,
396404
};
397405
}
398406
return updatedMessages;

src/app/LightspeedChatbot/types.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,10 @@ export interface StreamTokenData {
5757
}
5858

5959
export interface StreamEndData {
60-
referenced_documents: any[];
60+
referenced_documents: Array<{
61+
doc_url: string;
62+
doc_title: string;
63+
}>;
6164
truncated: any;
6265
input_tokens: number;
6366
output_tokens: number;

0 commit comments

Comments
 (0)