mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 08:46:22 +02:00
c979609041
- Introduced a new endpoint for batch document status retrieval, allowing users to check the status of multiple documents in a search space. - Enhanced the document upload process to return duplicate document IDs and improved response structure. - Updated schemas to include new response models for document status. - Removed unused attachment processing code from chat routes and UI components to streamline functionality.
52 lines
1.6 KiB
TypeScript
52 lines
1.6 KiB
TypeScript
import type { ThreadMessageLike } from "@assistant-ui/react";
|
|
import type { MessageRecord } from "./thread-persistence";
|
|
|
|
/**
|
|
* Convert backend message to assistant-ui ThreadMessageLike format
|
|
* Filters out 'thinking-steps' part as it's handled separately via messageThinkingSteps
|
|
*/
|
|
export function convertToThreadMessage(msg: MessageRecord): ThreadMessageLike {
|
|
let content: ThreadMessageLike["content"];
|
|
|
|
if (typeof msg.content === "string") {
|
|
content = [{ type: "text", text: msg.content }];
|
|
} else if (Array.isArray(msg.content)) {
|
|
// Filter out custom metadata parts - they're handled separately
|
|
const filteredContent = msg.content.filter((part: unknown) => {
|
|
if (typeof part !== "object" || part === null || !("type" in part)) return true;
|
|
const partType = (part as { type: string }).type;
|
|
// Filter out metadata parts not directly renderable by assistant-ui
|
|
return (
|
|
partType !== "thinking-steps" &&
|
|
partType !== "mentioned-documents" &&
|
|
partType !== "attachments"
|
|
);
|
|
});
|
|
content =
|
|
filteredContent.length > 0
|
|
? (filteredContent as ThreadMessageLike["content"])
|
|
: [{ type: "text", text: "" }];
|
|
} else {
|
|
content = [{ type: "text", text: String(msg.content) }];
|
|
}
|
|
|
|
// Build metadata.custom for author display in shared chats
|
|
const metadata = msg.author_id
|
|
? {
|
|
custom: {
|
|
author: {
|
|
displayName: msg.author_display_name ?? null,
|
|
avatarUrl: msg.author_avatar_url ?? null,
|
|
},
|
|
},
|
|
}
|
|
: undefined;
|
|
|
|
return {
|
|
id: `msg-${msg.id}`,
|
|
role: msg.role,
|
|
content,
|
|
createdAt: new Date(msg.created_at),
|
|
metadata,
|
|
};
|
|
}
|