Use cron to generate rule, store in KV

This commit is contained in:
Gabriel Simmer 2024-08-10 14:24:55 +01:00
parent b107181752
commit 64243f517e
Signed by: arch
SSH key fingerprint: SHA256:m3OEcdtrnBpMX+2BDGh/byv3hrCekCLzDYMdvGEKPPQ
3 changed files with 29 additions and 15 deletions

View file

@ -1,20 +1,33 @@
import index from "./index.html";
async function generateRule(env) {
const simple = {
prompt: 'Give me a rule for a Discord server that is surreal, bizzare, and outlandish while still being somewhat realistic on first glance. Keep it short and succinct. Do not add any additional information, greetings, or quotation marks.'
};
return await env.AI.run('@cf/meta/llama-3.1-8b-instruct', simple);
}
export default {
async scheduled(event, env, ctx) {
const rule = await generateRule(env);
console.log(`generated rule ${JSON.stringify(rule)}`);
ctx.waitUntil(env.LOG.put("LATEST", JSON.stringify(rule)));
ctx.waitUntil(env.LOG.put(Date.now(), JSON.stringify(rule), { expirationTtl: 31_536_000 }));
},
async fetch(request, env, ctx) {
const cacheUrl = new URL(request.url);
const url = new URL(request.url);
const { pathname } = url;
// Construct the cache key from the cache URL
const cacheKey = new Request(cacheUrl.toString(), request);
const cacheKey = new Request(url.toString(), request);
const cache = caches.default;
let response = await cache.match(cacheKey);
if (response) {
return response;
}
const url = new URL(request.url);
const { pathname } = url;
switch (pathname) {
case "/":
return new Response(index, {
@ -22,17 +35,11 @@ export default {
});
case "/rule.json":
const simple = {
prompt: 'Give me a rule for a Discord server that is surreal, bizzare, and outlandish while still being somewhat realistic on first glance. Keep it short and succinct. Do not add any additional information, greetings, or quotation marks.'
};
const reply = await env.AI.run('@cf/meta/llama-3.1-8b-instruct', simple);
let response = new Response(JSON.stringify(reply), {
const rule = await env.LOG.get("LATEST");
const response = new Response(rule, {
headers: { "Content-Type": "application/json",
"Cache-Control": "s-maxage=43200" }
"Cache-Control": "s-maxage=3600"}
});
ctx.waitUntil(cache.put(cacheKey, response.clone()));
return response;
default:

View file

@ -1,4 +1,4 @@
{
"name": "thirdrule",
"version": "1.0.0",
"version": "1.0.0"
}

View file

@ -5,3 +5,10 @@ account_id = "7dc420732ea679a530aee304ea49a63c"
[ai]
binding = "AI"
[triggers]
crons = ["0 0 * * *"]
[[kv_namespaces]]
binding = "LOG"
id = "b92ed82d29074a8eb892419703cfdfa5"