77a4ee1c37
当您提供代码差异后，我将按照以下格式生成： ``` <type>(<scope>): <subject> <body> ``` 其中type会根据更改类型选择（feat、fix、docs、style、refactor等），scope表示影响范围，subject简要描述变更内容，body详细说明修改内容。
120 lines
4.0 KiB
Swift
120 lines
4.0 KiB
Swift
import SwiftUI
|
|
|
|
/// 模型推理自检:加载 LLM 跑一段固定 prompt,流式显示输出 + tok/s。
|
|
/// 模型就绪后从「我的 · 模型管理」进入,用于现场快速验证本地推理是否正常。
|
|
struct ModelSelfTestView: View {
|
|
@State private var output = ""
|
|
@State private var phase: Phase = .idle
|
|
@State private var rate: Double = 0
|
|
|
|
private enum Phase: Equatable {
|
|
case idle, loading, running, done, failed(String)
|
|
|
|
var label: String {
|
|
switch self {
|
|
case .idle: return String(appLoc: "未开始")
|
|
case .loading: return String(appLoc: "加载模型…")
|
|
case .running: return String(appLoc: "推理中…")
|
|
case .done: return String(appLoc: "完成 ✓")
|
|
case .failed(let m): return String(appLoc: "失败:\(m)")
|
|
}
|
|
}
|
|
}
|
|
|
|
private let prompt = "用中文一句话介绍肝功能里 ALT 这个指标。"
|
|
|
|
private var isBusy: Bool { phase == .loading || phase == .running }
|
|
|
|
private var statusColor: Color {
|
|
switch phase {
|
|
case .failed: return Tj.Palette.brick
|
|
case .done: return Tj.Palette.leaf
|
|
default: return Tj.Palette.text2
|
|
}
|
|
}
|
|
|
|
var body: some View {
|
|
VStack(alignment: .leading, spacing: 16) {
|
|
VStack(alignment: .leading, spacing: 6) {
|
|
Text("测试 PROMPT")
|
|
.font(.tjScaled( 11, weight: .semibold))
|
|
.tracking(0.5)
|
|
.foregroundStyle(Tj.Palette.text3)
|
|
Text(prompt)
|
|
.font(.tjScaled( 14))
|
|
.foregroundStyle(Tj.Palette.text)
|
|
}
|
|
.padding(14)
|
|
.frame(maxWidth: .infinity, alignment: .leading)
|
|
.tjCard()
|
|
|
|
HStack {
|
|
Text(phase.label)
|
|
.font(.tjScaled( 13, weight: .medium))
|
|
.foregroundStyle(statusColor)
|
|
.lineLimit(1)
|
|
Spacer()
|
|
if rate > 0 {
|
|
Text(String(format: "%.1f tok/s", rate))
|
|
.font(.tjScaled( 12, design: .monospaced))
|
|
.foregroundStyle(Tj.Palette.text3)
|
|
}
|
|
}
|
|
|
|
Button {
|
|
Task { await run() }
|
|
} label: {
|
|
Text(isBusy ? "运行中…" : "运行推理自检").frame(maxWidth: .infinity)
|
|
}
|
|
.buttonStyle(TjPrimaryButton())
|
|
.disabled(isBusy)
|
|
|
|
ScrollView {
|
|
Text(output.isEmpty ? "(暂无输出)" : output)
|
|
.font(.system(.footnote, design: .monospaced))
|
|
.foregroundStyle(Tj.Palette.text)
|
|
.frame(maxWidth: .infinity, alignment: .leading)
|
|
.textSelection(.enabled)
|
|
.padding(12)
|
|
}
|
|
.frame(maxHeight: 280)
|
|
.background(
|
|
RoundedRectangle(cornerRadius: Tj.Radius.md, style: .continuous)
|
|
.fill(Tj.Palette.paper)
|
|
)
|
|
.overlay(
|
|
RoundedRectangle(cornerRadius: Tj.Radius.md, style: .continuous)
|
|
.strokeBorder(Tj.Palette.lineSoft, lineWidth: 1)
|
|
)
|
|
|
|
Spacer()
|
|
}
|
|
.padding(16)
|
|
.background(Tj.Palette.sand.ignoresSafeArea())
|
|
.navigationTitle("推理自检")
|
|
.navigationBarTitleDisplayMode(.inline)
|
|
}
|
|
|
|
@MainActor
|
|
private func run() async {
|
|
output = ""
|
|
rate = 0
|
|
phase = .loading
|
|
do {
|
|
try await AIRuntime.shared.prepare()
|
|
phase = .running
|
|
for try await chunk in await AIRuntime.shared.generate(prompt: prompt, maxTokens: 200) {
|
|
output += chunk.text
|
|
rate = chunk.decodeRate
|
|
}
|
|
phase = .done
|
|
} catch {
|
|
phase = .failed(error.localizedDescription)
|
|
}
|
|
}
|
|
}
|
|
|
|
#Preview {
|
|
NavigationStack { ModelSelfTestView() }
|
|
}
|