Kaynağa Gözat

feat(app/log): Add Referenced Content in Application Logs (#7082)

Kevin9703 8 ay önce
ebeveyn
işleme
2c188a45c8

+ 1 - 0
web/app/components/app/log/list.tsx

@@ -126,6 +126,7 @@ const getFormattedChatList = (messages: ChatMessage[], conversationId: string, t
         tokens: item.answer_tokens + item.message_tokens,
         latency: item.provider_response_latency.toFixed(2),
       },
+      citation: item.metadata?.retriever_resources,
       annotation: (() => {
         if (item.annotation_hit_history) {
           return {

+ 19 - 11
web/app/components/base/chat/chat/type.ts

@@ -13,8 +13,14 @@ export type Feedbacktype = {
   content?: string | null
 }
 
-export type FeedbackFunc = (messageId: string, feedback: Feedbacktype) => Promise<any>
-export type SubmitAnnotationFunc = (messageId: string, content: string) => Promise<any>
+export type FeedbackFunc = (
+  messageId: string,
+  feedback: Feedbacktype
+) => Promise<any>
+export type SubmitAnnotationFunc = (
+  messageId: string,
+  content: string
+) => Promise<any>
 
 export type DisplayScene = 'web' | 'console'
 
@@ -91,20 +97,22 @@ export type IChatItem = {
   input?: any
 }
 
-export type MessageEnd = {
-  id: string
-  metadata: {
-    retriever_resources?: CitationItem[]
-    annotation_reply: {
+export type Metadata = {
+  retriever_resources?: CitationItem[]
+  annotation_reply: {
+    id: string
+    account: {
       id: string
-      account: {
-        id: string
-        name: string
-      }
+      name: string
     }
   }
 }
 
+export type MessageEnd = {
+  id: string
+  metadata: Metadata
+}
+
 export type MessageReplace = {
   id: string
   task_id: string

+ 2 - 0
web/models/log.ts

@@ -4,6 +4,7 @@ import type {
   Edge,
   Node,
 } from '@/app/components/workflow/types'
+import type { Metadata } from '@/app/components/base/chat/chat/type'
 
 // Log type contains key:string conversation_id:string created_at:string quesiton:string answer:string
 export type Conversation = {
@@ -102,6 +103,7 @@ export type MessageContent = {
     from_end_user_id?: string
   }>
   message_files: VisionFile[]
+  metadata: Metadata
   agent_thoughts: any[] // TODO
   workflow_run_id: string
 }