gongxiangp2/yedir/temp/gemini_code.js
chenpei d0dc94e36d feat: 将弹幕功能替换为AI聊天助手
新功能:
- 创建AI聊天助手组件(AIChatWidget)
- 支持文字+图片多模态对话
- 集成Gemini AI API
- 支持图片编辑和内容生成
- 实时对话历史管理
- 优雅的UI设计,支持黑夜模式

技术实现:
- 后端API代理Gemini API(/api/ai-chat)
- 支持最多5张图片上传
- 流式响应处理
- 对话历史管理
- Base64图片处理

配置:
- 在.env中添加Gemini API配置
- GEMINI_API_KEY
- GEMINI_API_BASE_URL
- GEMINI_MODEL

替换原弹幕功能,提供更强大的AI助手体验!
2025-10-27 17:45:37 +08:00

25 lines
767 B
JavaScript
Executable File

if (model === 'gemini-2.0-flash-exp-image-generation') {
apiKey = process.env.GEMINI_FLASH_API_KEY;
if (!apiKey) throw new Error('GEMINI_FLASH_API_KEY environment variable not set.');
apiUrl = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent';
// Construct Gemini request body
const imageParts = imageFiles.map(file => ({
inline_data: {
mime_type: file.mimetype,
data: file.buffer.toString('base64')
}
}));
requestBody = {
contents: [
{
parts: [
{ text: prompt },
...imageParts
]
}
]
};
console.log('Using Gemini Flash API Key');
}