Class: CompletionKit::McpDispatcher

Inherits:
Object
  • Object
show all
Defined in:
app/services/completion_kit/mcp_dispatcher.rb

Defined Under Namespace

Classes: InvalidParams, MethodNotFound

Constant Summary collapse

PROTOCOL_VERSION = 
"2025-03-26"

Class Method Summary collapse

Class Method Details

.call_tool(name, arguments) ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'app/services/completion_kit/mcp_dispatcher.rb', line 42

def self.call_tool(name, arguments)
  case name
  when /\Aprompts_/              then McpTools::Prompts.call(name, arguments)
  when /\Aruns_/                 then McpTools::Runs.call(name, arguments)
  when /\Aresponses_/            then McpTools::Responses.call(name, arguments)
  when /\Adatasets_/             then McpTools::Datasets.call(name, arguments)
  when /\Ametric_versions_/      then McpTools::MetricVersions.call(name, arguments)
  when /\Ametric_groups_/        then McpTools::MetricGroups.call(name, arguments)
  when /\Ametrics_/              then McpTools::Metrics.call(name, arguments)
  when /\Aprovider_credentials_/ then McpTools::ProviderCredentials.call(name, arguments)
  when /\Atags_/                 then McpTools::Tags.call(name, arguments)
  when /\Acalibrations_/         then McpTools::Calibrations.call(name, arguments)
  when /\Ajudges_/               then McpTools::Judges.call(name, arguments)
  else raise MethodNotFound, "Unknown tool: #{name}"
  end
end

.dispatch(method, params) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
 
# File 'app/services/completion_kit/mcp_dispatcher.rb', line 17

def self.dispatch(method, params)
  case method
  when "tools/list"
    {tools: tool_definitions}
  when "tools/call"
    call_tool(params&.dig("name"), params&.dig("arguments") || {})
  else
    raise MethodNotFound, "Method not found: #{method}"
  end
end

.initialize_session ⇒ Object 



8
9
10
11
12
13
14
15
 
# File 'app/services/completion_kit/mcp_dispatcher.rb', line 8

def self.initialize_session
  {
    session_id: McpSession.start!,
    protocolVersion: PROTOCOL_VERSION,
    serverInfo: {name: "CompletionKit", version: CompletionKit::VERSION},
    capabilities: {tools: {listChanged: false}}
  }
end

.tool_definitions ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'app/services/completion_kit/mcp_dispatcher.rb', line 28

def self.tool_definitions
  McpTools::Prompts.definitions +
    McpTools::Runs.definitions +
    McpTools::Responses.definitions +
    McpTools::Datasets.definitions +
    McpTools::Metrics.definitions +
    McpTools::MetricGroups.definitions +
    McpTools::MetricVersions.definitions +
    McpTools::ProviderCredentials.definitions +
    McpTools::Tags.definitions +
    McpTools::Calibrations.definitions +
    McpTools::Judges.definitions
end