kairn-cli 2.2.1 → 2.2.2

package/dist/cli.js

@@ -1237,22 +1237,29 @@ function classifyError(err, provider) {
 }
 async function callLLM(config, userMessage, options) {
   const maxTokens = options.maxTokens ?? 8192;
-  const systemPrompt = options.systemPrompt;
+  const { systemPrompt } = options;
+  const jsonMode = options.jsonMode ?? false;
   const providerName = getProviderName(config.provider);
   if (config.provider === "anthropic") {
     const client2 = new Anthropic2({ apiKey: config.api_key });
+    const messages = [
+      { role: "user", content: userMessage }
+    ];
+    if (jsonMode) {
+      messages.push({ role: "assistant", content: "{" });
+    }
     try {
       const response = await client2.messages.create({
         model: config.model,
         max_tokens: maxTokens,
         system: systemPrompt,
-        messages: [{ role: "user", content: userMessage }]
+        messages
       });
       const textBlock = response.content.find((block) => block.type === "text");
       if (!textBlock || textBlock.type !== "text") {
         throw new Error("No text response from compiler LLM");
       }
-      return textBlock.text;
+      return jsonMode ? `{${textBlock.text}` : textBlock.text;
     } catch (err) {
       throw new Error(classifyError(err, providerName));
     }
@@ -1268,7 +1275,8 @@ async function callLLM(config, userMessage, options) {
       messages: [
         { role: "system", content: systemPrompt },
         { role: "user", content: userMessage }
-      ]
+      ],
+      ...jsonMode ? { response_format: { type: "json_object" } } : {}
     });
     const text = response.choices[0]?.message?.content;
     if (!text) {
@@ -4785,7 +4793,18 @@ function parseProposerResponse(raw) {
   try {
     parsed = JSON.parse(cleaned);
   } catch {
-    throw new Error(`Proposer returned invalid JSON: ${cleaned.slice(0, 200)}`);
+    const firstBrace = cleaned.indexOf("{");
+    const lastBrace = cleaned.lastIndexOf("}");
+    if (firstBrace !== -1 && lastBrace > firstBrace) {
+      const extracted = cleaned.slice(firstBrace, lastBrace + 1);
+      try {
+        parsed = JSON.parse(extracted);
+      } catch {
+        throw new Error(`Proposer returned invalid JSON: ${cleaned.slice(0, 200)}`);
+      }
+    } else {
+      throw new Error(`Proposer returned invalid JSON: ${cleaned.slice(0, 200)}`);
+    }
   }
   if (typeof parsed !== "object" || parsed === null) {
     throw new Error("Proposer response is not a JSON object");
@@ -4848,7 +4867,8 @@ async function propose(iteration, workspacePath, harnessPath, history, tasks, co
   const proposerConfig = { ...config, model: proposerModel };
   const response = await callLLM(proposerConfig, userMessage, {
     systemPrompt: PROPOSER_SYSTEM_PROMPT,
-    maxTokens: 8192
+    maxTokens: 8192,
+    jsonMode: true
   });
   return parseProposerResponse(response);
 }