Module: CompletionKit::McpTools::Criteria

Defined in:
app/services/completion_kit/mcp_tools/criteria.rb

Constant Summary collapse

TOOLS = 
{
  "criteria_list" => {
    description: "List all criteria",
    inputSchema: {type: "object", properties: {}, required: []},
    handler: :list
  },
  "criteria_get" => {
    description: "Get a criteria by ID",
    inputSchema: {type: "object", properties: {id: {type: "integer"}}, required: ["id"]},
    handler: :get
  },
  "criteria_create" => {
    description: "Create a criteria grouping metrics",
    inputSchema: {
      type: "object",
      properties: {
        name: {type: "string"}, description: {type: "string"},
        metric_ids: {type: "array", items: {type: "integer"}}
      },
      required: ["name"]
    },
    handler: :create
  },
  "criteria_update" => {
    description: "Update a criteria",
    inputSchema: {
      type: "object",
      properties: {
        id: {type: "integer"}, name: {type: "string"}, description: {type: "string"},
        metric_ids: {type: "array", items: {type: "integer"}}
      },
      required: ["id"]
    },
    handler: :update
  },
  "criteria_delete" => {
    description: "Delete a criteria",
    inputSchema: {type: "object", properties: {id: {type: "integer"}}, required: ["id"]},
    handler: :delete
  }
}.freeze

Class Method Summary collapse

Class Method Details

.call(name, arguments) ⇒ Object 



50
51
52
53
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 50

def self.call(name, arguments)
  tool = TOOLS.fetch(name)
  send(tool[:handler], arguments)
end

.create(args) ⇒ Object 



63
64
65
66
67
68
69
70
71
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 63

def self.create(args)
  criteria = CompletionKit::Criteria.new(args.slice("name", "description"))
  if criteria.save
    replace_metric_memberships(criteria, args["metric_ids"])
    text_result(criteria.reload.as_json)
  else
    error_result(criteria.errors.full_messages.join(", "))
  end
end

.definitions ⇒ Object 



46
47
48
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 46

def self.definitions
  TOOLS.map { |name, config| {name: name, description: config[:description], inputSchema: config[:inputSchema]} }
end

.delete(args) ⇒ Object 



83
84
85
86
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 83

def self.delete(args)
  CompletionKit::Criteria.find(args["id"]).destroy!
  text_result("Criteria #{args["id"]} deleted")
end

.error_result(message) ⇒ Object 



93
94
95
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 93

def self.error_result(message)
  {content: [{type: "text", text: message}], isError: true}
end

.get(args) ⇒ Object 



59
60
61
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 59

def self.get(args)
  text_result(CompletionKit::Criteria.find(args["id"]).as_json)
end

.list(_args) ⇒ Object 



55
56
57
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 55

def self.list(_args)
  text_result(CompletionKit::Criteria.order(created_at: :desc).map(&:as_json))
end

.replace_metric_memberships(criteria, metric_ids) ⇒ Object 



97
98
99
100
101
102
103
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 97

def self.replace_metric_memberships(criteria, metric_ids)
  return unless metric_ids
  criteria.criteria_memberships.delete_all
  Array(metric_ids).reject(&:blank?).each_with_index do |metric_id, index|
    criteria.criteria_memberships.create!(metric_id: metric_id, position: index + 1)
  end
end

.text_result(data) ⇒ Object 



88
89
90
91
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 88

def self.text_result(data)
  text = data.is_a?(String) ? data : data.to_json
  {content: [{type: "text", text: text}]}
end

.update(args) ⇒ Object 



73
74
75
76
77
78
79
80
81
 
# File 'app/services/completion_kit/mcp_tools/criteria.rb', line 73

def self.update(args)
  criteria = CompletionKit::Criteria.find(args["id"])
  if criteria.update(args.except("id", "metric_ids").slice("name", "description"))
    replace_metric_memberships(criteria, args["metric_ids"]) if args.key?("metric_ids")
    text_result(criteria.reload.as_json)
  else
    error_result(criteria.errors.full_messages.join(", "))
  end
end