Feat: Add Gemini moderation for new posts

This commit is contained in:
2025-10-02 15:02:32 -07:00
parent 02f94fb09d
commit 4f0e84e9aa

View File

@@ -46,6 +46,10 @@ export async function onRequestPost({request,env}){
const{sub,title,link,content}=body; const{sub,title,link,content}=body;
if(!sub||!title)return json({error:'Missing fields'},{status:400},request); if(!sub||!title)return json({error:'Missing fields'},{status:400},request);
const mod=await fetch(`https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent?key=${env.GOOGLE_KEY}`,{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({contents:[{parts:[{text:`Is this appropriate for a public forum? Respond ONLY "yes" or "no".\n\n${title}\n\n${content||''}`}]}]})});
if(!mod.ok){const err=await mod.text();return json({error:{message:`Moderation failed: ${err}`}},{status:500},request)}
if((await mod.json()).candidates?.[0]?.content.parts[0].text.trim().toLowerCase()!=='yes')return json({error:{message:'Post rejected by Gemini 2.5 Flash Lite.'}},{status:400},request);
let sub_row=await env.D1_SPCHCAP.prepare('SELECT id FROM subs WHERE name=?').bind(sub).first(); let sub_row=await env.D1_SPCHCAP.prepare('SELECT id FROM subs WHERE name=?').bind(sub).first();
if(!sub_row){ if(!sub_row){
await env.D1_SPCHCAP.prepare('INSERT INTO subs(name)VALUES(?)').bind(sub).run(); await env.D1_SPCHCAP.prepare('INSERT INTO subs(name)VALUES(?)').bind(sub).run();