@axiom-lattice/core 3.0.7 → 3.1.1

package/dist/index.mjs

@@ -2646,6 +2646,7 @@ var InMemoryTaskStore = class {
       projectId: params.projectId,
       dueDate: params.dueDate,
       metadata: params.metadata,
+      files: params.files,
       parentId: params.parentId,
       sourceId: params.sourceId,
       context: params.context,
@@ -2703,6 +2704,7 @@ var InMemoryTaskStore = class {
     const updated = {
       ...existing,
       ...updates,
+      files: updates.files !== void 0 ? updates.files : existing.files,
       updatedAt: /* @__PURE__ */ new Date()
     };
     tenantTasks.set(id, updated);
@@ -14207,6 +14209,11 @@ var manageTaskSchema = z41.object({
   dependencies: z41.array(z41.string()).optional().describe("List of task IDs that must be completed before this task can start"),
   result: z41.string().optional().describe("Result summary when task is completed"),
   failureReason: z41.string().optional().describe("Reason for failure (use when status='failed')"),
+  files: z41.array(z41.object({
+    uri: z41.string().describe("Uniquely locates the resource: http(s):// URL, /s/:token share, or sandbox path"),
+    name: z41.string().optional().describe("Display name"),
+    addedBy: z41.enum(["user", "agent"]).optional().describe("Who attached the file")
+  })).optional().describe("File references attached to this task"),
   summary: z41.string().optional().describe("Brief summary of the operation")
 });
 function buildReviewMarkdown(task) {
@@ -14265,7 +14272,8 @@ function createTaskMiddleware() {
           requireReview: input.requireReview,
           dependencies: input.dependencies,
           workspaceId,
-          projectId
+          projectId,
+          files: input.files
         });
         return JSON.stringify({ success: true, data: task });
       }
@@ -14344,7 +14352,8 @@ function createTaskMiddleware() {
           "result",
           "failureReason",
           "requireReview",
-          "dependencies"
+          "dependencies",
+          "files"
         ];
         for (const field of settableFields) {
           if (input[field] !== void 0) {
@@ -18182,11 +18191,11 @@ ${BASE_PROMPT}` : BASE_PROMPT;
       defaultInterruptOn: interruptOn,
       subagents,
       generalPurposeAgent: true
-    }),
-    // Enables Anthropic prompt caching for improved performance and reduced costs
-    anthropicPromptCachingMiddleware({
-      unsupportedModelBehavior: "ignore"
     })
+    // Enables Anthropic prompt caching for improved performance and reduced costs
+    // anthropicPromptCachingMiddleware({
+    //   unsupportedModelBehavior: "ignore",
+    // })
   ];
   if (interruptOn) {
     middleware.push(humanInTheLoopMiddleware2({ interruptOn }));
@@ -21789,7 +21798,7 @@ registerToolLattice(
   "update_agent",
   {
     name: "update_agent",
-    description: "Update an existing agent's configuration. Provide the agent ID and the fields to change. Returns the updated agent's ID and name. NOTE: For user approval/confirmation scenarios, add middleware with type: 'ask_user_to_clarify'.",
+    description: "Update an existing agent's configuration. Provide the agent ID and the fields to change. Returns the updated agent's ID and name. NOTE: For user approval/confirmation scenarios, add middleware with type: 'ask_user_to_clarify'. IMPORTANT: do NOT set 'modelKey' unless the user explicitly specified a model \u2014 leaving it unset makes the runtime use the 'default' model.",
     schema: updateAgentSchema
   },
   async (input, exeConfig) => {
@@ -21864,6 +21873,27 @@ registerToolLattice(
     }
   }
 );
+registerToolLattice(
+  "list_models",
+  {
+    name: "list_models",
+    description: "List all registered models. Returns each model's key (use this string as modelKey when creating eval projects via manage_eval create_project, or as an agent's modelKey in create_agent/update_agent) and its display name.",
+    schema: z49.object({})
+  },
+  async (_input) => {
+    try {
+      const lattices = modelLatticeManager.getAllLattices();
+      return JSON.stringify({
+        models: lattices.map((l) => ({
+          key: l.key,
+          name: l.client.name ?? l.key
+        }))
+      });
+    } catch (error) {
+      return JSON.stringify({ error: `Failed to list models: ${error.message}` });
+    }
+  }
+);
 registerToolLattice(
   "invoke_agent",
   {
@@ -21995,6 +22025,24 @@ The skills document WHY and HOW; these gates are the unskippable
 minimum. If you cannot satisfy a gate (e.g. user says skip), record it
 and proceed only on the user's explicit instruction.
 
+LEARNING ROUND KICKOFF \u2014 a message that names an existing target agent
+id AND an existing tracking task id (a "learning round"). This protocol
+OVERRIDES the defaults above:
+- The target agent ALREADY EXISTS (an empty placeholder). Build and
+  refine it via update_agent on that exact id. NEVER call create_agent \u2014
+  a new agent would disconnect the round's tracking.
+- The parent task ALREADY EXISTS \u2014 your create-a-task-first duty is
+  satisfied by it. Create subtasks with manage_task under its id
+  (parentId); NEVER create a new parent task for the round.
+- The round is pre-approved \u2014 skip the DESIGN\u2192CONFIRM gates: show the
+  design in your reply, then build directly.
+- Do NOT set modelKey in update_agent unless the user explicitly named a
+  model \u2014 leaving it unset makes the runtime use the 'default' model.
+- Every update_agent call must be mirrored by a manage_task work item
+  whose summary names the changed keys (e.g. "update_agent: prompt,
+  modelKey") \u2014 the round feed highlights these so the user can see what
+  changed between iterations.
+
 Your sub-skills (accessible via the MOC or direct loading):
 - [[learn-capability]] \u2014 Learn from any source material \u2192 skills + agents
 - [[agent-build]] \u2014 Design and build single agents (REACT/DEEP_AGENT)
@@ -22109,6 +22157,7 @@ var agentArchitectConfig = {
   tools: [
     "list_agents",
     "list_tools",
+    "list_models",
     "list_middleware_types",
     "list_connections",
     "get_agent",
@@ -26578,14 +26627,22 @@ var DEFAULT_CALIBRATION_PROBES = [
     expectedPass: false
   }
 ];
+function resolveJudgeModelKey(explicitKey, manager) {
+  if (explicitKey) return explicitKey;
+  const models = manager.getAllLattices();
+  const defaultModel = models.find((m) => m.key === "default");
+  if (defaultModel) return defaultModel.key;
+  const first = models[0];
+  if (!first) {
+    throw new Error("No model registered \u2014 register a model or provide a judgeModelKey");
+  }
+  return first.key;
+}
 var LatticeEvalProject = class {
   constructor(project, onCaseComplete) {
     this.suites = /* @__PURE__ */ new Map();
     this.project = project;
-    const judgeModelKey = this.project.judge_agent_config.modelKey;
-    if (!judgeModelKey) {
-      throw new Error("judge_agent_config.modelKey is required \u2014 only pre-registered models are allowed");
-    }
+    const judgeModelKey = resolveJudgeModelKey(this.project.judge_agent_config.modelKey, modelLatticeManager);
     if (!modelLatticeManager.hasLattice(judgeModelKey)) {
       throw new Error(
         `Judge model "${judgeModelKey}" not found. Register it first or provide a valid modelKey.`
@@ -28987,6 +29044,7 @@ function createManageEvalTool() {
     description: z66.string().optional(),
     judgeModelKey: z66.string().optional(),
     concurrency: z66.number().optional(),
+    targetAgentId: z66.string().optional().describe("Optional for create_project \u2014 the agent this eval project verifies. Recorded in targetServerConfig so the agent's detail page can find its eval data without relying on project naming."),
     suiteId: z66.string().optional(),
     caseId: z66.string().optional(),
     inputMessage: z66.string().optional(),
@@ -29015,10 +29073,16 @@ function createManageEvalTool() {
             data = await store.createProject(tid, uuidv48(), {
               name: input.name,
               description: input.description,
-              judgeModelConfig: { modelKey: input.judgeModelKey },
+              // No judge key given → default to the 'default' model key (the same
+              // convention the UI model selector uses); never an empty modelKey.
+              judgeModelConfig: { modelKey: input.judgeModelKey ?? "default" },
               targetServerConfig: {
                 workspace_id: ctx.workspaceId,
-                project_id: ctx.projectId
+                project_id: ctx.projectId,
+                // The project↔agent association lives here (not in the project
+                // name): an eval project is agent-agnostic by design, this
+                // pointer is merely the agent's designated verifier.
+                ...input.targetAgentId ? { targetAgentId: input.targetAgentId } : {}
               },
               concurrency: input.concurrency ?? 3
             });
@@ -29085,8 +29149,8 @@ function createManageEvalTool() {
       name: "manage_eval",
       description: `Create, update, delete evaluation projects, suites, and test cases.
  
-Project: create_project(name, description?, judgeModelKey?, concurrency?) | update_project | delete_project
-  judgeModelKey defaults to first available model. concurrency defaults to 3.
+Project: create_project(name, description?, judgeModelKey?, concurrency?, targetAgentId?) | update_project | delete_project
+  judgeModelKey defaults to the "default" model. concurrency defaults to 3.
   delete_project rejected if active runs exist.
   **When creating a project from within a workspace, the workspace/project context is
   automatically bound \u2014 eval runs will execute in the same workspace.**
@@ -29109,6 +29173,7 @@ function createRunEvalTool() {
     suiteIds: z66.array(z66.string()).optional().describe("Optional for start \u2014 only run these suites (e.g. dev set only). Omit to run all."),
     caseIds: z66.array(z66.string()).optional().describe("Optional for start \u2014 only run these cases across the selected suites. Omit to run all cases in those suites."),
     runId: z66.string().optional().describe("Required for status, resume, abort"),
+    taskId: z66.string().optional().describe("Optional for start \u2014 training task ID this run belongs to (round association)"),
     sleepMs: z66.number().int().min(0).max(12e4).optional().describe("Optional for status \u2014 sleep this many ms BEFORE checking the run, to pace polling (e.g. 15000 \u2192 30000 \u2192 60000 \u2192 120000). Omit to check immediately."),
     wait: z66.boolean().optional().describe("Optional for start \u2014 defaults to true: block synchronously (up to ~150s) and return final results in one call. Set false to return the runId immediately and poll.")
   });
@@ -29126,7 +29191,7 @@ function createRunEvalTool() {
           switch (input.action) {
             case "start": {
               const ctx = workspaceContext(exeConfig);
-              const runId = await svc.startRun(tid, input.projectId, input.suiteIds, input.caseIds, ctx);
+              const runId = await svc.startRun(tid, input.projectId, input.suiteIds, input.caseIds, ctx, input.taskId);
               if (input.wait === false) {
                 data = sanitize({ runId, message: "Run started. Poll with run_eval status (backoff: 15s\u219230s\u219260s\u2192max 120s)." });
                 break;
@@ -31363,6 +31428,7 @@ export {
   registerWorkflowRun,
   removeCollectionVectorStore,
   renderTemplate,
+  resolveJudgeModelKey,
   resolvePath,
   sandboxLatticeManager,
   sanitizeToolCallId,