Files
2026-07-13 12:58:18 +08:00

59 lines
1.9 KiB
TypeScript

import Card from "@leafygreen-ui/card";
import { Avatar, Format, AvatarSize } from "@leafygreen-ui/avatar";
import { Spinner } from "@leafygreen-ui/loading-indicator";
import Button from "@leafygreen-ui/button";
import Icon from "@leafygreen-ui/icon";
import "@copilotkit/react-ui/styles.css";
import { AssistantMessageProps, Markdown } from "@copilotkit/react-ui";
import { useCopilotChat } from "@copilotkit/react-core";
export const CustomAssistantMessage = (props: AssistantMessageProps) => {
const { message, isLoading, isGenerating, subComponent, rawData } = props;
const id = rawData?.id;
return (
<div className="py-2">
<div className="flex items-end gap-2">
{!subComponent && (
<Avatar format={Format.MongoDB} size={AvatarSize.XLarge} />
)}
{subComponent ? (
subComponent
) : (
<Card className="flex w-full justify-start flex-col">
{message && <Markdown content={message.content || ""} />}
{isLoading && (
<div className="flex justify-start">
<Spinner />
</div>
)}
{!isGenerating && !isLoading && <ResponseButtons id={id} />}
</Card>
)}
</div>
</div>
);
};
const ResponseButtons = ({ id }: { id: string }) => {
const { reloadMessages } = useCopilotChat();
return (
<div className="flex gap-2 items-center mt-6">
<p className="text-gray-500">How was this response?</p>
<Button size={"xsmall"} onClick={() => alert("Thumbs up sent")}>
<Icon glyph="ThumbsUp" />
</Button>
<Button size={"xsmall"} onClick={() => alert("Thumbs down sent")}>
<Icon glyph="ThumbsDown" />
</Button>
<div className="flex gap-2 items-center">
|
<Button size={"xsmall"} onClick={() => reloadMessages(id)}>
<Icon glyph="Refresh" />
</Button>
</div>
</div>
);
};