@lota-sdk/shared 0.4.5 → 0.4.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/shared",
-  "version": "0.4.5",
+  "version": "0.4.7",
   "type": "module",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/schemas/tools.ts

@@ -4,6 +4,7 @@ import { AgentPlanDraftSchema, AgentPlanEdgeDraftSchema } from './agent-plan-dra
 import { baseChatMessageSchema } from './chat-api'
 import type { AnyChatMessage } from './chat-message'
 import {
+  PlanArtifactSubmissionSchema,
   PlanEdgeSpecSchema,
   PlanNodeResultSubmissionSchema,
   PlanNodeRunStatusSchema,
@@ -67,9 +68,9 @@ export const SubmitExecutionNodeResultArgsSchema = PlanNodeResultSubmissionSchem
 export const SubmitPlanTurnResultArgsSchema = PlanNodeResultSubmissionSchema
 
 export const ExecutionPlanActionSchema = z
-  .enum(['create', 'create-project', 'replace', 'resume'])
+  .enum(['create', 'create-project', 'replace', 'resume', 'update-node'])
   .describe(
-    'create: inline plan (1-2 nodes). create-project: plan in a dedicated project thread (3+ nodes). replace: replace active plan. resume: resume interrupted plan.',
+    'create: inline plan (1-2 nodes). create-project: plan in a dedicated project thread (3+ nodes). replace: replace active plan. resume: resume interrupted plan. update-node: submit result for a specific plan node.',
   )
 export type ExecutionPlanAction = z.infer<typeof ExecutionPlanActionSchema>
 
@@ -117,6 +118,22 @@ export const ExecutionPlanResumeArgsSchema = z
   .object({ action: z.literal('resume'), runId: runIdSchema.describe('Required for replace and resume.') })
   .strict()
 
+const ExecutionPlanNodeUpdateSchema = z
+  .object({
+    id: z.string().trim().min(1).describe('The node ID to update.'),
+    latestNotes: z.string().trim().min(1).max(4000).describe('Summary of what was accomplished.'),
+    deliverables: z.array(PlanArtifactSubmissionSchema).optional().describe('Artifacts produced by the node.'),
+  })
+  .strict()
+
+export const ExecutionPlanUpdateNodeArgsSchema = z
+  .object({
+    action: z.literal('update-node'),
+    runId: runIdSchema.describe('The plan run containing the node.'),
+    node: ExecutionPlanNodeUpdateSchema,
+  })
+  .strict()
+
 const ExecutionPlanBaseArgsSchema = z
   .object({
     action: ExecutionPlanActionSchema,
@@ -127,8 +144,9 @@ const ExecutionPlanBaseArgsSchema = z
     projectTitle: projectTitleSchema.optional(),
     targetThreadId: targetThreadIdSchema.optional(),
     requireApproval: requireApprovalSchema,
-    runId: runIdSchema.describe('Required for replace and resume.').optional(),
+    runId: runIdSchema.describe('Required for replace, resume, and update-node.').optional(),
     reason: replaceReasonSchema.describe('Required for replace.').optional(),
+    node: ExecutionPlanNodeUpdateSchema.optional().describe('Required for update-node.'),
   })
   .strict()
 
@@ -221,6 +239,43 @@ export const ExecutionPlanArgsSchema = ExecutionPlanBaseArgsSchema.superRefine((
       }
     }
   }
+
+  if (value.action === 'update-node') {
+    if (value.runId === undefined) {
+      addMissingExecutionPlanFieldIssue(ctx, 'runId', 'runId is required for update-node.')
+    }
+    if (value.node === undefined) {
+      addMissingExecutionPlanFieldIssue(ctx, 'node', 'node is required for update-node.')
+    }
+    for (const path of [
+      'title',
+      'objective',
+      'nodes',
+      'edges',
+      'projectTitle',
+      'targetThreadId',
+      'reason',
+    ] as const) {
+      if (value[path] !== undefined) {
+        addUnexpectedExecutionPlanFieldIssue(
+          ctx,
+          path,
+          `${path} is not allowed when action is "update-node". Use only action, runId, and node.`,
+        )
+      }
+    }
+    if (value.requireApproval !== undefined) {
+      addUnexpectedExecutionPlanFieldIssue(
+        ctx,
+        'requireApproval',
+        'requireApproval is not allowed when action is "update-node".',
+      )
+    }
+  }
+
+  if (value.action !== 'update-node' && value.node !== undefined) {
+    addUnexpectedExecutionPlanFieldIssue(ctx, 'node', 'node is only allowed when action is "update-node".')
+  }
 })
 
 export type ExecutionPlanArgs =
@@ -228,6 +283,7 @@ export type ExecutionPlanArgs =
   | z.infer<typeof ExecutionPlanCreateProjectArgsSchema>
   | z.infer<typeof ExecutionPlanReplaceArgsSchema>
   | z.infer<typeof ExecutionPlanResumeArgsSchema>
+  | z.infer<typeof ExecutionPlanUpdateNodeArgsSchema>
 
 export const ExecutionPlanQueryArgsSchema = z
   .object({

package/src/utils/assistant-text.ts

@@ -103,10 +103,20 @@ function stripRawToolPayloadObjects(value: string): string {
   return result
 }
 
+/** Strip bracketed plan/node status labels the LLM sometimes appends to titles (e.g. "Plan Title [blocked]") */
+const BRACKETED_STATUS_PATTERN = /\s*\[(blocked|failed|running|stalled|inactive|paused|cancelled|errored)\]/gi
+
+/** Strip raw execution plan run/node IDs the LLM sometimes echoes (e.g. "Run: 019d6382-...") */
+const RAW_RUN_ID_PATTERN = /\bRun:\s*[\da-f]{8}-[\da-f]{4}-[\da-f]{4}-[\da-f]{4}-[\da-f]{12}\b/gi
+const RAW_PLAN_RUN_ID_PATTERN = /\bplanRun:[\w⟨⟩-]+/g
+
 export function sanitizeAssistantVisibleText(value: string): string {
   let next = value
 
   next = next.replace(/\[REDACTED\]/gi, '')
+  next = next.replace(BRACKETED_STATUS_PATTERN, '')
+  next = next.replace(RAW_RUN_ID_PATTERN, '')
+  next = next.replace(RAW_PLAN_RUN_ID_PATTERN, '')
 
   for (const pattern of INTERNAL_BLOCK_PATTERNS) {
     next = next.replace(pattern, '')