Enhancement: chat message document links (#12670)

src-ui/src/app/services/chat.service.spec.ts

@@ -9,7 +9,11 @@ import {
 } from '@angular/common/http/testing'
 import { TestBed } from '@angular/core/testing'
 import { environment } from 'src/environments/environment'
-import { ChatService } from './chat.service'
+import {
+  CHAT_METADATA_DELIMITER,
+  ChatService,
+  parseChatResponse,
+} from './chat.service'
 
 describe('ChatService', () => {
   let service: ChatService
@@ -55,4 +59,22 @@ describe('ChatService', () => {
       partialText: mockResponse,
     } as any)
   })
+
+  it('should parse chat references from the metadata trailer', () => {
+    const parsed = parseChatResponse(
+      `Answer text${CHAT_METADATA_DELIMITER}{"references":[{"id":1,"title":"Document 1"}]}`
+    )
+
+    expect(parsed.content).toBe('Answer text')
+    expect(parsed.references).toEqual([{ id: 1, title: 'Document 1' }])
+  })
+
+  it('should hide incomplete metadata trailer from the visible content', () => {
+    const parsed = parseChatResponse(
+      `Answer text${CHAT_METADATA_DELIMITER}{"references"`
+    )
+
+    expect(parsed.content).toBe('Answer text')
+    expect(parsed.references).toBeUndefined()
+  })
 })

src-ui/src/app/services/chat.service.ts

@@ -11,6 +11,46 @@ export interface ChatMessage {
   role: 'user' | 'assistant'
   content: string
   isStreaming?: boolean
+  references?: ChatReference[]
+}
+
+export interface ChatReference {
+  id: number
+  title: string
+}
+
+export interface ParsedChatResponse {
+  content: string
+  references?: ChatReference[]
+}
+
+export const CHAT_METADATA_DELIMITER = '\n\n__PAPERLESS_CHAT_METADATA__'
+
+export function parseChatResponse(response: string): ParsedChatResponse {
+  const delimiterIndex = response.indexOf(CHAT_METADATA_DELIMITER)
+
+  if (delimiterIndex === -1) {
+    return { content: response }
+  }
+
+  const metadataString = response.slice(
+    delimiterIndex + CHAT_METADATA_DELIMITER.length
+  )
+
+  try {
+    const metadata = JSON.parse(metadataString) as {
+      references?: ChatReference[]
+    }
+
+    return {
+      content: response.slice(0, delimiterIndex),
+      references: metadata.references ?? [],
+    }
+  } catch {
+    return {
+      content: response.slice(0, delimiterIndex),
+    }
+  }
 }
 
 @Injectable({