@lota-sdk/core 0.4.0 → 0.4.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/core",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "type": "module",
   "main": "./src/index.ts",
   "types": "./src/index.ts",
@@ -32,7 +32,7 @@
     "@chat-adapter/slack": "^4.23.0",
     "@chat-adapter/state-ioredis": "^4.23.0",
     "@logtape/logtape": "^2.0.5",
-    "@lota-sdk/shared": "0.4.0",
+    "@lota-sdk/shared": "0.4.1",
     "@mendable/firecrawl-js": "^4.18.1",
     "@surrealdb/node": "^3.0.3",
     "ai": "^6.0.145",

package/src/tools/execution-plan.tool.ts

@@ -1,11 +1,16 @@
 import {
   ExecutionPlanArgsSchema,
+  ExecutionPlanCreateArgsSchema,
+  ExecutionPlanCreateProjectArgsSchema,
   ExecutionPlanQueryArgsSchema,
+  ExecutionPlanReplaceArgsSchema,
+  ExecutionPlanResumeArgsSchema,
   SubmitExecutionNodeResultArgsSchema,
+  AgentPlanDraftSchema,
   expandAgentPlanDraft,
   getLatestExecutionPlanResult,
 } from '@lota-sdk/shared'
-import type { CreateProjectWithPlanResultData } from '@lota-sdk/shared'
+import type { CreateProjectWithPlanResultData, ExecutionPlanArgs } from '@lota-sdk/shared'
 import { tool } from 'ai'
 
 import type { RecordIdRef } from '../db/record-id'
@@ -44,17 +49,16 @@ export function createExecutionPlanTool(params: {
       'Manage execution plans. Actions: create (inline, 1-2 nodes), create-project (dedicated project thread, 3+ nodes), replace (swap active plan), resume (resume interrupted plan).',
     inputSchema: ExecutionPlanArgsSchema,
     execute: async (input) => {
-      const parsed = ExecutionPlanArgsSchema.parse(input)
+      const parsed = parseExecutionPlanArgs(input)
       let result: unknown
 
       switch (parsed.action) {
         case 'create': {
-          const { action: _action, targetThreadId, ...draftInput } = parsed
-          const draft = expandAgentPlanDraft(draftInput)
+          const draft = extractAgentPlanDraft(parsed)
           params.validateInlinePlan?.(draft)
           result = await resolvedEpService.createPlan({
             organizationId: params.orgId,
-            threadId: targetThreadId ?? params.threadId,
+            threadId: parsed.targetThreadId ?? params.threadId,
             leadAgentId: params.agentId,
             input: draft,
           })
@@ -62,19 +66,18 @@ export function createExecutionPlanTool(params: {
         }
 
         case 'create-project': {
-          const { action: _action, projectTitle, targetThreadId, ...draftInput } = parsed
-          const draft = expandAgentPlanDraft(draftInput)
-          const targetThread = targetThreadId
-            ? await resolvedWsService.getThread(targetThreadId)
+          const draft = extractAgentPlanDraft(parsed)
+          const targetThread = parsed.targetThreadId
+            ? await resolvedWsService.getThread(parsed.targetThreadId)
             : await (() => {
-                if (!projectTitle) {
+                if (!parsed.projectTitle) {
                   throw new Error('projectTitle is required when action is "create-project".')
                 }
 
                 return resolvedWsService.createThread({
                   userId: params.userId,
                   organizationId: params.orgId,
-                  title: projectTitle,
+                  title: parsed.projectTitle,
                   type: 'group',
                 })
               })()
@@ -93,7 +96,7 @@ export function createExecutionPlanTool(params: {
             )
           }
 
-          const createdThread = !targetThreadId
+          const createdThread = !parsed.targetThreadId
           try {
             const created = await resolvedEpService.createPlan({
               organizationId: params.orgId,
@@ -117,13 +120,12 @@ export function createExecutionPlanTool(params: {
         }
 
         case 'replace': {
-          const { action: _action, runId, reason, ...draftInput } = parsed
-          const draft = expandAgentPlanDraft(draftInput)
+          const draft = extractAgentPlanDraft(parsed)
           result = await resolvedEpService.replacePlan({
             organizationId: params.orgId,
             threadId: params.threadId,
             leadAgentId: params.agentId,
-            input: { runId, reason, ...draft },
+            input: { runId: parsed.runId, reason: parsed.reason, ...draft },
           })
           break
         }
@@ -143,6 +145,32 @@ export function createExecutionPlanTool(params: {
   })
 }
 
+function parseExecutionPlanArgs(input: unknown): ExecutionPlanArgs {
+  const parsed = ExecutionPlanArgsSchema.parse(input)
+
+  switch (parsed.action) {
+    case 'create':
+      return ExecutionPlanCreateArgsSchema.parse(parsed)
+    case 'create-project':
+      return ExecutionPlanCreateProjectArgsSchema.parse(parsed)
+    case 'replace':
+      return ExecutionPlanReplaceArgsSchema.parse(parsed)
+    case 'resume':
+      return ExecutionPlanResumeArgsSchema.parse(parsed)
+  }
+}
+
+function extractAgentPlanDraft(input: Extract<ExecutionPlanArgs, { action: 'create' | 'create-project' | 'replace' }>) {
+  return expandAgentPlanDraft(
+    AgentPlanDraftSchema.parse({
+      title: input.title,
+      objective: input.objective,
+      nodes: input.nodes,
+      edges: input.edges,
+    }),
+  )
+}
+
 export function createExecutionPlanQueryTool(params: { threadId: RecordIdRef }) {
   return tool({
     description: