@neutrome/lilsdk 0.6.4 → 0.6.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neutrome/lilsdk",
-  "version": "0.6.4",
+  "version": "0.6.5",
   "type": "module",
   "exports": {
     ".": "./src/index.ts",

package/src/managed/collapsed-reasoner.ts

@@ -25,6 +25,8 @@ const INITIAL_PROMPT =
 
 /** Options accepted by {@link createCollapsedReasoner}. */
 export type CollapsedReasonerOptions = {
+  /** Prompt used to initialize the reasoner's state. */
+  initialPrompt?: string;
   /** Prompt used for periodic progress updates. */
   prompt?: string;
   /** Prompt used when an update must be attempted. */
@@ -87,7 +89,7 @@ export function createCollapsedReasoner(
         wake?.();
         wakePromise = nextWake();
       };
-      const initial = summarizeInitial(ctx, summarizer, request).then(
+      const initial = summarizeInitial(ctx, summarizer, request, settings.initialPrompt).then(
         (summary) => {
           if (summary && acceptsSummaries && phase === 0) ready.push(summary);
           notify();
@@ -218,6 +220,7 @@ function parseOptions(options: CollapsedReasonerOptions) {
   return {
     interval,
     forcedInterval,
+    initialPrompt: options.initialPrompt ?? INITIAL_PROMPT,
     prompt: options.prompt ?? DEFAULT_PROMPT,
     forcedPrompt: options.forcedPrompt ?? DEFAULT_FORCED_PROMPT,
   };
@@ -247,12 +250,13 @@ async function summarizeInitial(
   ctx: ExecutorContext,
   summarizer: ExecutorInput,
   request: Program,
+  prompt: string,
 ): Promise<string> {
-  const prompt = new ProgramView(request).messages
+  const content = new ProgramView(request).messages
     .filter((message) => message.role === "user")
     .at(-1)?.text.trim();
-  if (!prompt) return "";
-  return summarize(ctx, summarizer, INITIAL_PROMPT, prompt);
+  if (!content) return "";
+  return summarize(ctx, summarizer, prompt, content);
 }
 
 function withoutThinking(program: Program): Program {