feat: add reasoning tests (#856)

This commit is contained in:
Jeremy 2025-03-09 21:02:19 -07:00 committed by GitHub
parent 39729644df
commit 9628c54755
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 420 additions and 155 deletions

View file

@ -47,7 +47,7 @@ const PurePreviewMessage = ({
return (
<AnimatePresence>
<motion.div
data-testid={`message-${message.role}-${index}`}
data-testid={`message-${message.role}`}
className="w-full mx-auto max-w-3xl px-4 group/message"
initial={{ y: 5, opacity: 0 }}
animate={{ y: 0, opacity: 1 }}
@ -73,7 +73,7 @@ const PurePreviewMessage = ({
<div className="flex flex-col gap-4 w-full">
{message.experimental_attachments && (
<div
data-testid={`message-attachments-${index}`}
data-testid={`message-attachments`}
className="flex flex-row justify-end gap-2"
>
{message.experimental_attachments.map((attachment) => (
@ -93,12 +93,15 @@ const PurePreviewMessage = ({
)}
{(message.content || message.reasoning) && mode === 'view' && (
<div className="flex flex-row gap-2 items-start">
<div
data-testid="message-content"
className="flex flex-row gap-2 items-start"
>
{message.role === 'user' && !isReadonly && (
<Tooltip>
<TooltipTrigger asChild>
<Button
data-testid={`edit-${message.role}-${index}`}
data-testid={`message-edit`}
variant="ghost"
className="px-2 h-fit rounded-full text-muted-foreground opacity-0 group-hover/message:opacity-100"
onClick={() => {
@ -243,6 +246,7 @@ export const ThinkingMessage = () => {
return (
<motion.div
data-testid="message-assistant-loading"
className="w-full mx-auto max-w-3xl px-4 group/message "
initial={{ y: 5, opacity: 0 }}
animate={{ y: 0, opacity: 1, transition: { delay: 1 } }}