feat: P4 多模态 — 图片上传 + 沙盒 vision 支持

后端:
- ChatMessage 接口支持 text | image_url 多部分内容
- sandbox chat 接受 images[] 参数
- 自动将图片转为 image_url 格式发送到 AI
- 上传模块支持 jpg/png/gif/webp/svg

前端:
- 沙盒输入区 + 图片上传按钮
- 图片缩略图预览 + 删除
- 上传后随消息发送到 API
- uploads/ 目录持久化存储
This commit is contained in:
yuzhiran-dev
2026-05-18 14:15:56 +08:00
parent 0ed017b212
commit 79d7535c1c
6 changed files with 71 additions and 15 deletions
+3 -2
View File
@@ -2,7 +2,7 @@ import { Injectable, Logger } from '@nestjs/common';
interface ChatMessage {
role: 'user' | 'assistant' | 'system';
content: string;
content: string | { type: 'text' | 'image_url'; text?: string; image_url?: { url: string } }[];
}
export interface ChatOptions {
@@ -112,7 +112,8 @@ export class AIGatewayService {
}
private fallback(messages: ChatMessage[]): string {
const lastMsg = messages[messages.length - 1]?.content || '';
const lastMsg = typeof messages[messages.length - 1]?.content === 'string'
? messages[messages.length - 1]?.content as string : '';
const mockReplies: Record<string, string> = {
'你好': '你好!我是宇之然 AI 助手,很高兴为你服务!',
'hello': 'Hello! I am YuZhiRan AI assistant, nice to meet you!',
@@ -12,9 +12,9 @@ export class SandboxController {
constructor(private sandboxService: SandboxService) {}
@Post('chat')
@ApiBody({ schema: { example: { conversationId: 'uuid', model: 'general', messages: [{ role: 'user', content: 'hi' }], temperature: 0.7, top_p: 1, max_tokens: 2000 } } })
async chat(@Req() req: any, @Body() body: { conversationId?: string; model: string; messages: { role: string; content: string }[] } & ChatOptions) {
return this.sandboxService.chat(req.user.userId, body.conversationId, body.model, body.messages, body);
@ApiBody({ schema: { example: { conversationId: 'uuid', model: 'general', messages: [{ role: 'user', content: 'hi' }], images: ['https://example.com/img.png'], temperature: 0.7, top_p: 1, max_tokens: 2000 } } })
async chat(@Req() req: any, @Body() body: { conversationId?: string; model: string; messages: { role: string; content: string }[]; images?: string[] } & ChatOptions) {
return this.sandboxService.chat(req.user.userId, body.conversationId, body.model, body.messages, body, body.images);
}
@Get('sessions')
+17 -2
View File
@@ -10,7 +10,7 @@ export class SandboxService {
private aiGateway: AIGatewayService,
) {}
async chat(userId: number, conversationId: string | undefined, model: string, messages: { role: string; content: string }[], options?: ChatOptions) {
async chat(userId: number, conversationId: string | undefined, model: string, messages: { role: string; content: string }[], options?: ChatOptions, images?: string[]) {
const user = await this.prisma.user.findUnique({ where: { id: userId } });
if (!user || user.status !== 'ACTIVE') {
throw new HttpException('用户不可用', HttpStatus.FORBIDDEN);
@@ -33,7 +33,22 @@ export class SandboxService {
}
}
let reply = await this.aiGateway.chat(model, messages as any, options);
// Convert images to structured message content
let aiMessages = messages as any[];
if (images && images.length > 0) {
aiMessages = messages.map(m => {
if (m.role === 'user' && m === messages[messages.length - 1]) {
const parts: any[] = [{ type: 'text', text: m.content }];
for (const img of images) {
parts.push({ type: 'image_url', image_url: { url: img } });
}
return { role: m.role, content: parts };
}
return m;
});
}
let reply = await this.aiGateway.chat(model, aiMessages, options);
if (typeof reply !== 'string') {
reply = '抱歉,AI 返回了无效的回复,请重试。';
}
File diff suppressed because one or more lines are too long
+5 -5
View File
@@ -28,13 +28,13 @@
| **Mobile 全站 18 页** | 批量替换 hardcoded color → CSS vars | ✅ |
| **PC 响应式** | header/compare/sandbox/member/workshop 移动端修复 | ✅ |
### P3 — 体验增强(当前阶段
### P3 — 体验增强(全部完成 ✅
| 优先级 | 项目 | 说明 | 状态 |
|--------|------|------|------|
| **P3a** | 代码语法高亮 | 沙盒聊天中代码块用 Shiki/prism 渲染 | |
| **P3b** | 会话重命名 | 侧边栏对话标题可编辑 | |
| **P3c** | LLM 集成层增强 | 模型能力检测 / 用量统计 | |
| **P3d** | Mobile 样式复查 | 剩余页面 CSS 变量覆盖率检查 | |
| **P3a** | 代码语法高亮 | highlight.js + CodeBlock 组件,10+ 语言 | |
| **P3b** | 会话重命名 | PATCH endpoint + 双击编辑 | |
| **P3c** | LLM 集成层增强 | 模型能力目录 + 用量统计 + getModelInfo | |
| **P3d** | Mobile 样式复查 | 0 残留硬编码颜色 | |
| — | 多模态(文件上传/图片理解) | 远期规划 | 📋 |
| — | SDK / 共享 UI 包 | 远期规划 | 📋 |
+42 -2
View File
@@ -75,6 +75,9 @@ function SandboxPage() {
const [currentSessionId, setCurrentSessionId] = useState<number | null>(null);
const [renamingId, setRenamingId] = useState<number | null>(null);
const [renameValue, setRenameValue] = useState('');
const [uploadedImages, setUploadedImages] = useState<string[]>([]);
const [uploading, setUploading] = useState(false);
const fileInputRef = useRef<HTMLInputElement>(null);
const messagesEndRef = useRef<HTMLDivElement>(null);
const messagesContainerRef = useRef<HTMLDivElement>(null);
const { isLoggedIn } = useAuth();
@@ -156,7 +159,7 @@ function SandboxPage() {
'Content-Type': 'application/json',
Authorization: `Bearer ${tk}`,
},
body: JSON.stringify({ conversationId, model, messages: apiMessages, temperature, top_p: topP, max_tokens: maxTokens }),
body: JSON.stringify({ conversationId, model, messages: apiMessages, temperature, top_p: topP, max_tokens: maxTokens, ...(uploadedImages.length > 0 ? { images: uploadedImages } : {}) }),
});
const data = await res.json();
if (!res.ok) throw new Error(data.message || '请求失败');
@@ -164,6 +167,7 @@ function SandboxPage() {
if (data.conversationId) setConversationId(data.conversationId);
if (data.sessionId) setCurrentSessionId(data.sessionId);
if (quota) setQuota({ ...quota, used: quota.used + 1, remaining: quota.remaining - 1 });
setUploadedImages([]);
loadSessions(tk);
} else {
await new Promise(r => setTimeout(r, 600));
@@ -515,11 +519,47 @@ function SandboxPage() {
{t.sandbox.dailyQuota.replace('{used}', String(quota.used)).replace('{remaining}', String(quota.remaining))}
</div>
)}
{uploadedImages.length > 0 && (
<div className="flex gap-2 mb-2 flex-wrap">
{uploadedImages.map((img, i) => (
<div key={i} className="relative group">
<img src={img} alt="" className="w-16 h-16 object-cover rounded-lg border border-border" />
<button onClick={() => setUploadedImages(prev => prev.filter((_, j) => j !== i))}
className="absolute -top-1.5 -right-1.5 w-5 h-5 bg-red-500 text-white rounded-full text-xs flex items-center justify-center opacity-0 group-hover:opacity-100 transition-opacity">×</button>
</div>
))}
</div>
)}
<form onSubmit={handleSend} className="flex gap-2">
<input ref={fileInputRef} type="file" accept="image/*" className="hidden"
onChange={async e => {
const file = e.target.files?.[0]
if (!file) return
setUploading(true)
const formData = new FormData()
formData.append('file', file)
try {
const tk = getToken()
const res = await fetch(`${API_BASE}/upload`, {
method: 'POST',
headers: tk ? { Authorization: `Bearer ${tk}` } : {},
body: formData,
})
const data = await res.json()
if (data.url) setUploadedImages(prev => [...prev, data.url])
} catch {}
setUploading(false)
}} />
<button type="button" onClick={() => fileInputRef.current?.click()} disabled={uploading || !isLoggedIn}
className="px-2.5 py-2.5 text-muted-foreground hover:text-foreground border border-input rounded-xl hover:bg-accent disabled:opacity-50">
<svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
<path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M4 16l4.586-4.586a2 2 0 012.828 0L16 16m-2-2l1.586-1.586a2 2 0 012.828 0L20 14m-6-6h.01M6 20h12a2 2 0 002-2V6a2 2 0 00-2-2H6a2 2 0 00-2 2v12a2 2 0 002 2z" />
</svg>
</button>
<input type="text" value={input} onChange={e => setInput(e.target.value)}
placeholder={t.sandbox.placeholder} disabled={sending}
className="flex-1 px-4 py-2.5 bg-background border border-input rounded-xl text-sm focus:outline-none focus:ring-2 focus:ring-ring disabled:opacity-50" />
<button type="submit" disabled={sending || !input.trim()}
<button type="submit" disabled={sending || (!input.trim() && uploadedImages.length === 0)}
className="px-5 py-2.5 bg-brand-600 text-white text-sm font-medium rounded-xl hover:bg-brand-700 disabled:opacity-50">
{sending ? t.sandbox.sending : t.sandbox.send}
</button>