Refactor (8)

index.ts

@@ -95,11 +95,12 @@ type Job =
     };
 
 async function processPostJob() {
-  await using postJobSession = new LlmSession(model, postJobPrompt);
-  await postJobSession.init();
   const notes = await getNotes(10, 0, 5);
   const input = notes.map(formatNote).join("\n");
-  const text = parseResponse(
+  const text = await (async () => {
+    await using postJobSession = new LlmSession(model, postJobPrompt);
+    await postJobSession.init();
+    return parseResponse(
       grammar,
       await postJobSession.prompt(input, {
         ...baseChatPromptOptions,
@@ -111,6 +112,7 @@ async function processPostJob() {
         },
       }),
     );
+  })();
   if (text) {
     const rephrased = await rephrase(text);
     if (values.test) return;
@@ -129,9 +131,10 @@ async function processReplyJob(job: Extract<Job, { type: "reply" }>) {
       text: formatNote(n),
     } as ChatHistoryItem;
   });
+  const text = await (async () => {
     await using session = new LlmSession(model, replyJobPrompt, history);
     await session.init();
-  const text = parseResponse(
+    return parseResponse(
       grammar,
       await session.prompt(formatNote(job.last), {
         ...baseChatPromptOptions,
@@ -143,6 +146,8 @@ async function processReplyJob(job: Extract<Job, { type: "reply" }>) {
         },
       }),
     );
+  })();
+
   if (text) {
     const rephrased = await rephrase(text);
     if (values.test) return;
@@ -268,6 +273,8 @@ async function test() {
     await processJob({ type: "post" });
     await processJob({ type: "post" });
     await processJob({ type: "post" });
+    await processJob({ type: "post" });
+    await processJob({ type: "post" });
   } catch (e) {
     console.error(e);
     if (e instanceof Error) console.log(e.stack);