feat: add ultra mode

This commit is contained in:
hetao
2026-02-06 15:42:53 +08:00
parent 70989a949e
commit 449ffbad75
18 changed files with 272 additions and 41 deletions

View File

@@ -8,6 +8,7 @@ import {
PaperclipIcon,
PlusIcon,
SparklesIcon,
RocketIcon,
ZapIcon,
} from "lucide-react";
import { useSearchParams } from "next/navigation";
@@ -80,9 +81,9 @@ export function InputBox({
disabled?: boolean;
context: Omit<
AgentThreadContext,
"thread_id" | "is_plan_mode" | "thinking_enabled"
"thread_id" | "is_plan_mode" | "thinking_enabled" | "subagent_enabled"
> & {
mode: "flash" | "thinking" | "pro" | undefined;
mode: "flash" | "thinking" | "pro" | "ultra" | undefined;
};
extraHeader?: React.ReactNode;
isNewThread?: boolean;
@@ -90,9 +91,9 @@ export function InputBox({
onContextChange?: (
context: Omit<
AgentThreadContext,
"thread_id" | "is_plan_mode" | "thinking_enabled"
"thread_id" | "is_plan_mode" | "thinking_enabled" | "subagent_enabled"
> & {
mode: "flash" | "thinking" | "pro" | undefined;
mode: "flash" | "thinking" | "pro" | "ultra" | undefined;
},
) => void;
onSubmit?: (message: PromptInputMessage) => void;
@@ -131,7 +132,7 @@ export function InputBox({
[onContextChange, context],
);
const handleModeSelect = useCallback(
(mode: "flash" | "thinking" | "pro") => {
(mode: "flash" | "thinking" | "pro" | "ultra") => {
onContextChange?.({
...context,
mode,
@@ -205,11 +206,15 @@ export function InputBox({
{context.mode === "pro" && (
<GraduationCapIcon className="size-3" />
)}
{context.mode === "ultra" && (
<RocketIcon className="size-3" />
)}
</div>
<div className="text-xs font-normal">
{(context.mode === "flash" && t.inputBox.flashMode) ||
(context.mode === "thinking" && t.inputBox.reasoningMode) ||
(context.mode === "pro" && t.inputBox.proMode)}
(context.mode === "pro" && t.inputBox.proMode) ||
(context.mode === "ultra" && t.inputBox.ultraMode)}
</div>
</PromptInputActionMenuTrigger>
<PromptInputActionMenuContent className="w-80">
@@ -306,6 +311,34 @@ export function InputBox({
<div className="ml-auto size-4" />
)}
</PromptInputActionMenuItem>
<PromptInputActionMenuItem
className={cn(
context.mode === "ultra"
? "text-accent-foreground"
: "text-muted-foreground/65",
)}
onSelect={() => handleModeSelect("ultra")}
>
<div className="flex flex-col gap-2">
<div className="flex items-center gap-1 font-bold">
<RocketIcon
className={cn(
"mr-2 size-4",
context.mode === "ultra" && "text-accent-foreground",
)}
/>
{t.inputBox.ultraMode}
</div>
<div className="pl-7 text-xs">
{t.inputBox.ultraModeDescription}
</div>
</div>
{context.mode === "ultra" ? (
<CheckIcon className="ml-auto size-4" />
) : (
<div className="ml-auto size-4" />
)}
</PromptInputActionMenuItem>
</PromptInputActionMenu>
</DropdownMenuGroup>
</PromptInputActionMenuContent>

View File

@@ -0,0 +1,117 @@
"use client";
import { CheckCircleIcon, Loader2Icon, SquareTerminalIcon, WrenchIcon, XCircleIcon } from "lucide-react";
import { MessageResponse } from "@/components/ai-elements/message";
import { useI18n } from "@/core/i18n/hooks";
import { cn } from "@/lib/utils";
import type { SubagentState } from "@/core/threads/types";
interface SubagentCardProps {
subagentType: string;
state?: SubagentState;
isLoading?: boolean;
prompt?: string;
}
export function SubagentCard({ subagentType, state, isLoading, prompt }: SubagentCardProps) {
const { t } = useI18n();
const getSubagentIcon = (type: string) => {
switch (type) {
case "bash":
return SquareTerminalIcon;
case "general-purpose":
return WrenchIcon;
default:
return WrenchIcon;
}
};
const getSubagentLabel = (type: string) => {
switch (type) {
case "bash":
return t.subagents.bash;
case "general-purpose":
return t.subagents.generalPurpose;
default:
return t.subagents.unknown;
}
};
const IconComponent = getSubagentIcon(subagentType);
const label = getSubagentLabel(subagentType);
// Determine status based on state, not isLoading
const status = state?.status || "running";
const isRunning = status === "running";
const isCompleted = status === "completed";
const isFailed = status === "failed";
const getStatusIcon = () => {
if (isCompleted) {
return <CheckCircleIcon className="size-4 text-green-600" />;
}
if (isFailed) {
return <XCircleIcon className="size-4 text-red-600" />;
}
if (isRunning) {
return <Loader2Icon className="size-4 animate-spin text-blue-600" />;
}
return null;
};
const borderColorClass = isCompleted
? "border-green-200 bg-green-50/30"
: isFailed
? "border-red-200 bg-red-50/30"
: "border-blue-200 bg-blue-50/30";
return (
<div className={cn(
"rounded-lg border-l-2 p-4 transition-colors space-y-3",
borderColorClass
)}>
{/* Header */}
<div className="flex items-start gap-2">
<div className="mt-0.5 flex-shrink-0">
<IconComponent className="size-4" />
</div>
<div className="flex-1 min-w-0">
<div className="flex items-center gap-2">
<span className="font-medium text-sm">{label}</span>
{getStatusIcon()}
</div>
{prompt && (
<div className="mt-1 text-xs text-muted-foreground">
{prompt}
</div>
)}
</div>
</div>
{/* Status message for running state */}
{isRunning && !state?.result && (
<div className="text-sm text-muted-foreground ml-6">
{t.subagents.running}
</div>
)}
{/* Result */}
{state?.result && (
<div className="ml-6 text-sm">
<MessageResponse>{state.result}</MessageResponse>
</div>
)}
{/* Error */}
{state?.status === "failed" && state.error && (
<div className="ml-6 rounded-md bg-red-50 p-3 text-sm text-red-700 border border-red-200">
<div className="font-medium">{t.subagents.failed}</div>
<div className="mt-1 text-xs">{state.error}</div>
</div>
)}
</div>
);
}