Remove unused stuff

Author: cte

webview-ui/src/components/ui/chat/ChatAnnotations.tsx

@@ -1,4 +1,4 @@
-import { Fragment, useMemo } from "react"
+import { useMemo } from "react"
 import { CopyIcon, CheckIcon } from "@radix-ui/react-icons"
 import { BrainCircuit, CircleUserRound } from "lucide-react"
 
@@ -9,15 +9,7 @@ import { Badge } from "@/components/ui"
 import { BadgeData, ChatHandler, Message, MessageAnnotationType } from "./types"
 import { ChatMessageProvider } from "./ChatMessageProvider"
 import { useChatMessage } from "./useChatMessage"
-import {
-	EventAnnotations,
-	AgentEventAnnotations,
-	ImageAnnotations,
-	MarkdownAnnotations,
-	DocumentFileAnnotations,
-	SourceAnnotations,
-	SuggestedQuestionsAnnotations,
-} from "./ChatAnnotations"
+import { Markdown } from "./markdown"
 
 /**
  * ChatMessage
@@ -100,82 +92,18 @@ function ChatMessageAvatar() {
  * ChatMessageContent
  */
 
-export enum ContentPosition {
-	TOP = -9999,
-	CHAT_EVENTS = 0,
-	AFTER_EVENTS = 1,
-	CHAT_AGENT_EVENTS = 2,
-	AFTER_AGENT_EVENTS = 3,
-	CHAT_IMAGE = 4,
-	AFTER_IMAGE = 5,
-	BEFORE_MARKDOWN = 6,
-	MARKDOWN = 7,
-	AFTER_MARKDOWN = 8,
-	CHAT_DOCUMENT_FILES = 9,
-	AFTER_DOCUMENT_FILES = 10,
-	CHAT_SOURCES = 11,
-	AFTER_SOURCES = 12,
-	SUGGESTED_QUESTIONS = 13,
-	AFTER_SUGGESTED_QUESTIONS = 14,
-	BOTTOM = 9999,
-}
-
-type ContentDisplayConfig = {
-	position: ContentPosition
-	component: React.ReactNode | null
-}
-
 interface ChatMessageContentProps {
 	isHeaderVisible: boolean
-	isLoading?: boolean
-	content?: ContentDisplayConfig[]
-	append?: ChatHandler["append"]
 }
 
-function ChatMessageContent({ isHeaderVisible, isLoading, content, append }: ChatMessageContentProps) {
-	const { message, isLast } = useChatMessage()
-
-	const contents = useMemo<ContentDisplayConfig[]>(() => {
-		const displayMap: {
-			[key in ContentPosition]?: React.ReactNode | null
-		} = {
-			[ContentPosition.CHAT_EVENTS]: (
-				<EventAnnotations message={message} showLoading={(isLast && isLoading) ?? false} />
-			),
-			[ContentPosition.CHAT_AGENT_EVENTS]: <AgentEventAnnotations message={message} />,
-			[ContentPosition.CHAT_IMAGE]: <ImageAnnotations message={message} />,
-			[ContentPosition.MARKDOWN]: <MarkdownAnnotations message={message} />,
-			[ContentPosition.CHAT_DOCUMENT_FILES]: <DocumentFileAnnotations message={message} />,
-			[ContentPosition.CHAT_SOURCES]: <SourceAnnotations message={message} />,
-			...(isLast &&
-				append && {
-					[ContentPosition.SUGGESTED_QUESTIONS]: (
-						<SuggestedQuestionsAnnotations message={message} append={append} />
-					),
-				}),
-		}
-
-		// Override the default display map with the custom content.
-		content?.forEach((content) => {
-			displayMap[content.position] = content.component
-		})
-
-		return Object.entries(displayMap).map(([position, component]) => ({
-			position: parseInt(position),
-			component,
-		}))
-	}, [isLast, isLoading, content, append, message])
+function ChatMessageContent({ isHeaderVisible }: ChatMessageContentProps) {
+	const {
+		message: { content },
+	} = useChatMessage()
 
 	return (
-		<div
-			className={cn("flex flex-col gap-4 flex-1 min-w-0 px-4 pb-6", {
-				"pt-4": isHeaderVisible,
-			})}>
-			{contents
-				.sort((a, b) => a.position - b.position)
-				.map(({ component }, index) => (
-					<Fragment key={index}>{component}</Fragment>
-				))}
+		<div className={cn("flex flex-col gap-4 flex-1 min-w-0 px-4 pb-6", { "pt-4": isHeaderVisible })}>
+			<Markdown content={content} />
 		</div>
 	)
 }