From 0340c24eccce3d90c03934dec14d574128fb32ef Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Wed, 17 Jul 2024 12:06:40 -0400
Subject: added image citation highlights

---
 src/client/views/nodes/ChatBox/AnswerParser.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'src/client/views/nodes/ChatBox/AnswerParser.ts')

diff --git a/src/client/views/nodes/ChatBox/AnswerParser.ts b/src/client/views/nodes/ChatBox/AnswerParser.ts
index f77d2261d..1162d46b0 100644
--- a/src/client/views/nodes/ChatBox/AnswerParser.ts
+++ b/src/client/views/nodes/ChatBox/AnswerParser.ts
@@ -1,4 +1,5 @@
 import { ASSISTANT_ROLE, AssistantMessage, Citation, getChunkType } from './types';
+import { v4 as uuid } from 'uuid';
 
 export class AnswerParser {
     static parse(xml: string): AssistantMessage {
@@ -33,7 +34,8 @@ export class AnswerParser {
                 direct_text: direct_text.trim(),
                 type: getChunkType(type),
                 chunk_id: chunk_id,
-                location: citationPlainStart,
+                text_location: citationPlainStart,
+                citation_id: uuid(),
             });
 
             citationOffset += fullMatch.length;
-- 
cgit v1.2.3-70-g09d2