Module: CompletionKit::McpTools::Calibrations

Extended by:
Base
Defined in:
app/services/completion_kit/mcp_tools/calibrations.rb

Constant Summary collapse

TOOLS = 
{
  "calibrations_list" => {
    description: "List calibrations. Filter by run_id, response_id, metric_id, or created_by.",
    inputSchema: {
      type: "object",
      properties: {
        run_id: {type: "integer"},
        response_id: {type: "integer"},
        metric_id: {type: "integer"},
        created_by: {type: "string"}
      },
      required: []
    },
    handler: :list
  },
  "calibrations_create" => {
    description: "Upsert a calibration for (run, response, metric, created_by). Verdict is one of agree, disagree, borderline. corrected_score (1..5) is required when verdict is 'disagree'.",
    inputSchema: {
      type: "object",
      properties: {
        run_id: {type: "integer"},
        response_id: {type: "integer"},
        metric_id: {type: "integer"},
        verdict: {type: "string", enum: %w[agree disagree borderline]},
        corrected_score: {type: "number"},
        note: {type: "string"},
        created_by: {type: "string"}
      },
      required: ["run_id", "response_id", "metric_id", "verdict"]
    },
    handler: :create
  }
}.freeze

Class Method Summary collapse

Methods included from Base

call, definitions, error_result, text_result

Class Method Details

.create(args) ⇒ Object 



49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
# File 'app/services/completion_kit/mcp_tools/calibrations.rb', line 49

def self.create(args)
  run = CompletionKit::Run.find(args["run_id"])
  response = run.responses.find(args["response_id"])
  metric = CompletionKit::Metric.find(args["metric_id"])
  created_by = args["created_by"].presence || "mcp"

  calibration = CompletionKit::Calibration.find_or_initialize_by(
    run_id: run.id, response_id: response.id, metric_id: metric.id, created_by: created_by
  )
  calibration.assign_attributes(
    judge_version: CompletionKit::JudgeVersion.ensure_current_for(metric),
    verdict: args["verdict"],
    corrected_score: args["corrected_score"],
    note: args["note"]
  )

  if calibration.save
    text_result(calibration.as_json)
  else
    error_result(calibration.errors.full_messages.join(", "))
  end
end

.list(args) ⇒ Object 



40
41
42
43
44
45
46
47
 
# File 'app/services/completion_kit/mcp_tools/calibrations.rb', line 40

def self.list(args)
  scope = CompletionKit::Calibration.all
  scope = scope.where(run_id: args["run_id"]) if args["run_id"]
  scope = scope.where(response_id: args["response_id"]) if args["response_id"]
  scope = scope.where(metric_id: args["metric_id"]) if args["metric_id"]
  scope = scope.where(created_by: args["created_by"]) if args["created_by"]
  text_result(scope.order(:created_at).map(&:as_json))
end