新功能: - 创建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助手体验!
25 lines
767 B
JavaScript
Executable File
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');
|
|
}
|