import { tool, type Tool } from "@lmstudio/sdk";
import { z } from "zod";
import { okResult } from "../../core/result";
import type { AgentFinal } from "../runStore";
import type { AgentToolContext } from "./context";

/** `finish_run` — the single completion call. */
export function createFinishRunTool(ctx: AgentToolContext): Tool {
  const { state, completionFlag, beforeTool } = ctx;
  return tool({
    name: "finish_run",
    description:
      "Finish the run with a factual summary, evidence, remaining work, and confidence. Call only after verification or when a blocker is proven.",
    parameters: {
      summary: z.string(),
      evidence: z.array(z.string()),
      remaining: z.array(z.string()),
      confidence: z.number().min(0).max(1),
    },
    implementation: async (input: AgentFinal) => {
      await beforeTool("finish_run");
      state.final = {
        summary: input.summary.slice(0, 4000),
        evidence: input.evidence.slice(0, 30).map((item) => item.slice(0, 1000)),
        remaining: input.remaining.slice(0, 30).map((item) => item.slice(0, 1000)),
        confidence: input.confidence,
      };
      completionFlag.done = true;
      await ctx.store.event(state, {
        type: "finish",
        summary: state.final.summary,
        details: {
          evidence: state.final.evidence,
          remaining: state.final.remaining,
          confidence: state.final.confidence,
        },
      });
      return okResult({
        operation: "agent.finish",
        summary: "Run completion recorded.",
        data: state.final,
        importance: "critical",
        facts: [state.final.summary, ...state.final.evidence],
      });
    },
  });
}

import { tool, type Tool } from "@lmstudio/sdk";
import { z } from "zod";
import { okResult } from "../../core/result";
import type { AgentFinal } from "../runStore";
import type { AgentToolContext } from "./context";

/** `finish_run` — the single completion call. */
export function createFinishRunTool(ctx: AgentToolContext): Tool {
  const { state, completionFlag, beforeTool } = ctx;
  return tool({
    name: "finish_run",
    description:
      "Finish the run with a factual summary, evidence, remaining work, and confidence. Call only after verification or when a blocker is proven.",
    parameters: {
      summary: z.string(),
      evidence: z.array(z.string()),
      remaining: z.array(z.string()),
      confidence: z.number().min(0).max(1),
    },
    implementation: async (input: AgentFinal) => {
      await beforeTool("finish_run");
      state.final = {
        summary: input.summary.slice(0, 4000),
        evidence: input.evidence.slice(0, 30).map((item) => item.slice(0, 1000)),
        remaining: input.remaining.slice(0, 30).map((item) => item.slice(0, 1000)),
        confidence: input.confidence,
      };
      completionFlag.done = true;
      await ctx.store.event(state, {
        type: "finish",
        summary: state.final.summary,
        details: {
          evidence: state.final.evidence,
          remaining: state.final.remaining,
          confidence: state.final.confidence,
        },
      });
      return okResult({
        operation: "agent.finish",
        summary: "Run completion recorded.",
        data: state.final,
        importance: "critical",
        facts: [state.final.summary, ...state.final.evidence],
      });
    },
  });
}