TradingAgents/frontend/components/analysis/AnalysisForm.tsx

792 lines
32 KiB
TypeScript
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/**
* Analysis form component
*/
"use client";
import { useState, useEffect } from "react";
import { useForm, ControllerRenderProps } from "react-hook-form";
import { zodResolver } from "@hookform/resolvers/zod";
import * as z from "zod";
import { format } from "date-fns";
import { CheckIcon } from "lucide-react";
import { getApiSettingsAsync } from "@/lib/storage";
import { getBaseUrlForModel, getApiKeyForModel } from "@/lib/api-helpers";
import { cn } from "@/lib/utils";
import { Button } from "@/components/ui/button";
import { DatePicker } from "@/components/ui/date-picker";
import {
Form,
FormControl,
FormDescription,
FormField,
FormItem,
FormLabel,
FormMessage,
} from "@/components/ui/form";
import { Input } from "@/components/ui/input";
import { Checkbox } from "@/components/ui/checkbox";
import {
Select,
SelectContent,
SelectItem,
SelectTrigger,
SelectValue,
} from "@/components/ui/select";
import {
Card,
CardContent,
CardDescription,
CardHeader,
CardTitle,
} from "@/components/ui/card";
import type { AnalysisRequest } from "@/lib/types";
const formSchema = z.object({
ticker: z.string().min(1, "股票代碼為必填").max(10),
analysis_date: z
.string()
.regex(/^\d{4}-\d{2}-\d{2}$/, "日期格式必須為 YYYY-MM-DD"),
analysts: z.array(z.string()).min(1, "請至少選擇一位分析師"),
research_depth: z.number().int().min(1).max(5),
quick_think_llm: z.string().min(1, "請選擇快速思維模型"),
deep_think_llm: z.string().min(1, "請選擇深層思維模型"),
// Market type selection: us=美股, twse=上市, tpex=上櫃/興櫃
market_type: z.enum(["us", "twse", "tpex"]),
// Custom model names (when "custom" is selected)
custom_quick_think_model: z.string().optional(),
custom_deep_think_model: z.string().optional(),
// API Configuration (hidden from UI, populated from localStorage)
quick_think_base_url: z
.string()
.url("請輸入有效的 URL")
.optional()
.or(z.literal("")),
deep_think_base_url: z
.string()
.url("請輸入有效的 URL")
.optional()
.or(z.literal("")),
quick_think_api_key: z.string().min(1, "請輸入快速思維模型 API Key"),
deep_think_api_key: z.string().min(1, "請輸入深層思維模型 API Key"),
embedding_base_url: z
.string()
.url("請輸入有效的 URL")
.optional()
.or(z.literal("")),
embedding_api_key: z.string().min(1, "請輸入嵌入模型 API Key"),
alpha_vantage_api_key: z.string().optional().or(z.literal("")), // 選填
finmind_api_key: z.string().optional().or(z.literal("")), // 選填
});
interface AnalysisFormProps {
onSubmit: (data: AnalysisRequest) => void;
loading?: boolean;
}
const ANALYSTS = [
{ value: "market", label: "市場分析師" },
{ value: "social", label: "社群媒體分析師" },
{ value: "news", label: "新聞分析師" },
{ value: "fundamentals", label: "基本面分析師" },
];
export function AnalysisForm({ onSubmit, loading = false }: AnalysisFormProps) {
const form = useForm<z.infer<typeof formSchema>>({
resolver: zodResolver(formSchema),
defaultValues: {
ticker: "NVDA",
analysis_date: format(new Date(), "yyyy-MM-dd"),
analysts: ["market", "social", "news", "fundamentals"], // 預設全選
research_depth: 3, // 預設中等層級
market_type: "us", // 預設美股
quick_think_llm: "gpt-5-mini",
deep_think_llm: "gpt-5-mini",
custom_quick_think_model: "",
custom_deep_think_model: "",
quick_think_base_url: "https://api.openai.com/v1",
deep_think_base_url: "https://api.openai.com/v1",
quick_think_api_key: "",
deep_think_api_key: "",
embedding_base_url: "https://api.openai.com/v1",
embedding_api_key: "",
alpha_vantage_api_key: "",
finmind_api_key: "",
},
});
// Load API settings from localStorage and update when models change
const quickThinkLlm = form.watch("quick_think_llm");
const deepThinkLlm = form.watch("deep_think_llm");
const marketType = form.watch("market_type");
const isQuickThinkCustom = quickThinkLlm === "custom";
const isDeepThinkCustom = deepThinkLlm === "custom";
useEffect(() => {
// Use async version to get decrypted API keys
const loadSettings = async () => {
const savedSettings = await getApiSettingsAsync();
// For custom models, always use custom base URL and API key
if (isQuickThinkCustom) {
form.setValue(
"quick_think_base_url",
savedSettings.custom_base_url || ""
);
form.setValue(
"quick_think_api_key",
savedSettings.custom_api_key || ""
);
} else {
form.setValue(
"quick_think_base_url",
getBaseUrlForModel(quickThinkLlm, savedSettings.custom_base_url)
);
form.setValue(
"quick_think_api_key",
getApiKeyForModel(quickThinkLlm, savedSettings)
);
}
if (isDeepThinkCustom) {
form.setValue(
"deep_think_base_url",
savedSettings.custom_base_url || ""
);
form.setValue("deep_think_api_key", savedSettings.custom_api_key || "");
} else {
form.setValue(
"deep_think_base_url",
getBaseUrlForModel(deepThinkLlm, savedSettings.custom_base_url)
);
form.setValue(
"deep_think_api_key",
getApiKeyForModel(deepThinkLlm, savedSettings)
);
}
form.setValue(
"embedding_base_url",
savedSettings.custom_base_url || "https://api.openai.com/v1"
);
form.setValue(
"embedding_api_key",
savedSettings.custom_api_key || savedSettings.openai_api_key
);
form.setValue(
"alpha_vantage_api_key",
savedSettings.alpha_vantage_api_key || ""
);
form.setValue("finmind_api_key", savedSettings.finmind_api_key || "");
};
loadSettings();
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [quickThinkLlm, deepThinkLlm, isQuickThinkCustom, isDeepThinkCustom]);
// 當市場類型改變時,更新預設股票代碼和提示
useEffect(() => {
const currentTicker = form.getValues("ticker");
// 只在用戶未修改預設值時才自動切換
const isTwStock = marketType === "twse" || marketType === "tpex";
const isDefaultUsTicker =
currentTicker === "NVDA" || currentTicker === "AAPL";
const isDefaultTwTicker =
currentTicker === "2330" ||
currentTicker === "2317" ||
currentTicker === "6488";
if (isTwStock && isDefaultUsTicker) {
form.setValue("ticker", marketType === "twse" ? "2330" : "6488");
} else if (marketType === "us" && isDefaultTwTicker) {
form.setValue("ticker", "NVDA");
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [marketType]);
// 全選/取消全選
const toggleSelectAll = () => {
const currentAnalysts = form.getValues("analysts");
if (currentAnalysts.length === ANALYSTS.length) {
form.setValue("analysts", []);
} else {
form.setValue(
"analysts",
ANALYSTS.map((a) => a.value)
);
}
};
function handleSubmit(values: z.infer<typeof formSchema>) {
// Use custom model names if "custom" is selected
const finalQuickThinkLlm =
values.quick_think_llm === "custom"
? values.custom_quick_think_model || ""
: values.quick_think_llm;
const finalDeepThinkLlm =
values.deep_think_llm === "custom"
? values.custom_deep_think_model || ""
: values.deep_think_llm;
// Validate custom model names
if (
values.quick_think_llm === "custom" &&
!values.custom_quick_think_model
) {
form.setError("custom_quick_think_model", {
type: "manual",
message: "請輸入快速思維模型的完整名稱",
});
return;
}
if (values.deep_think_llm === "custom" && !values.custom_deep_think_model) {
form.setError("custom_deep_think_model", {
type: "manual",
message: "請輸入深層思維模型的完整名稱",
});
return;
}
const request: AnalysisRequest = {
...values,
quick_think_llm: finalQuickThinkLlm,
deep_think_llm: finalDeepThinkLlm,
};
onSubmit(request);
}
return (
<Card className="shadow-lg hover-lift animate-scale-up">
<CardContent className="pt-6">
<Form {...form}>
<form
onSubmit={form.handleSubmit(handleSubmit)}
className="space-y-6"
>
<div className="grid grid-cols-1 md:grid-cols-2 gap-6">
{/* 分析師選擇區塊 - 全寬 */}
<div className="md:col-span-2 border-b pb-6">
<div className="flex justify-between items-center mb-4">
<FormLabel className="text-base font-semibold">
</FormLabel>
<Button
type="button"
variant="outline"
size="sm"
onClick={toggleSelectAll}
>
{form.watch("analysts").length === ANALYSTS.length
? "取消全選"
: "全選"}
</Button>
</div>
<FormField
control={form.control}
name="analysts"
render={({ field }) => (
<FormItem>
<div className="grid grid-cols-2 md:grid-cols-4 gap-4">
{ANALYSTS.map((analyst) => {
const isSelected = field.value?.includes(
analyst.value
);
return (
<FormItem key={analyst.value} className="space-y-0">
<FormControl>
<div
onClick={() => {
const newValue = isSelected
? field.value?.filter(
(v: string) => v !== analyst.value
)
: [...(field.value ?? []), analyst.value];
field.onChange(newValue);
}}
className={cn(
"relative flex cursor-pointer flex-row items-center gap-3 rounded-lg border-2 p-4 transition-all hover:bg-accent",
isSelected
? "border-primary bg-primary/5 text-primary"
: "border-muted-foreground/25 bg-card text-muted-foreground"
)}
>
<div
className={cn(
"flex h-5 w-5 shrink-0 items-center justify-center rounded-sm border transition-colors",
isSelected
? "border-primary bg-primary text-primary-foreground"
: "border-muted-foreground"
)}
>
{isSelected && (
<CheckIcon className="h-3.5 w-3.5" />
)}
</div>
<span className="font-medium select-none">
{analyst.label}
</span>
</div>
</FormControl>
</FormItem>
);
})}
</div>
<FormMessage />
</FormItem>
)}
/>
</div>
{/* 第一行市場類型、股票代碼、分析日期3列 */}
<div className="md:col-span-2 grid grid-cols-1 md:grid-cols-3 gap-6">
{/* 市場類型選擇 */}
<FormField
control={form.control}
name="market_type"
render={({ field }) => (
<FormItem>
<FormLabel></FormLabel>
<Select
onValueChange={field.onChange}
defaultValue={field.value}
>
<FormControl>
<SelectTrigger>
<SelectValue placeholder="選擇市場" />
</SelectTrigger>
</FormControl>
<SelectContent>
<SelectItem
value="us"
className="py-3 cursor-pointer"
>
🇺🇸
</SelectItem>
<SelectItem
value="twse"
className="py-3 cursor-pointer"
>
🇹🇼
</SelectItem>
<SelectItem
value="tpex"
className="py-3 cursor-pointer"
>
🇹🇼 /
</SelectItem>
</SelectContent>
</Select>
<FormDescription></FormDescription>
<FormMessage />
</FormItem>
)}
/>
{/* 股票代碼 */}
<FormField
control={form.control}
name="ticker"
render={({ field }) => (
<FormItem>
<FormLabel></FormLabel>
<FormControl>
<Input
placeholder={
marketType === "us"
? "NVDA"
: marketType === "twse"
? "2330"
: "6488"
}
{...field}
/>
</FormControl>
<FormDescription>
{marketType === "us"
? "輸入美股代碼例如NVDA、AAPL"
: marketType === "twse"
? "輸入上市股票代碼例如2330、2317"
: "輸入上櫃/興櫃股票代碼例如6488、5765"}
</FormDescription>
<FormMessage />
</FormItem>
)}
/>
<FormField
control={form.control}
name="analysis_date"
render={({ field }) => (
<FormItem>
<FormLabel></FormLabel>
<FormControl>
<DatePicker
date={field.value ? new Date(field.value) : undefined}
onDateChange={(date) => {
field.onChange(
date ? format(date, "yyyy-MM-dd") : ""
);
}}
placeholder="選擇分析日期"
className="w-full"
/>
</FormControl>
<FormDescription></FormDescription>
<FormMessage />
</FormItem>
)}
/>
</div>
{/* 第二行研究深度、快速思維模型、深層思維模型3列 */}
<div className="md:col-span-2 grid grid-cols-1 md:grid-cols-3 gap-6">
<FormField
control={form.control}
name="research_depth"
render={({ field }) => (
<FormItem>
<FormLabel></FormLabel>
<Select
onValueChange={(value) =>
field.onChange(parseInt(value))
}
defaultValue={field.value?.toString() ?? "3"}
>
<FormControl>
<SelectTrigger>
<SelectValue placeholder="選擇研究深度" />
</SelectTrigger>
</FormControl>
<SelectContent className="max-h-80">
<SelectItem value="1" className="py-3 cursor-pointer">
-
</SelectItem>
<SelectItem value="3" className="py-3 cursor-pointer">
-
</SelectItem>
<SelectItem value="5" className="py-3 cursor-pointer">
-
</SelectItem>
</SelectContent>
</Select>
<FormDescription></FormDescription>
<FormMessage />
</FormItem>
)}
/>
<FormField
control={form.control}
name="quick_think_llm"
render={({ field }) => (
<FormItem>
<FormLabel></FormLabel>
<Select
onValueChange={field.onChange}
defaultValue={field.value}
>
<FormControl>
<SelectTrigger>
<SelectValue placeholder="選擇模型" />
</SelectTrigger>
</FormControl>
<SelectContent>
{/* OpenAI */}
<SelectItem value="gpt-5.2-2025-12-11">
OpenAI: GPT-5.2
</SelectItem>
<SelectItem value="gpt-5.1">
OpenAI: GPT-5.1
</SelectItem>
<SelectItem value="gpt-5-mini">
OpenAI: GPT-5 Mini
</SelectItem>
<SelectItem value="gpt-5-nano">
OpenAI: GPT-5 Nano
</SelectItem>
<SelectItem value="gpt-4.1-mini">
OpenAI: GPT-4.1 Mini
</SelectItem>
<SelectItem value="gpt-4.1-nano">
OpenAI: GPT-4.1 Nano
</SelectItem>
<SelectItem value="o4-mini">
OpenAI: o4-mini
</SelectItem>
{/* Anthropic (Official model IDs) */}
<SelectItem value="claude-sonnet-4-5-20250929">
Anthropic: Claude Sonnet 4.5
</SelectItem>
<SelectItem value="claude-haiku-4-5-20251001">
Anthropic: Claude Haiku 4.5
</SelectItem>
<SelectItem value="claude-sonnet-4-20250514">
Anthropic: Claude Sonnet 4
</SelectItem>
<SelectItem value="claude-3-7-sonnet-20250219">
Anthropic: Claude 3.7 Sonnet
</SelectItem>
<SelectItem value="claude-3-5-haiku-20241022">
Anthropic: Claude 3.5 Haiku
</SelectItem>
<SelectItem value="claude-3-haiku-20240307">
Anthropic: Claude 3 Haiku
</SelectItem>
{/* Google */}
<SelectItem value="gemini-2.5-pro">
Google: Gemini 2.5 Pro
</SelectItem>
<SelectItem value="gemini-2.5-flash">
Google: Gemini 2.5 Flash
</SelectItem>
<SelectItem value="gemini-2.5-flash-lite">
Google: Gemini 2.5 Flash Lite
</SelectItem>
<SelectItem value="gemini-2.0-flash">
Google: Gemini 2.0 Flash
</SelectItem>
<SelectItem value="gemini-2.0-flash-lite">
Google: Gemini 2.0 Flash Lite
</SelectItem>
{/* Grok */}
<SelectItem value="grok-4-1-fast-reasoning">
Grok: 4.1 Fast Reasoning
</SelectItem>
<SelectItem value="grok-4-1-fast-non-reasoning">
Grok: 4.1 Fast Non Reasoning
</SelectItem>
<SelectItem value="grok-4-fast-reasoning">
Grok: 4 Fast Reasoning
</SelectItem>
<SelectItem value="grok-4-fast-non-reasoning">
Grok: 4 Fast Non Reasoning
</SelectItem>
<SelectItem value="grok-4-0709">Grok: 4</SelectItem>
<SelectItem value="grok-3">Grok: 3</SelectItem>
<SelectItem value="grok-3-mini">
Grok: 3 Mini
</SelectItem>
{/* DeepSeek */}
<SelectItem value="deepseek-reasoner">
DeepSeek: Reasoner
</SelectItem>
<SelectItem value="deepseek-chat">
DeepSeek: Chat
</SelectItem>
{/* Qwen */}
<SelectItem value="qwen3-max">Qwen: 3 Max</SelectItem>
<SelectItem value="qwen-plus">Qwen: Plus</SelectItem>
<SelectItem value="qwen-flash">
Qwen: Flash
</SelectItem>
{/* Custom Model */}
<SelectItem value="custom">
Other
</SelectItem>
</SelectContent>
</Select>
<FormDescription></FormDescription>
<FormMessage />
</FormItem>
)}
/>
{/* Custom Quick Think Model Input */}
{isQuickThinkCustom && (
<FormField
control={form.control}
name="custom_quick_think_model"
render={({ field }) => (
<FormItem className="md:col-span-3 animate-scale-up">
<FormLabel></FormLabel>
<FormControl>
<Input placeholder="例如deepseek-chat" {...field} />
</FormControl>
<FormDescription>
使
</FormDescription>
<FormMessage />
</FormItem>
)}
/>
)}
<FormField
control={form.control}
name="deep_think_llm"
render={({ field }) => (
<FormItem>
<FormLabel></FormLabel>
<Select
onValueChange={field.onChange}
defaultValue={field.value}
>
<FormControl>
<SelectTrigger>
<SelectValue placeholder="選擇模型" />
</SelectTrigger>
</FormControl>
<SelectContent>
{/* OpenAI */}
<SelectItem value="gpt-5.2-2025-12-11">
OpenAI: GPT-5.2
</SelectItem>
<SelectItem value="gpt-5.1">
OpenAI: GPT-5.1
</SelectItem>
<SelectItem value="gpt-5-mini">
OpenAI: GPT-5 Mini
</SelectItem>
<SelectItem value="gpt-5-nano">
OpenAI: GPT-5 Nano
</SelectItem>
<SelectItem value="gpt-4.1-mini">
OpenAI: GPT-4.1 Mini
</SelectItem>
<SelectItem value="gpt-4.1-nano">
OpenAI: GPT-4.1 Nano
</SelectItem>
<SelectItem value="o4-mini">
OpenAI: o4-mini
</SelectItem>
{/* Anthropic (Official model IDs) */}
<SelectItem value="claude-sonnet-4-5-20250929">
Anthropic: Claude Sonnet 4.5
</SelectItem>
<SelectItem value="claude-haiku-4-5-20251001">
Anthropic: Claude Haiku 4.5
</SelectItem>
<SelectItem value="claude-sonnet-4-20250514">
Anthropic: Claude Sonnet 4
</SelectItem>
<SelectItem value="claude-3-7-sonnet-20250219">
Anthropic: Claude 3.7 Sonnet
</SelectItem>
<SelectItem value="claude-3-5-haiku-20241022">
Anthropic: Claude 3.5 Haiku
</SelectItem>
<SelectItem value="claude-3-haiku-20240307">
Anthropic: Claude 3 Haiku
</SelectItem>
{/* Google */}
<SelectItem value="gemini-2.5-pro">
Google: Gemini 2.5 Pro
</SelectItem>
<SelectItem value="gemini-2.5-flash">
Google: Gemini 2.5 Flash
</SelectItem>
<SelectItem value="gemini-2.5-flash-lite">
Google: Gemini 2.5 Flash Lite
</SelectItem>
<SelectItem value="gemini-2.0-flash">
Google: Gemini 2.0 Flash
</SelectItem>
<SelectItem value="gemini-2.0-flash-lite">
Google: Gemini 2.0 Flash Lite
</SelectItem>
{/* Grok */}
<SelectItem value="grok-4-1-fast-reasoning">
Grok: 4.1 Fast Reasoning
</SelectItem>
<SelectItem value="grok-4-1-fast-non-reasoning">
Grok: 4.1 Fast Non Reasoning
</SelectItem>
<SelectItem value="grok-4-fast-reasoning">
Grok: 4 Fast Reasoning
</SelectItem>
<SelectItem value="grok-4-fast-non-reasoning">
Grok: 4 Fast Non Reasoning
</SelectItem>
<SelectItem value="grok-4-0709">Grok: 4</SelectItem>
<SelectItem value="grok-3">Grok: 3</SelectItem>
<SelectItem value="grok-3-mini">
Grok: 3 Mini
</SelectItem>
{/* DeepSeek */}
<SelectItem value="deepseek-reasoner">
DeepSeek: Reasoner
</SelectItem>
<SelectItem value="deepseek-chat">
DeepSeek: Chat
</SelectItem>
{/* Qwen */}
<SelectItem value="qwen3-max">Qwen: 3 Max</SelectItem>
<SelectItem value="qwen-plus">Qwen: Plus</SelectItem>
<SelectItem value="qwen-flash">
Qwen: Flash
</SelectItem>
{/* Custom Model */}
<SelectItem value="custom">
Other
</SelectItem>
</SelectContent>
</Select>
<FormDescription></FormDescription>
<FormMessage />
</FormItem>
)}
/>
{/* Custom Deep Think Model Input */}
{isDeepThinkCustom && (
<FormField
control={form.control}
name="custom_deep_think_model"
render={({ field }) => (
<FormItem className="md:col-span-3 animate-scale-up">
<FormLabel></FormLabel>
<FormControl>
<Input placeholder="例如deepseek-chat" {...field} />
</FormControl>
<FormDescription>
使
</FormDescription>
<FormMessage />
</FormItem>
)}
/>
)}
</div>
</div>
<Button
type="submit"
className="w-full bg-gradient-to-r from-blue-500 to-pink-500 dark:from-blue-600 dark:to-purple-600 hover:from-blue-600 hover:to-pink-600 dark:hover:from-blue-700 dark:hover:to-purple-700 shadow-lg hover:shadow-xl transition-all animate-heartbeat"
disabled={loading}
size="lg"
style={{
touchAction: "manipulation",
WebkitTapHighlightColor: "transparent",
}}
onClick={(e) => {
// Ensure touch events work on Safari mobile
e.currentTarget.blur();
}}
>
{loading ? "執行分析中..." : "執行分析"}
</Button>
</form>
</Form>
</CardContent>
</Card>
);
}