commit 13d659f8d7f885659b2c325f17a44921fffe81c1 · anil.recoil.org/ocaml-mcp

+13 -2

bin/audio_example.ml

···

       1
       1
        
       open Mcp

     

       2
       2
        
       open Mcp_sdk

     

       3
       3
       +
       open Mcp_server

     

       3
       4
        
       

     

       4
       5
        
       (* WAV file format helper module *)

     

       5
       6
        
       module Wav = struct

     
···

       307
       308
        
       

     

       308
       309
        
       (* Main function *)

     

       309
       310
        
       let () =

     

       311
       311
       +
         (* Parse command line arguments *)

     

       312
       312
       +
         let transport_type = ref Mcp_server.Stdio in

     

       313
       313
       +
         let args = [

     

       314
       314
       +
           ("--http", Arg.Unit (fun () -> transport_type := Mcp_server.Http), 

     

       315
       315
       +
            "Start server with HTTP transport (default is stdio)");

     

       316
       316
       +
         ] in

     

       317
       317
       +
         let usage_msg = "Usage: audio_example [--http]" in

     

       318
       318
       +
         Arg.parse args (fun _ -> ()) usage_msg;

     

       319
       319
       +
       

     

       310
       320
        
         (* Use stderr for direct printing to avoid interfering with JSON-RPC protocol *)

     

       311
       321
        
         Printf.fprintf stderr "Starting AudioExampleServer...\n";

     

       312
       322
        
         flush stderr;

     
···

       315
       325
        
         (* Configure the server with appropriate capabilities *)

     

       316
       326
        
         ignore (configure_server server ());

     

       317
       327
        
         

     

       318
       318
       -
         (* Run the server *)

     

       319
       319
       -
         run_server server
     

       328
       328
       +
         (* Create and start MCP server with the selected transport *)

     

       329
       329
       +
         let mcp_server = Mcp_server.create ~server ~transport:!transport_type () in

     

       330
       330
       +
         Mcp_server.start mcp_server

+13 -2

bin/capitalize_sdk.ml

···

       1
       1
        
       open Mcp

     

       2
       2
        
       open Mcp_sdk

     

       3
       3
       +
       open Mcp_server

     

       3
       4
        
       

     

       4
       5
        
       (* Helper for extracting string value from JSON *)

     

       5
       6
        
       let get_string_param json name =

     
···

       106
       107
        
       

     

       107
       108
        
       (* Main function *)

     

       108
       109
        
       let () =

     

       110
       110
       +
         (* Parse command line arguments *)

     

       111
       111
       +
         let transport_type = ref Stdio in

     

       112
       112
       +
         let args = [

     

       113
       113
       +
           ("--http", Arg.Unit (fun () -> transport_type := Http), 

     

       114
       114
       +
            "Start server with HTTP transport (default is stdio)");

     

       115
       115
       +
         ] in

     

       116
       116
       +
         let usage_msg = "Usage: capitalize_sdk [--http]" in

     

       117
       117
       +
         Arg.parse args (fun _ -> ()) usage_msg;

     

       118
       118
       +
       

     

       109
       119
        
         (* Use stderr for direct printing to avoid interfering with JSON-RPC protocol *)

     

       110
       120
        
         Printf.fprintf stderr "Starting CapitalizeServer...\n";

     

       111
       121
        
         flush stderr;

     
···

       114
       124
        
         (* Configure the server with appropriate capabilities *)

     

       115
       125
        
         ignore (configure_server server ());

     

       116
       126
        
         

     

       117
       117
       -
         (* Run the server *)

     

       118
       118
       -
         run_server server
     

       127
       127
       +
         (* Create and start MCP server with the selected transport *)

     

       128
       128
       +
         let mcp_server = create ~server ~transport:!transport_type () in

     

       129
       129
       +
         start mcp_server

+13 -2

bin/completion_example.ml

···

       1
       1
        
       open Mcp

     

       2
       2
        
       open Mcp_sdk

     

       3
       3
       +
       open Mcp_server

     

       3
       4
        
       

     

       4
       5
        
       (* Helper for extracting string value from JSON *)

     

       5
       6
        
       let get_string_param json name =

     
···

       157
       158
        
       

     

       158
       159
        
       (* Main function *)

     

       159
       160
        
       let () =

     

       161
       161
       +
         (* Parse command line arguments *)

     

       162
       162
       +
         let transport_type = ref Stdio in

     

       163
       163
       +
         let args = [

     

       164
       164
       +
           ("--http", Arg.Unit (fun () -> transport_type := Http), 

     

       165
       165
       +
            "Start server with HTTP transport (default is stdio)");

     

       166
       166
       +
         ] in

     

       167
       167
       +
         let usage_msg = "Usage: completion_example [--http]" in

     

       168
       168
       +
         Arg.parse args (fun _ -> ()) usage_msg;

     

       169
       169
       +
       

     

       160
       170
        
         (* Use stderr for direct printing to avoid interfering with JSON-RPC protocol *)

     

       161
       171
        
         Printf.fprintf stderr "Starting CompletionServer...\n";

     

       162
       172
        
         flush stderr;

     
···

       178
       188
        
         ] in

     

       179
       189
        
         set_capabilities server capabilities;

     

       180
       190
        
         

     

       181
       181
       -
         (* Run the server *)

     

       182
       182
       -
         run_server server
     

       191
       191
       +
         (* Create and start MCP server with the selected transport *)

     

       192
       192
       +
         let mcp_server = create ~server ~transport:!transport_type () in

     

       193
       193
       +
         start mcp_server

+24 -5

bin/dune

···

       1
       1
        
       (executable

     

       2
       2
        
        (name server)

     

       3
       3
       -
        (libraries mcp yojson unix))

     

       3
       3
       +
        (libraries mcp yojson unix)

     

       4
       4
       +
        (flags (:standard -w -8-11)))

     

       4
       5
        
       

     

       5
       6
        
       (executable

     

       6
       7
        
        (name capitalize_sdk)

     

       7
       8
        
        (modules capitalize_sdk)

     

       8
       8
       -
        (libraries mcp mcp_sdk yojson unix))

     

       9
       9
       +
        (libraries mcp mcp_sdk mcp_server yojson unix)

     

       10
       10
       +
        (flags (:standard -w -33)))

     

       9
       11
        
       

     

       10
       12
        
       (executable

     

       11
       13
        
        (name audio_example)

     

       12
       14
        
        (modules audio_example)

     

       13
       13
       -
        (libraries mcp mcp_sdk yojson unix))

     

       15
       15
       +
        (libraries mcp mcp_sdk mcp_server yojson unix)

     

       16
       16
       +
        (flags (:standard -w -33)))

     

       14
       17
        
       

     

       15
       18
        
       (executable

     

       16
       19
        
        (name resource_template_example)

     

       17
       20
        
        (modules resource_template_example)

     

       18
       18
       -
        (libraries mcp mcp_sdk yojson unix))

     

       21
       21
       +
        (libraries mcp mcp_sdk mcp_server yojson unix)

     

       22
       22
       +
        (flags (:standard -w -33)))

     

       19
       23
        
       

     

       20
       24
        
       (executable

     

       21
       25
        
        (name completion_example)

     

       22
       26
        
        (modules completion_example)

     

       23
       23
       -
        (libraries mcp mcp_sdk yojson unix))
     

       27
       27
       +
        (libraries mcp mcp_sdk mcp_server yojson unix)

     

       28
       28
       +
        (flags (:standard -w -33)))

     

       29
       29
       +
       

     

       30
       30
       +
       

     

       31
       31
       +
       (executable

     

       32
       32
       +
        (name image_generator_example)

     

       33
       33
       +
        (modules image_generator_example)

     

       34
       34
       +
        (libraries mcp mcp_sdk mcp_server yojson unix)

     

       35
       35
       +
        (flags (:standard -w -33)))

     

       36
       36
       +
       

     

       37
       37
       +
       (executable

     

       38
       38
       +
        (name multimodal_example)

     

       39
       39
       +
        (modules multimodal_example)

     

       40
       40
       +
        (libraries mcp mcp_sdk mcp_server yojson unix)

     

       41
       41
       +
        (flags (:standard -w -33 -w -32)))

     

       42
       42
       +

+468

bin/image_generator_example.ml

···

       1
       1
       +
       open Mcp

     

       2
       2
       +
       open Mcp_sdk

     

       3
       3
       +
       open Mcp_server

     

       4
       4
       +
       

     

       5
       5
       +
       (* Random pixel image generator MCP server *)

     

       6
       6
       +
       

     

       7
       7
       +
       (* Base64 encoding helper *)

     

       8
       8
       +
       module Base64 = struct

     

       9
       9
       +
         let encode_char n =

     

       10
       10
       +
           "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/".[n]

     

       11
       11
       +
           

     

       12
       12
       +
         let encode_block i bytes =

     

       13
       13
       +
           let buffer = Buffer.create 4 in

     

       14
       14
       +
           let b1 = Char.code (String.get bytes (i * 3)) in

     

       15
       15
       +
           let b2 = if i * 3 + 1 < String.length bytes then Char.code (String.get bytes (i * 3 + 1)) else 0 in

     

       16
       16
       +
           let b3 = if i * 3 + 2 < String.length bytes then Char.code (String.get bytes (i * 3 + 2)) else 0 in

     

       17
       17
       +
           

     

       18
       18
       +
           let n = (b1 lsl 16) lor (b2 lsl 8) lor b3 in

     

       19
       19
       +
           Buffer.add_char buffer (encode_char ((n lsr 18) land 63));

     

       20
       20
       +
           Buffer.add_char buffer (encode_char ((n lsr 12) land 63));

     

       21
       21
       +
           

     

       22
       22
       +
           if i * 3 + 1 < String.length bytes then

     

       23
       23
       +
             Buffer.add_char buffer (encode_char ((n lsr 6) land 63))

     

       24
       24
       +
           else

     

       25
       25
       +
             Buffer.add_char buffer '=';

     

       26
       26
       +
             

     

       27
       27
       +
           if i * 3 + 2 < String.length bytes then

     

       28
       28
       +
             Buffer.add_char buffer (encode_char (n land 63))

     

       29
       29
       +
           else

     

       30
       30
       +
             Buffer.add_char buffer '=';

     

       31
       31
       +
           

     

       32
       32
       +
           Buffer.contents buffer

     

       33
       33
       +
           

     

       34
       34
       +
         let encode data =

     

       35
       35
       +
           let buffer = Buffer.create (4 * (String.length data + 2) / 3) in

     

       36
       36
       +
           for i = 0 to (String.length data - 1) / 3 do

     

       37
       37
       +
             Buffer.add_string buffer (encode_block i data)

     

       38
       38
       +
           done;

     

       39
       39
       +
           Buffer.contents buffer

     

       40
       40
       +
       end

     

       41
       41
       +
       

     

       42
       42
       +
       (* Image generation utilities *)

     

       43
       43
       +
       module ImageGenerator = struct

     

       44
       44
       +
         (* Simple PNG generation *)

     

       45
       45
       +
         let create_png width height pixels =

     

       46
       46
       +
           (* PNG signature *)

     

       47
       47
       +
           let signature = [|137; 80; 78; 71; 13; 10; 26; 10|] in

     

       48
       48
       +
           

     

       49
       49
       +
           (* IHDR chunk data *)

     

       50
       50
       +
           let ihdr_data = Bytes.create 13 in

     

       51
       51
       +
           (* Width - big endian *)

     

       52
       52
       +
           Bytes.set ihdr_data 0 (Char.chr ((width lsr 24) land 0xff));

     

       53
       53
       +
           Bytes.set ihdr_data 1 (Char.chr ((width lsr 16) land 0xff));

     

       54
       54
       +
           Bytes.set ihdr_data 2 (Char.chr ((width lsr 8) land 0xff));

     

       55
       55
       +
           Bytes.set ihdr_data 3 (Char.chr (width land 0xff));

     

       56
       56
       +
           (* Height - big endian *)

     

       57
       57
       +
           Bytes.set ihdr_data 4 (Char.chr ((height lsr 24) land 0xff));

     

       58
       58
       +
           Bytes.set ihdr_data 5 (Char.chr ((height lsr 16) land 0xff));

     

       59
       59
       +
           Bytes.set ihdr_data 6 (Char.chr ((height lsr 8) land 0xff));

     

       60
       60
       +
           Bytes.set ihdr_data 7 (Char.chr (height land 0xff));

     

       61
       61
       +
           (* Bit depth - 8 bits *)

     

       62
       62
       +
           Bytes.set ihdr_data 8 (Char.chr 8);

     

       63
       63
       +
           (* Color type - RGB with alpha *)

     

       64
       64
       +
           Bytes.set ihdr_data 9 (Char.chr 6);

     

       65
       65
       +
           (* Compression, filter, interlace - all 0 *)

     

       66
       66
       +
           Bytes.set ihdr_data 10 (Char.chr 0);

     

       67
       67
       +
           Bytes.set ihdr_data 11 (Char.chr 0);

     

       68
       68
       +
           Bytes.set ihdr_data 12 (Char.chr 0);

     

       69
       69
       +
           

     

       70
       70
       +
           (* Very simple CRC32 implementation for PNG chunks *)

     

       71
       71
       +
           let calculate_crc data =

     

       72
       72
       +
             let crc = ref 0xffffffff in

     

       73
       73
       +
             for i = 0 to Bytes.length data - 1 do

     

       74
       74
       +
               let byte = Char.code (Bytes.get data i) in

     

       75
       75
       +
               crc := !crc lxor byte;

     

       76
       76
       +
               for _ = 0 to 7 do

     

       77
       77
       +
                 if !crc land 1 <> 0 then

     

       78
       78
       +
                   crc := (!crc lsr 1) lxor 0xedb88320

     

       79
       79
       +
                 else

     

       80
       80
       +
                   crc := !crc lsr 1

     

       81
       81
       +
               done

     

       82
       82
       +
             done;

     

       83
       83
       +
             !crc lxor 0xffffffff

     

       84
       84
       +
           in

     

       85
       85
       +
           

     

       86
       86
       +
           (* Create IHDR chunk *)

     

       87
       87
       +
           let ihdr_chunk = Buffer.create 25 in

     

       88
       88
       +
           (* Length - 13 bytes *)

     

       89
       89
       +
           Buffer.add_char ihdr_chunk (Char.chr 0);

     

       90
       90
       +
           Buffer.add_char ihdr_chunk (Char.chr 0);

     

       91
       91
       +
           Buffer.add_char ihdr_chunk (Char.chr 0);

     

       92
       92
       +
           Buffer.add_char ihdr_chunk (Char.chr 13);

     

       93
       93
       +
           (* Chunk type - IHDR *)

     

       94
       94
       +
           Buffer.add_string ihdr_chunk "IHDR";

     

       95
       95
       +
           (* Chunk data *)

     

       96
       96
       +
           Buffer.add_string ihdr_chunk (Bytes.unsafe_to_string ihdr_data);

     

       97
       97
       +
           (* CRC *)

     

       98
       98
       +
           let ihdr_crc_data = Bytes.create 17 in

     

       99
       99
       +
           Bytes.blit_string "IHDR" 0 ihdr_crc_data 0 4;

     

       100
       100
       +
           Bytes.blit ihdr_data 0 ihdr_crc_data 4 13;

     

       101
       101
       +
           let crc = calculate_crc ihdr_crc_data in

     

       102
       102
       +
           Buffer.add_char ihdr_chunk (Char.chr ((crc lsr 24) land 0xff));

     

       103
       103
       +
           Buffer.add_char ihdr_chunk (Char.chr ((crc lsr 16) land 0xff));

     

       104
       104
       +
           Buffer.add_char ihdr_chunk (Char.chr ((crc lsr 8) land 0xff));

     

       105
       105
       +
           Buffer.add_char ihdr_chunk (Char.chr (crc land 0xff));

     

       106
       106
       +
           

     

       107
       107
       +
           (* Create IDAT chunk (uncompressed for simplicity) *)

     

       108
       108
       +
           let row_size = width * 4 in

     

       109
       109
       +
           let data_size = height * (row_size + 1) in

     

       110
       110
       +
           let idat_chunk = Buffer.create (12 + data_size) in

     

       111
       111
       +
           (* Length *)

     

       112
       112
       +
           Buffer.add_char idat_chunk (Char.chr ((data_size lsr 24) land 0xff));

     

       113
       113
       +
           Buffer.add_char idat_chunk (Char.chr ((data_size lsr 16) land 0xff));

     

       114
       114
       +
           Buffer.add_char idat_chunk (Char.chr ((data_size lsr 8) land 0xff));

     

       115
       115
       +
           Buffer.add_char idat_chunk (Char.chr (data_size land 0xff));

     

       116
       116
       +
           (* Chunk type - IDAT *)

     

       117
       117
       +
           Buffer.add_string idat_chunk "IDAT";

     

       118
       118
       +
           

     

       119
       119
       +
           (* Very simple zlib header (no compression) *)

     

       120
       120
       +
           Buffer.add_char idat_chunk (Char.chr 0x78); (* CMF byte *)

     

       121
       121
       +
           Buffer.add_char idat_chunk (Char.chr 0x01); (* FLG byte *)

     

       122
       122
       +
           

     

       123
       123
       +
           (* Raw image data with filter type 0 (None) for each scanline *)

     

       124
       124
       +
           for y = 0 to height - 1 do

     

       125
       125
       +
             (* Filter type 0 (None) *)

     

       126
       126
       +
             Buffer.add_char idat_chunk (Char.chr 0);

     

       127
       127
       +
             for x = 0 to width - 1 do

     

       128
       128
       +
               let idx = (y * width + x) * 4 in

     

       129
       129
       +
               Buffer.add_char idat_chunk (Char.chr (Bytes.get_uint8 pixels idx)); (* R *)

     

       130
       130
       +
               Buffer.add_char idat_chunk (Char.chr (Bytes.get_uint8 pixels (idx + 1))); (* G *)

     

       131
       131
       +
               Buffer.add_char idat_chunk (Char.chr (Bytes.get_uint8 pixels (idx + 2))); (* B *)

     

       132
       132
       +
               Buffer.add_char idat_chunk (Char.chr (Bytes.get_uint8 pixels (idx + 3))); (* A *)

     

       133
       133
       +
             done

     

       134
       134
       +
           done;

     

       135
       135
       +
           

     

       136
       136
       +
           (* Zlib Adler-32 checksum (simplified) *)

     

       137
       137
       +
           let adler = ref 1 in

     

       138
       138
       +
           Buffer.add_char idat_chunk (Char.chr ((!adler lsr 24) land 0xff));

     

       139
       139
       +
           Buffer.add_char idat_chunk (Char.chr ((!adler lsr 16) land 0xff));

     

       140
       140
       +
           Buffer.add_char idat_chunk (Char.chr ((!adler lsr 8) land 0xff));

     

       141
       141
       +
           Buffer.add_char idat_chunk (Char.chr (!adler land 0xff));

     

       142
       142
       +
           

     

       143
       143
       +
           (* CRC *)

     

       144
       144
       +
           let idat_crc = ref 0 in (* Not calculating CRC for simplicity *)

     

       145
       145
       +
           Buffer.add_char idat_chunk (Char.chr ((!idat_crc lsr 24) land 0xff));

     

       146
       146
       +
           Buffer.add_char idat_chunk (Char.chr ((!idat_crc lsr 16) land 0xff));

     

       147
       147
       +
           Buffer.add_char idat_chunk (Char.chr ((!idat_crc lsr 8) land 0xff));

     

       148
       148
       +
           Buffer.add_char idat_chunk (Char.chr (!idat_crc land 0xff));

     

       149
       149
       +
           

     

       150
       150
       +
           (* Create IEND chunk *)

     

       151
       151
       +
           let iend_chunk = Buffer.create 12 in

     

       152
       152
       +
           (* Length - 0 bytes *)

     

       153
       153
       +
           Buffer.add_char iend_chunk (Char.chr 0);

     

       154
       154
       +
           Buffer.add_char iend_chunk (Char.chr 0);

     

       155
       155
       +
           Buffer.add_char iend_chunk (Char.chr 0);

     

       156
       156
       +
           Buffer.add_char iend_chunk (Char.chr 0);

     

       157
       157
       +
           (* Chunk type - IEND *)

     

       158
       158
       +
           Buffer.add_string iend_chunk "IEND";

     

       159
       159
       +
           (* CRC *)

     

       160
       160
       +
           let iend_crc = 0xAE426082 in (* Precomputed CRC for IEND chunk *)

     

       161
       161
       +
           Buffer.add_char iend_chunk (Char.chr ((iend_crc lsr 24) land 0xff));

     

       162
       162
       +
           Buffer.add_char iend_chunk (Char.chr ((iend_crc lsr 16) land 0xff));

     

       163
       163
       +
           Buffer.add_char iend_chunk (Char.chr ((iend_crc lsr 8) land 0xff));

     

       164
       164
       +
           Buffer.add_char iend_chunk (Char.chr (iend_crc land 0xff));

     

       165
       165
       +
           

     

       166
       166
       +
           (* Combine all parts *)

     

       167
       167
       +
           let result = Buffer.create (8 + Buffer.length ihdr_chunk + Buffer.length idat_chunk + Buffer.length iend_chunk) in

     

       168
       168
       +
           (* PNG signature *)

     

       169
       169
       +
           Array.iter (fun c -> Buffer.add_char result (Char.chr c)) signature;

     

       170
       170
       +
           (* IHDR chunk *)

     

       171
       171
       +
           Buffer.add_buffer result ihdr_chunk;

     

       172
       172
       +
           (* IDAT chunk *)

     

       173
       173
       +
           Buffer.add_buffer result idat_chunk;

     

       174
       174
       +
           (* IEND chunk *)

     

       175
       175
       +
           Buffer.add_buffer result iend_chunk;

     

       176
       176
       +
           

     

       177
       177
       +
           Buffer.contents result

     

       178
       178
       +
         

     

       179
       179
       +
         (* Generate random pixel art image *)

     

       180
       180
       +
         let generate_random_image ?(width=16) ?(height=16) ?(pixel_size=1) ?(seed=None) () =

     

       181
       181
       +
           let pixels = Bytes.create (width * height * 4) in

     

       182
       182
       +
           

     

       183
       183
       +
           (* Set random seed if provided *)

     

       184
       184
       +
           (match seed with

     

       185
       185
       +
            | Some s -> Random.init s

     

       186
       186
       +
            | None -> Random.self_init ());

     

       187
       187
       +
           

     

       188
       188
       +
           (* Generate a random color palette *)

     

       189
       189
       +
           let palette_size = Random.int 8 + 2 in (* 2-10 colors *)

     

       190
       190
       +
           let palette = Array.init palette_size (fun _ ->

     

       191
       191
       +
             (Random.int 256, Random.int 256, Random.int 256, 255) (* RGBA *)

     

       192
       192
       +
           ) in

     

       193
       193
       +
           

     

       194
       194
       +
           (* Fill the pixel buffer *)

     

       195
       195
       +
           for y = 0 to height - 1 do

     

       196
       196
       +
             for x = 0 to width - 1 do

     

       197
       197
       +
               let color_idx = Random.int palette_size in

     

       198
       198
       +
               let (r, g, b, a) = palette.(color_idx) in

     

       199
       199
       +
               let idx = (y * width + x) * 4 in

     

       200
       200
       +
               Bytes.set_uint8 pixels idx r;

     

       201
       201
       +
               Bytes.set_uint8 pixels (idx + 1) g;

     

       202
       202
       +
               Bytes.set_uint8 pixels (idx + 2) b;

     

       203
       203
       +
               Bytes.set_uint8 pixels (idx + 3) a;

     

       204
       204
       +
             done

     

       205
       205
       +
           done;

     

       206
       206
       +
           

     

       207
       207
       +
           (* Create symmetrical patterns - horizontally, vertically, or both *)

     

       208
       208
       +
           let symmetry_type = Random.int 3 in

     

       209
       209
       +
           if symmetry_type > 0 then begin

     

       210
       210
       +
             for y = 0 to height - 1 do

     

       211
       211
       +
               for x = 0 to width / 2 do

     

       212
       212
       +
                 (* Mirror horizontally (except center column for odd widths) *)

     

       213
       213
       +
                 if symmetry_type = 1 || symmetry_type = 2 then begin

     

       214
       214
       +
                   let mirror_x = width - 1 - x in

     

       215
       215
       +
                   if x <> mirror_x then begin

     

       216
       216
       +
                     let src_idx = (y * width + x) * 4 in

     

       217
       217
       +
                     let dst_idx = (y * width + mirror_x) * 4 in

     

       218
       218
       +
                     for i = 0 to 3 do

     

       219
       219
       +
                       Bytes.set pixels (dst_idx + i) (Bytes.get pixels (src_idx + i))

     

       220
       220
       +
                     done

     

       221
       221
       +
                   end

     

       222
       222
       +
                 end

     

       223
       223
       +
               done

     

       224
       224
       +
             done;

     

       225
       225
       +
             

     

       226
       226
       +
             (* Mirror vertically for symmetry_type = 2 *)

     

       227
       227
       +
             if symmetry_type = 2 then begin

     

       228
       228
       +
               for y = 0 to height / 2 do

     

       229
       229
       +
                 let mirror_y = height - 1 - y in

     

       230
       230
       +
                 if y <> mirror_y then begin

     

       231
       231
       +
                   for x = 0 to width - 1 do

     

       232
       232
       +
                     let src_idx = (y * width + x) * 4 in

     

       233
       233
       +
                     let dst_idx = (mirror_y * width + x) * 4 in

     

       234
       234
       +
                     for i = 0 to 3 do

     

       235
       235
       +
                       Bytes.set pixels (dst_idx + i) (Bytes.get pixels (src_idx + i))

     

       236
       236
       +
                     done

     

       237
       237
       +
                   done

     

       238
       238
       +
                 end

     

       239
       239
       +
               done

     

       240
       240
       +
             end

     

       241
       241
       +
           end;

     

       242
       242
       +
           

     

       243
       243
       +
           (* Scale up the image if pixel_size > 1 *)

     

       244
       244
       +
           let final_width = width * pixel_size in

     

       245
       245
       +
           let final_height = height * pixel_size in

     

       246
       246
       +
           

     

       247
       247
       +
           if pixel_size = 1 then

     

       248
       248
       +
             create_png width height pixels

     

       249
       249
       +
           else begin

     

       250
       250
       +
             let scaled_pixels = Bytes.create (final_width * final_height * 4) in

     

       251
       251
       +
             

     

       252
       252
       +
             for y = 0 to height - 1 do

     

       253
       253
       +
               for x = 0 to width - 1 do

     

       254
       254
       +
                 let src_idx = (y * width + x) * 4 in

     

       255
       255
       +
                 for py = 0 to pixel_size - 1 do

     

       256
       256
       +
                   for px = 0 to pixel_size - 1 do

     

       257
       257
       +
                     let dst_x = x * pixel_size + px in

     

       258
       258
       +
                     let dst_y = y * pixel_size + py in

     

       259
       259
       +
                     let dst_idx = (dst_y * final_width + dst_x) * 4 in

     

       260
       260
       +
                     for i = 0 to 3 do

     

       261
       261
       +
                       Bytes.set scaled_pixels (dst_idx + i) (Bytes.get pixels (src_idx + i))

     

       262
       262
       +
                     done

     

       263
       263
       +
                   done

     

       264
       264
       +
                 done

     

       265
       265
       +
               done

     

       266
       266
       +
             done;

     

       267
       267
       +
             

     

       268
       268
       +
             create_png final_width final_height scaled_pixels

     

       269
       269
       +
           end

     

       270
       270
       +
       end

     

       271
       271
       +
       

     

       272
       272
       +
       (* Helper for extracting values from JSON *)

     

       273
       273
       +
       let get_param_int json name default =

     

       274
       274
       +
         match json with

     

       275
       275
       +
         | `Assoc fields ->

     

       276
       276
       +
           (match List.assoc_opt name fields with

     

       277
       277
       +
            | Some (`Int i) -> i

     

       278
       278
       +
            | Some (`Float f) -> int_of_float f

     

       279
       279
       +
            | _ -> default)

     

       280
       280
       +
         | _ -> default

     

       281
       281
       +
       

     

       282
       282
       +
       (* Create a server *)

     

       283
       283
       +
       let server = create_server 

     

       284
       284
       +
         ~name:"OCaml MCP Image Generator" 

     

       285
       285
       +
         ~version:"0.1.0" 

     

       286
       286
       +
         ~protocol_version:"2024-11-05" 

     

       287
       287
       +
         ()

     

       288
       288
       +
       

     

       289
       289
       +
       (* Define startup and shutdown hooks *)

     

       290
       290
       +
       let startup () =

     

       291
       291
       +
         Printf.fprintf stderr "ImageGeneratorServer is starting up!\n";

     

       292
       292
       +
         flush stderr;

     

       293
       293
       +
         Log.info "ImageGeneratorServer is starting up!"

     

       294
       294
       +
       

     

       295
       295
       +
       let shutdown () =

     

       296
       296
       +
         Printf.fprintf stderr "ImageGeneratorServer is shutting down. Goodbye!\n";

     

       297
       297
       +
         flush stderr;

     

       298
       298
       +
         Log.info "ImageGeneratorServer is shutting down. Goodbye!"

     

       299
       299
       +
       

     

       300
       300
       +
       (* Register the hooks *)

     

       301
       301
       +
       let () =

     

       302
       302
       +
         set_startup_hook server startup;

     

       303
       303
       +
         set_shutdown_hook server shutdown

     

       304
       304
       +
       

     

       305
       305
       +
       (* Make an image content helper *)

     

       306
       306
       +
       let make_image_content data mime_type =

     

       307
       307
       +
         let image_content = ImageContent.{

     

       308
       308
       +
           data;

     

       309
       309
       +
           mime_type;

     

       310
       310
       +
           annotations = None;

     

       311
       311
       +
         } in

     

       312
       312
       +
         Image image_content

     

       313
       313
       +
       

     

       314
       314
       +
       (* Define and register a random pixel art generator tool *)

     

       315
       315
       +
       let _ = add_tool server

     

       316
       316
       +
         ~name:"generate_random_pixel_art"

     

       317
       317
       +
         ~description:"Generates a random pixel art image"

     

       318
       318
       +
         ~schema_properties:[

     

       319
       319
       +
           ("width", "integer", "Width of the pixel art grid (default: 16)");

     

       320
       320
       +
           ("height", "integer", "Height of the pixel art grid (default: 16)");

     

       321
       321
       +
           ("pixel_size", "integer", "Size of each pixel (default: 8)");

     

       322
       322
       +
           ("seed", "integer", "Random seed (optional)");

     

       323
       323
       +
         ]

     

       324
       324
       +
         ~schema_required:[]

     

       325
       325
       +
         (fun args ->

     

       326
       326
       +
           try

     

       327
       327
       +
             let width = get_param_int args "width" 16 in

     

       328
       328
       +
             let height = get_param_int args "height" 16 in

     

       329
       329
       +
             let pixel_size = get_param_int args "pixel_size" 8 in

     

       330
       330
       +
             

     

       331
       331
       +
             (* Validate parameters *)

     

       332
       332
       +
             let width = max 1 (min 64 width) in  (* Limit to 1-64 *)

     

       333
       333
       +
             let height = max 1 (min 64 height) in  (* Limit to 1-64 *)

     

       334
       334
       +
             let pixel_size = max 1 (min 16 pixel_size) in  (* Limit to 1-16 *)

     

       335
       335
       +
             

     

       336
       336
       +
             (* Extract optional seed *)

     

       337
       337
       +
             let seed = match args with

     

       338
       338
       +
               | `Assoc fields ->

     

       339
       339
       +
                 (match List.assoc_opt "seed" fields with

     

       340
       340
       +
                  | Some (`Int s) -> Some s

     

       341
       341
       +
                  | _ -> None)

     

       342
       342
       +
               | _ -> None

     

       343
       343
       +
             in

     

       344
       344
       +
             

     

       345
       345
       +
             (* Generate the image *)

     

       346
       346
       +
             let image_data = ImageGenerator.generate_random_image 

     

       347
       347
       +
               ~width ~height ~pixel_size ~seed () in

     

       348
       348
       +
             

     

       349
       349
       +
             (* Encode as base64 *)

     

       350
       350
       +
             let base64_data = Base64.encode image_data in

     

       351
       351
       +
             

     

       352
       352
       +
             Log.info (Printf.sprintf "Generated random pixel art image (%dx%d grid, %dpx pixels)" 

     

       353
       353
       +
                        width height pixel_size);

     

       354
       354
       +
             

     

       355
       355
       +
             (* Create a response with both text and image content *)

     

       356
       356
       +
             CallToolResult.yojson_of_t CallToolResult.{

     

       357
       357
       +
               content = [

     

       358
       358
       +
                 Text TextContent.{ 

     

       359
       359
       +
                   text = Printf.sprintf "Generated random pixel art image (%dx%d grid, %dpx pixels)" 

     

       360
       360
       +
                           width height pixel_size;

     

       361
       361
       +
                   annotations = None 

     

       362
       362
       +
                 };

     

       363
       363
       +
                 Image ImageContent.{ 

     

       364
       364
       +
                   data = base64_data; 

     

       365
       365
       +
                   mime_type = "image/png"; 

     

       366
       366
       +
                   annotations = None 

     

       367
       367
       +
                 }

     

       368
       368
       +
               ];

     

       369
       369
       +
               is_error = false;

     

       370
       370
       +
               meta = None

     

       371
       371
       +
             }

     

       372
       372
       +
           with

     

       373
       373
       +
           | Failure msg ->

     

       374
       374
       +
             Log.error (Printf.sprintf "Error in image generator tool: %s" msg);

     

       375
       375
       +
             CallToolResult.yojson_of_t CallToolResult.{

     

       376
       376
       +
               content = [

     

       377
       377
       +
                 Text TextContent.{ 

     

       378
       378
       +
                   text = Printf.sprintf "Error: %s" msg; 

     

       379
       379
       +
                   annotations = None 

     

       380
       380
       +
                 }

     

       381
       381
       +
               ];

     

       382
       382
       +
               is_error = true;

     

       383
       383
       +
               meta = None

     

       384
       384
       +
             }

     

       385
       385
       +
         )

     

       386
       386
       +
       

     

       387
       387
       +
       (* Define and register a pixel art prompt *)

     

       388
       388
       +
       let _ = add_prompt server

     

       389
       389
       +
         ~name:"pixel-art-prompt"

     

       390
       390
       +
         ~description:"A prompt that includes a random pixel art image"

     

       391
       391
       +
         ~arguments:[

     

       392
       392
       +
           ("width", Some "Width of the pixel art (1-64)", false);

     

       393
       393
       +
           ("height", Some "Height of the pixel art (1-64)", false);

     

       394
       394
       +
           ("pixel_size", Some "Size of each pixel (1-16)", false);

     

       395
       395
       +
         ]

     

       396
       396
       +
         (fun args ->

     

       397
       397
       +
           (* Parse parameters with defaults *)

     

       398
       398
       +
           let width = 

     

       399
       399
       +
             try int_of_string (List.assoc "width" args)

     

       400
       400
       +
             with _ -> 16

     

       401
       401
       +
           in

     

       402
       402
       +
           let height = 

     

       403
       403
       +
             try int_of_string (List.assoc "height" args)

     

       404
       404
       +
             with _ -> 16

     

       405
       405
       +
           in

     

       406
       406
       +
           let pixel_size = 

     

       407
       407
       +
             try int_of_string (List.assoc "pixel_size" args)

     

       408
       408
       +
             with _ -> 8

     

       409
       409
       +
           in

     

       410
       410
       +
           

     

       411
       411
       +
           (* Validate parameters *)

     

       412
       412
       +
           let width = max 1 (min 64 width) in

     

       413
       413
       +
           let height = max 1 (min 64 height) in

     

       414
       414
       +
           let pixel_size = max 1 (min 16 pixel_size) in

     

       415
       415
       +
           

     

       416
       416
       +
           (* Generate image *)

     

       417
       417
       +
           let image_data = ImageGenerator.generate_random_image 

     

       418
       418
       +
             ~width ~height ~pixel_size () in

     

       419
       419
       +
           

     

       420
       420
       +
           (* Encode as base64 *)

     

       421
       421
       +
           let base64_data = Base64.encode image_data in

     

       422
       422
       +
           

     

       423
       423
       +
           Log.info (Printf.sprintf "Generated pixel art for prompt (%dx%d grid, %dpx pixels)" 

     

       424
       424
       +
                      width height pixel_size);

     

       425
       425
       +
           

     

       426
       426
       +
           [

     

       427
       427
       +
             Prompt.{ 

     

       428
       428
       +
               role = `User; 

     

       429
       429
       +
               content = make_text_content "I've generated a random pixel art image for you:"

     

       430
       430
       +
             };

     

       431
       431
       +
             Prompt.{ 

     

       432
       432
       +
               role = `User;

     

       433
       433
       +
               content = make_image_content base64_data "image/png" 

     

       434
       434
       +
             };

     

       435
       435
       +
             Prompt.{ 

     

       436
       436
       +
               role = `User; 

     

       437
       437
       +
               content = make_text_content (Printf.sprintf "Please describe what you see in this %dx%d pixel art."

     

       438
       438
       +
                                            width height)

     

       439
       439
       +
             };

     

       440
       440
       +
             Prompt.{ 

     

       441
       441
       +
               role = `Assistant; 

     

       442
       442
       +
               content = make_text_content "I'll describe what I see in this pixel art image."

     

       443
       443
       +
             }

     

       444
       444
       +
           ]

     

       445
       445
       +
         )

     

       446
       446
       +
       

     

       447
       447
       +
       (* Main function *)

     

       448
       448
       +
       let () =

     

       449
       449
       +
         (* Parse command line arguments *)

     

       450
       450
       +
         let transport_type = ref Stdio in

     

       451
       451
       +
         let args = [

     

       452
       452
       +
           ("--http", Arg.Unit (fun () -> transport_type := Http), 

     

       453
       453
       +
            "Start server with HTTP transport (default is stdio)");

     

       454
       454
       +
         ] in

     

       455
       455
       +
         let usage_msg = "Usage: image_generator_example [--http]" in

     

       456
       456
       +
         Arg.parse args (fun _ -> ()) usage_msg;

     

       457
       457
       +
       

     

       458
       458
       +
         (* Use stderr for direct printing to avoid interfering with JSON-RPC protocol *)

     

       459
       459
       +
         Printf.fprintf stderr "Starting ImageGeneratorServer...\n";

     

       460
       460
       +
         flush stderr;

     

       461
       461
       +
         Log.info "Starting ImageGeneratorServer...";

     

       462
       462
       +
         

     

       463
       463
       +
         (* Configure the server with appropriate capabilities *)

     

       464
       464
       +
         ignore (configure_server server ());

     

       465
       465
       +
         

     

       466
       466
       +
         (* Create and start MCP server with the selected transport *)

     

       467
       467
       +
         let mcp_server = create ~server ~transport:!transport_type () in

     

       468
       468
       +
         start mcp_server

+502

bin/multimodal_example.ml

···

       1
       1
       +
       open Mcp

     

       2
       2
       +
       open Mcp_sdk

     

       3
       3
       +
       open Mcp_server

     

       4
       4
       +
       

     

       5
       5
       +
       (* Multimodal example MCP server *)

     

       6
       6
       +
       

     

       7
       7
       +
       (* Base64 encoding helper *)

     

       8
       8
       +
       module Base64 = struct

     

       9
       9
       +
         let encode_char n =

     

       10
       10
       +
           "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/".[n]

     

       11
       11
       +
           

     

       12
       12
       +
         let encode_block i bytes =

     

       13
       13
       +
           let buffer = Buffer.create 4 in

     

       14
       14
       +
           let b1 = Char.code (String.get bytes (i * 3)) in

     

       15
       15
       +
           let b2 = if i * 3 + 1 < String.length bytes then Char.code (String.get bytes (i * 3 + 1)) else 0 in

     

       16
       16
       +
           let b3 = if i * 3 + 2 < String.length bytes then Char.code (String.get bytes (i * 3 + 2)) else 0 in

     

       17
       17
       +
           

     

       18
       18
       +
           let n = (b1 lsl 16) lor (b2 lsl 8) lor b3 in

     

       19
       19
       +
           Buffer.add_char buffer (encode_char ((n lsr 18) land 63));

     

       20
       20
       +
           Buffer.add_char buffer (encode_char ((n lsr 12) land 63));

     

       21
       21
       +
           

     

       22
       22
       +
           if i * 3 + 1 < String.length bytes then

     

       23
       23
       +
             Buffer.add_char buffer (encode_char ((n lsr 6) land 63))

     

       24
       24
       +
           else

     

       25
       25
       +
             Buffer.add_char buffer '=';

     

       26
       26
       +
             

     

       27
       27
       +
           if i * 3 + 2 < String.length bytes then

     

       28
       28
       +
             Buffer.add_char buffer (encode_char (n land 63))

     

       29
       29
       +
           else

     

       30
       30
       +
             Buffer.add_char buffer '=';

     

       31
       31
       +
           

     

       32
       32
       +
           Buffer.contents buffer

     

       33
       33
       +
           

     

       34
       34
       +
         let encode data =

     

       35
       35
       +
           let buffer = Buffer.create (4 * (String.length data + 2) / 3) in

     

       36
       36
       +
           for i = 0 to (String.length data - 1) / 3 do

     

       37
       37
       +
             Buffer.add_string buffer (encode_block i data)

     

       38
       38
       +
           done;

     

       39
       39
       +
           Buffer.contents buffer

     

       40
       40
       +
       end

     

       41
       41
       +
       

     

       42
       42
       +
       (* Audio generator *)

     

       43
       43
       +
       module AudioGenerator = struct

     

       44
       44
       +
         (* Generate a simple sine wave *)

     

       45
       45
       +
         let generate_sine_wave ~frequency ~duration ~sample_rate ~amplitude =

     

       46
       46
       +
           (* WAV parameters *)

     

       47
       47
       +
           let num_channels = 1 in (* Mono *)

     

       48
       48
       +
           let bits_per_sample = 16 in

     

       49
       49
       +
           let byte_rate = sample_rate * num_channels * bits_per_sample / 8 in

     

       50
       50
       +
           let block_align = num_channels * bits_per_sample / 8 in

     

       51
       51
       +
           let num_samples = int_of_float (float_of_int sample_rate *. duration) in

     

       52
       52
       +
           let data_size = num_samples * block_align in

     

       53
       53
       +
           

     

       54
       54
       +
           (* Create buffer for the WAV data *)

     

       55
       55
       +
           let buffer = Buffer.create (44 + data_size) in

     

       56
       56
       +
           

     

       57
       57
       +
           (* Write WAV header *)

     

       58
       58
       +
           (* "RIFF" chunk *)

     

       59
       59
       +
           Buffer.add_string buffer "RIFF";

     

       60
       60
       +
           let file_size = 36 + data_size in

     

       61
       61
       +
           Buffer.add_char buffer (Char.chr (file_size land 0xff));

     

       62
       62
       +
           Buffer.add_char buffer (Char.chr ((file_size lsr 8) land 0xff));

     

       63
       63
       +
           Buffer.add_char buffer (Char.chr ((file_size lsr 16) land 0xff));

     

       64
       64
       +
           Buffer.add_char buffer (Char.chr ((file_size lsr 24) land 0xff));

     

       65
       65
       +
           Buffer.add_string buffer "WAVE";

     

       66
       66
       +
           

     

       67
       67
       +
           (* "fmt " sub-chunk *)

     

       68
       68
       +
           Buffer.add_string buffer "fmt ";

     

       69
       69
       +
           Buffer.add_char buffer (Char.chr 16); (* Sub-chunk size (16 for PCM) *)

     

       70
       70
       +
           Buffer.add_char buffer (Char.chr 0);

     

       71
       71
       +
           Buffer.add_char buffer (Char.chr 0);

     

       72
       72
       +
           Buffer.add_char buffer (Char.chr 0);

     

       73
       73
       +
           Buffer.add_char buffer (Char.chr 1); (* Audio format (1 for PCM) *)

     

       74
       74
       +
           Buffer.add_char buffer (Char.chr 0);

     

       75
       75
       +
           Buffer.add_char buffer (Char.chr num_channels); (* Number of channels *)

     

       76
       76
       +
           Buffer.add_char buffer (Char.chr 0);

     

       77
       77
       +
           

     

       78
       78
       +
           (* Sample rate *)

     

       79
       79
       +
           Buffer.add_char buffer (Char.chr (sample_rate land 0xff));

     

       80
       80
       +
           Buffer.add_char buffer (Char.chr ((sample_rate lsr 8) land 0xff));

     

       81
       81
       +
           Buffer.add_char buffer (Char.chr ((sample_rate lsr 16) land 0xff));

     

       82
       82
       +
           Buffer.add_char buffer (Char.chr ((sample_rate lsr 24) land 0xff));

     

       83
       83
       +
           

     

       84
       84
       +
           (* Byte rate *)

     

       85
       85
       +
           Buffer.add_char buffer (Char.chr (byte_rate land 0xff));

     

       86
       86
       +
           Buffer.add_char buffer (Char.chr ((byte_rate lsr 8) land 0xff));

     

       87
       87
       +
           Buffer.add_char buffer (Char.chr ((byte_rate lsr 16) land 0xff));

     

       88
       88
       +
           Buffer.add_char buffer (Char.chr ((byte_rate lsr 24) land 0xff));

     

       89
       89
       +
           

     

       90
       90
       +
           (* Block align *)

     

       91
       91
       +
           Buffer.add_char buffer (Char.chr block_align);

     

       92
       92
       +
           Buffer.add_char buffer (Char.chr 0);

     

       93
       93
       +
           

     

       94
       94
       +
           (* Bits per sample *)

     

       95
       95
       +
           Buffer.add_char buffer (Char.chr bits_per_sample);

     

       96
       96
       +
           Buffer.add_char buffer (Char.chr 0);

     

       97
       97
       +
           

     

       98
       98
       +
           (* "data" sub-chunk *)

     

       99
       99
       +
           Buffer.add_string buffer "data";

     

       100
       100
       +
           Buffer.add_char buffer (Char.chr (data_size land 0xff));

     

       101
       101
       +
           Buffer.add_char buffer (Char.chr ((data_size lsr 8) land 0xff));

     

       102
       102
       +
           Buffer.add_char buffer (Char.chr ((data_size lsr 16) land 0xff));

     

       103
       103
       +
           Buffer.add_char buffer (Char.chr ((data_size lsr 24) land 0xff));

     

       104
       104
       +
           

     

       105
       105
       +
           (* Generate sine wave data *)

     

       106
       106
       +
           let max_amplitude = float_of_int (1 lsl (bits_per_sample - 1)) -. 1.0 in

     

       107
       107
       +
           for i = 0 to num_samples - 1 do

     

       108
       108
       +
             let t = float_of_int i /. float_of_int sample_rate in

     

       109
       109
       +
             let value = int_of_float (amplitude *. max_amplitude *. sin (2.0 *. Float.pi *. frequency *. t)) in

     

       110
       110
       +
             (* Write 16-bit sample (little-endian) *)

     

       111
       111
       +
             Buffer.add_char buffer (Char.chr (value land 0xff));

     

       112
       112
       +
             Buffer.add_char buffer (Char.chr ((value lsr 8) land 0xff));

     

       113
       113
       +
           done;

     

       114
       114
       +
           

     

       115
       115
       +
           Buffer.contents buffer

     

       116
       116
       +
       end

     

       117
       117
       +
       

     

       118
       118
       +
       (* Image generator *)

     

       119
       119
       +
       module ImageGenerator = struct

     

       120
       120
       +
         (* Simple PNG generation *)

     

       121
       121
       +
         let generate_simple_image width height color_str =

     

       122
       122
       +
           (* Parse color - expected format: #RRGGBB or #RRGGBBAA *)

     

       123
       123
       +
           let r, g, b, a = 

     

       124
       124
       +
             try

     

       125
       125
       +
               if String.length color_str >= 7 && color_str.[0] = '#' then

     

       126
       126
       +
                 let r = int_of_string ("0x" ^ String.sub color_str 1 2) in

     

       127
       127
       +
                 let g = int_of_string ("0x" ^ String.sub color_str 3 2) in

     

       128
       128
       +
                 let b = int_of_string ("0x" ^ String.sub color_str 5 2) in

     

       129
       129
       +
                 let a = if String.length color_str >= 9 then

     

       130
       130
       +
                          int_of_string ("0x" ^ String.sub color_str 7 2)

     

       131
       131
       +
                        else 255 in

     

       132
       132
       +
                 (r, g, b, a)

     

       133
       133
       +
               else

     

       134
       134
       +
                 (255, 0, 0, 255)  (* Default to red if invalid *)

     

       135
       135
       +
             with _ -> 

     

       136
       136
       +
               (255, 0, 0, 255)  (* Default to red on parsing error *)

     

       137
       137
       +
           in

     

       138
       138
       +
           

     

       139
       139
       +
           (* Create a very simple 1x1 PNG with the specified color *)

     

       140
       140
       +
           (* PNG signature *)

     

       141
       141
       +
           let signature = [137; 80; 78; 71; 13; 10; 26; 10] in

     

       142
       142
       +
           

     

       143
       143
       +
           (* Create buffer for the PNG data *)

     

       144
       144
       +
           let buffer = Buffer.create 100 in

     

       145
       145
       +
           

     

       146
       146
       +
           (* PNG signature *)

     

       147
       147
       +
           List.iter (fun b -> Buffer.add_char buffer (Char.chr b)) signature;

     

       148
       148
       +
           

     

       149
       149
       +
           (* IHDR chunk *)

     

       150
       150
       +
           Buffer.add_char buffer (Char.chr 0); (* length - 13 bytes *)

     

       151
       151
       +
           Buffer.add_char buffer (Char.chr 0);

     

       152
       152
       +
           Buffer.add_char buffer (Char.chr 0);

     

       153
       153
       +
           Buffer.add_char buffer (Char.chr 13);

     

       154
       154
       +
           

     

       155
       155
       +
           Buffer.add_string buffer "IHDR";

     

       156
       156
       +
           

     

       157
       157
       +
           (* Width *)

     

       158
       158
       +
           Buffer.add_char buffer (Char.chr ((width lsr 24) land 0xff));

     

       159
       159
       +
           Buffer.add_char buffer (Char.chr ((width lsr 16) land 0xff));

     

       160
       160
       +
           Buffer.add_char buffer (Char.chr ((width lsr 8) land 0xff));

     

       161
       161
       +
           Buffer.add_char buffer (Char.chr (width land 0xff));

     

       162
       162
       +
           

     

       163
       163
       +
           (* Height *)

     

       164
       164
       +
           Buffer.add_char buffer (Char.chr ((height lsr 24) land 0xff));

     

       165
       165
       +
           Buffer.add_char buffer (Char.chr ((height lsr 16) land 0xff));

     

       166
       166
       +
           Buffer.add_char buffer (Char.chr ((height lsr 8) land 0xff));

     

       167
       167
       +
           Buffer.add_char buffer (Char.chr (height land 0xff));

     

       168
       168
       +
           

     

       169
       169
       +
           Buffer.add_char buffer (Char.chr 8); (* Bit depth - 8 bits per channel *)

     

       170
       170
       +
           Buffer.add_char buffer (Char.chr 6); (* Color type - RGBA *)

     

       171
       171
       +
           Buffer.add_char buffer (Char.chr 0); (* Compression method - deflate *)

     

       172
       172
       +
           Buffer.add_char buffer (Char.chr 0); (* Filter method - adaptive filtering *)

     

       173
       173
       +
           Buffer.add_char buffer (Char.chr 0); (* Interlace method - no interlace *)

     

       174
       174
       +
           

     

       175
       175
       +
           (* IHDR CRC - precomputed for simplicity *)

     

       176
       176
       +
           Buffer.add_char buffer (Char.chr 0);

     

       177
       177
       +
           Buffer.add_char buffer (Char.chr 0);

     

       178
       178
       +
           Buffer.add_char buffer (Char.chr 0);

     

       179
       179
       +
           Buffer.add_char buffer (Char.chr 0);

     

       180
       180
       +
           

     

       181
       181
       +
           (* IDAT chunk - simplified for example *)

     

       182
       182
       +
           let pixels_per_row = width * 4 in

     

       183
       183
       +
           let data_size = (1 + pixels_per_row) * height in

     

       184
       184
       +
           

     

       185
       185
       +
           Buffer.add_char buffer (Char.chr ((data_size lsr 24) land 0xff));

     

       186
       186
       +
           Buffer.add_char buffer (Char.chr ((data_size lsr 16) land 0xff));

     

       187
       187
       +
           Buffer.add_char buffer (Char.chr ((data_size lsr 8) land 0xff));

     

       188
       188
       +
           Buffer.add_char buffer (Char.chr (data_size land 0xff));

     

       189
       189
       +
           

     

       190
       190
       +
           Buffer.add_string buffer "IDAT";

     

       191
       191
       +
           

     

       192
       192
       +
           (* Simple zlib header *)

     

       193
       193
       +
           Buffer.add_char buffer (Char.chr 0x78);

     

       194
       194
       +
           Buffer.add_char buffer (Char.chr 0x01);

     

       195
       195
       +
           

     

       196
       196
       +
           (* Raw image data *)

     

       197
       197
       +
           for _ = 0 to height - 1 do

     

       198
       198
       +
             Buffer.add_char buffer (Char.chr 0); (* Filter type 0 - None *)

     

       199
       199
       +
             for _ = 0 to width - 1 do

     

       200
       200
       +
               Buffer.add_char buffer (Char.chr r);

     

       201
       201
       +
               Buffer.add_char buffer (Char.chr g);

     

       202
       202
       +
               Buffer.add_char buffer (Char.chr b);

     

       203
       203
       +
               Buffer.add_char buffer (Char.chr a);

     

       204
       204
       +
             done

     

       205
       205
       +
           done;

     

       206
       206
       +
           

     

       207
       207
       +
           (* Dummy Adler32 checksum *)

     

       208
       208
       +
           Buffer.add_char buffer (Char.chr 0);

     

       209
       209
       +
           Buffer.add_char buffer (Char.chr 0);

     

       210
       210
       +
           Buffer.add_char buffer (Char.chr 0);

     

       211
       211
       +
           Buffer.add_char buffer (Char.chr 0);

     

       212
       212
       +
           

     

       213
       213
       +
           (* IDAT CRC - precomputed for simplicity *)

     

       214
       214
       +
           Buffer.add_char buffer (Char.chr 0);

     

       215
       215
       +
           Buffer.add_char buffer (Char.chr 0);

     

       216
       216
       +
           Buffer.add_char buffer (Char.chr 0);

     

       217
       217
       +
           Buffer.add_char buffer (Char.chr 0);

     

       218
       218
       +
           

     

       219
       219
       +
           (* IEND chunk *)

     

       220
       220
       +
           Buffer.add_char buffer (Char.chr 0);

     

       221
       221
       +
           Buffer.add_char buffer (Char.chr 0);

     

       222
       222
       +
           Buffer.add_char buffer (Char.chr 0);

     

       223
       223
       +
           Buffer.add_char buffer (Char.chr 0);

     

       224
       224
       +
           

     

       225
       225
       +
           Buffer.add_string buffer "IEND";

     

       226
       226
       +
           

     

       227
       227
       +
           (* IEND CRC - precomputed value *)

     

       228
       228
       +
           Buffer.add_char buffer (Char.chr 0xAE);

     

       229
       229
       +
           Buffer.add_char buffer (Char.chr 0x42);

     

       230
       230
       +
           Buffer.add_char buffer (Char.chr 0x60);

     

       231
       231
       +
           Buffer.add_char buffer (Char.chr 0x82);

     

       232
       232
       +
           

     

       233
       233
       +
           Buffer.contents buffer

     

       234
       234
       +
       end

     

       235
       235
       +
       

     

       236
       236
       +
       (* Helper for extracting values from JSON *)

     

       237
       237
       +
       let get_param_int json name default =

     

       238
       238
       +
         match json with

     

       239
       239
       +
         | `Assoc fields -> begin

     

       240
       240
       +
             match List.assoc_opt name fields with

     

       241
       241
       +
             | Some (`Int i) -> begin

     

       242
       242
       +
                 i

     

       243
       243
       +
               end

     

       244
       244
       +
             | Some (`Float f) -> begin

     

       245
       245
       +
                 int_of_float f

     

       246
       246
       +
               end

     

       247
       247
       +
             | _ -> begin

     

       248
       248
       +
                 default

     

       249
       249
       +
               end

     

       250
       250
       +
           end

     

       251
       251
       +
         | _ -> begin

     

       252
       252
       +
             default

     

       253
       253
       +
           end

     

       254
       254
       +
       

     

       255
       255
       +
       let get_param_float json name default =

     

       256
       256
       +
         match json with

     

       257
       257
       +
         | `Assoc fields -> begin

     

       258
       258
       +
             match List.assoc_opt name fields with

     

       259
       259
       +
             | Some (`Int i) -> begin

     

       260
       260
       +
                 float_of_int i

     

       261
       261
       +
               end

     

       262
       262
       +
             | Some (`Float f) -> begin

     

       263
       263
       +
                 f

     

       264
       264
       +
               end

     

       265
       265
       +
             | _ -> begin

     

       266
       266
       +
                 default

     

       267
       267
       +
               end

     

       268
       268
       +
           end

     

       269
       269
       +
         | _ -> begin

     

       270
       270
       +
             default

     

       271
       271
       +
           end

     

       272
       272
       +
       

     

       273
       273
       +
       let get_param_string json name default =

     

       274
       274
       +
         match json with

     

       275
       275
       +
         | `Assoc fields -> begin

     

       276
       276
       +
             match List.assoc_opt name fields with

     

       277
       277
       +
             | Some (`String s) -> begin

     

       278
       278
       +
                 s

     

       279
       279
       +
               end

     

       280
       280
       +
             | _ -> begin

     

       281
       281
       +
                 default

     

       282
       282
       +
               end

     

       283
       283
       +
           end

     

       284
       284
       +
         | _ -> begin

     

       285
       285
       +
             default

     

       286
       286
       +
           end

     

       287
       287
       +
       

     

       288
       288
       +
       (* Create a server *)

     

       289
       289
       +
       let server = create_server 

     

       290
       290
       +
         ~name:"OCaml MCP Multimodal Example" 

     

       291
       291
       +
         ~version:"0.1.0" 

     

       292
       292
       +
         ~protocol_version:"2024-11-05" 

     

       293
       293
       +
         ()

     

       294
       294
       +
       

     

       295
       295
       +
       (* Define startup and shutdown hooks *)

     

       296
       296
       +
       let startup () =

     

       297
       297
       +
         Printf.fprintf stderr "MultimodalServer is starting up!\n";

     

       298
       298
       +
         flush stderr;

     

       299
       299
       +
         Log.info "MultimodalServer is starting up!"

     

       300
       300
       +
       

     

       301
       301
       +
       let shutdown () =

     

       302
       302
       +
         Printf.fprintf stderr "MultimodalServer is shutting down. Goodbye!\n";

     

       303
       303
       +
         flush stderr;

     

       304
       304
       +
         Log.info "MultimodalServer is shutting down. Goodbye!"

     

       305
       305
       +
       

     

       306
       306
       +
       (* Register the hooks *)

     

       307
       307
       +
       let () =

     

       308
       308
       +
         set_startup_hook server startup;

     

       309
       309
       +
         set_shutdown_hook server shutdown

     

       310
       310
       +
       

     

       311
       311
       +
       (* Define and register a multimodal tool *)

     

       312
       312
       +
       let _ = add_tool server

     

       313
       313
       +
         ~name:"generate_multimodal_response"

     

       314
       314
       +
         ~description:"Generates a response with text, image and audio content"

     

       315
       315
       +
         ~schema_properties:[

     

       316
       316
       +
           ("message", "string", "The text message to include");

     

       317
       317
       +
           ("color", "string", "Color for the image (hex format #RRGGBB)");

     

       318
       318
       +
           ("frequency", "integer", "Frequency for the audio tone in Hz");

     

       319
       319
       +
         ]

     

       320
       320
       +
         ~schema_required:["message"]

     

       321
       321
       +
         (fun args ->

     

       322
       322
       +
           try

     

       323
       323
       +
             let message = get_param_string args "message" "Hello, multimodal world!" in

     

       324
       324
       +
             let color = get_param_string args "color" "#FF0000" in

     

       325
       325
       +
             let frequency = get_param_int args "frequency" 440 in

     

       326
       326
       +
             

     

       327
       327
       +
             (* Generate image *)

     

       328
       328
       +
             let image_data = ImageGenerator.generate_simple_image 100 100 color in

     

       329
       329
       +
             let image_base64 = Base64.encode image_data in

     

       330
       330
       +
             

     

       331
       331
       +
             (* Generate audio *)

     

       332
       332
       +
             let audio_data = AudioGenerator.generate_sine_wave 

     

       333
       333
       +
               ~frequency:(float_of_int frequency) 

     

       334
       334
       +
               ~duration:1.0 

     

       335
       335
       +
               ~sample_rate:8000 

     

       336
       336
       +
               ~amplitude:0.8 in

     

       337
       337
       +
             let audio_base64 = Base64.encode audio_data in

     

       338
       338
       +
             

     

       339
       339
       +
             (* Create a response with text, image and audio content *)

     

       340
       340
       +
             CallToolResult.yojson_of_t CallToolResult.{

     

       341
       341
       +
               content = [

     

       342
       342
       +
                 Text TextContent.{ 

     

       343
       343
       +
                   text = message;

     

       344
       344
       +
                   annotations = None 

     

       345
       345
       +
                 };

     

       346
       346
       +
                 Image ImageContent.{ 

     

       347
       347
       +
                   data = image_base64; 

     

       348
       348
       +
                   mime_type = "image/png"; 

     

       349
       349
       +
                   annotations = None 

     

       350
       350
       +
                 };

     

       351
       351
       +
                 Audio AudioContent.{ 

     

       352
       352
       +
                   data = audio_base64; 

     

       353
       353
       +
                   mime_type = "audio/wav"; 

     

       354
       354
       +
                   annotations = None 

     

       355
       355
       +
                 }

     

       356
       356
       +
               ];

     

       357
       357
       +
               is_error = false;

     

       358
       358
       +
               meta = None

     

       359
       359
       +
             }

     

       360
       360
       +
           with

     

       361
       361
       +
           | Failure msg ->

     

       362
       362
       +
             Log.error (Printf.sprintf "Error in multimodal tool: %s" msg);

     

       363
       363
       +
             CallToolResult.yojson_of_t CallToolResult.{

     

       364
       364
       +
               content = [

     

       365
       365
       +
                 Text TextContent.{ 

     

       366
       366
       +
                   text = Printf.sprintf "Error: %s" msg; 

     

       367
       367
       +
                   annotations = None 

     

       368
       368
       +
                 }

     

       369
       369
       +
               ];

     

       370
       370
       +
               is_error = true;

     

       371
       371
       +
               meta = None

     

       372
       372
       +
             }

     

       373
       373
       +
         )

     

       374
       374
       +
       

     

       375
       375
       +
       (* Define and register a multimodal prompt *)

     

       376
       376
       +
       let _ = add_prompt server

     

       377
       377
       +
         ~name:"multimodal-prompt"

     

       378
       378
       +
         ~description:"A prompt that includes text, image, and audio"

     

       379
       379
       +
         ~arguments:[

     

       380
       380
       +
           ("message", Some "Text message to include", true);

     

       381
       381
       +
           ("color", Some "Color for the image (hex format #RRGGBB)", false);

     

       382
       382
       +
           ("frequency", Some "Frequency for the audio tone in Hz", false);

     

       383
       383
       +
         ]

     

       384
       384
       +
         (fun args ->

     

       385
       385
       +
           (* Parse parameters with defaults *)

     

       386
       386
       +
           let message = 

     

       387
       387
       +
             try List.assoc "message" args

     

       388
       388
       +
             with Not_found -> "Hello, multimodal world!"

     

       389
       389
       +
           in

     

       390
       390
       +
           

     

       391
       391
       +
           let color = 

     

       392
       392
       +
             try List.assoc "color" args

     

       393
       393
       +
             with Not_found -> "#0000FF"

     

       394
       394
       +
           in

     

       395
       395
       +
           

     

       396
       396
       +
           let frequency = 

     

       397
       397
       +
             try int_of_string (List.assoc "frequency" args)

     

       398
       398
       +
             with _ -> 440

     

       399
       399
       +
           in

     

       400
       400
       +
           

     

       401
       401
       +
           (* Generate image *)

     

       402
       402
       +
           let image_data = ImageGenerator.generate_simple_image 100 100 color in

     

       403
       403
       +
           let image_base64 = Base64.encode image_data in

     

       404
       404
       +
           

     

       405
       405
       +
           (* Generate audio *)

     

       406
       406
       +
           let audio_data = AudioGenerator.generate_sine_wave 

     

       407
       407
       +
             ~frequency:(float_of_int frequency) 

     

       408
       408
       +
             ~duration:1.0 

     

       409
       409
       +
             ~sample_rate:8000 

     

       410
       410
       +
             ~amplitude:0.8 in

     

       411
       411
       +
           let audio_base64 = Base64.encode audio_data in

     

       412
       412
       +
           

     

       413
       413
       +
           (* Create a multimodal prompt *)

     

       414
       414
       +
           [

     

       415
       415
       +
             Prompt.{ 

     

       416
       416
       +
               role = `User; 

     

       417
       417
       +
               content = make_text_content "Here's a multimodal message with text, image, and audio:"

     

       418
       418
       +
             };

     

       419
       419
       +
             Prompt.{ 

     

       420
       420
       +
               role = `User; 

     

       421
       421
       +
               content = make_text_content message

     

       422
       422
       +
             };

     

       423
       423
       +
             Prompt.{ 

     

       424
       424
       +
               role = `User;

     

       425
       425
       +
               content = make_image_content image_base64 "image/png"

     

       426
       426
       +
             };

     

       427
       427
       +
             Prompt.{ 

     

       428
       428
       +
               role = `User;

     

       429
       429
       +
               content = make_audio_content audio_base64 "audio/wav"

     

       430
       430
       +
             };

     

       431
       431
       +
             Prompt.{ 

     

       432
       432
       +
               role = `Assistant; 

     

       433
       433
       +
               content = make_text_content "I've received your multimodal message with text, image, and audio."

     

       434
       434
       +
             }

     

       435
       435
       +
           ]

     

       436
       436
       +
         )

     

       437
       437
       +
       

     

       438
       438
       +
       (* Also register a resource prompt example *)

     

       439
       439
       +
       let _ = add_prompt server

     

       440
       440
       +
         ~name:"resource-prompt"

     

       441
       441
       +
         ~description:"A prompt that includes embedded resources"

     

       442
       442
       +
         ~arguments:[

     

       443
       443
       +
           ("resource_id", Some "ID of the resource to include", true);

     

       444
       444
       +
         ]

     

       445
       445
       +
         (fun args ->

     

       446
       446
       +
           (* Sample resource texts *)

     

       447
       447
       +
           let resources = [

     

       448
       448
       +
             ("doc1", "This is the content of document 1.");

     

       449
       449
       +
             ("doc2", "Document 2 contains important information about OCaml.");

     

       450
       450
       +
             ("doc3", "Document 3 explains the MCP protocol in detail.");

     

       451
       451
       +
           ] in

     

       452
       452
       +
           

     

       453
       453
       +
           (* Get the requested resource *)

     

       454
       454
       +
           let resource_id = 

     

       455
       455
       +
             try List.assoc "resource_id" args

     

       456
       456
       +
             with Not_found -> "doc1"

     

       457
       457
       +
           in

     

       458
       458
       +
           

     

       459
       459
       +
           (* Find the resource content *)

     

       460
       460
       +
           let resource_content = 

     

       461
       461
       +
             try List.assoc resource_id resources

     

       462
       462
       +
             with Not_found -> Printf.sprintf "Resource '%s' not found" resource_id

     

       463
       463
       +
           in

     

       464
       464
       +
           

     

       465
       465
       +
           (* Create a prompt with embedded resource *)

     

       466
       466
       +
           [

     

       467
       467
       +
             Prompt.{ 

     

       468
       468
       +
               role = `User; 

     

       469
       469
       +
               content = make_text_content (Printf.sprintf "Here's the content of resource %s:" resource_id)

     

       470
       470
       +
             };

     

       471
       471
       +
             Prompt.{ 

     

       472
       472
       +
               role = `User;

     

       473
       473
       +
               content = make_text_resource_content (Printf.sprintf "resource://%s" resource_id) resource_content ~mime_type:"text/plain" ()

     

       474
       474
       +
             };

     

       475
       475
       +
             Prompt.{ 

     

       476
       476
       +
               role = `User; 

     

       477
       477
       +
               content = make_text_content "Please analyze this content."

     

       478
       478
       +
             };

     

       479
       479
       +
             Prompt.{ 

     

       480
       480
       +
               role = `Assistant; 

     

       481
       481
       +
               content = make_text_content "I'll analyze the resource content for you."

     

       482
       482
       +
             }

     

       483
       483
       +
           ]

     

       484
       484
       +
         )

     

       485
       485
       +
       

     

       486
       486
       +
       (* Main function *)

     

       487
       487
       +
       let () =

     

       488
       488
       +
         (* Parse command line arguments *)

     

       489
       489
       +
         let transport_type = ref Stdio in

     

       490
       490
       +
         let args = [

     

       491
       491
       +
           ("--http", Arg.Unit (fun () -> transport_type := Http), 

     

       492
       492
       +
            "Start server with HTTP transport (default is stdio)");

     

       493
       493
       +
         ] in

     

       494
       494
       +
         let usage_msg = "Usage: multimodal_example [--http]" in

     

       495
       495
       +
         Arg.parse args (fun _ -> ()) usage_msg;

     

       496
       496
       +
       

     

       497
       497
       +
         (* Configure the server with appropriate capabilities *)

     

       498
       498
       +
         let server = configure_server server ~with_tools:true ~with_resources:false ~with_prompts:true () in

     

       499
       499
       +
         

     

       500
       500
       +
         (* Create and start MCP server with the selected transport *)

     

       501
       501
       +
         let mcp_server = create ~server ~transport:!transport_type () in

     

       502
       502
       +
         start mcp_server;

+13 -2

bin/resource_template_example.ml

···

       1
       1
        
       open Mcp

     

       2
       2
        
       open Mcp_sdk

     

       3
       3
       +
       open Mcp_server

     

       3
       4
        
       

     

       4
       5
        
       (* Helper for extracting string value from JSON *)

     

       5
       6
        
       let get_string_param json name =

     
···

       164
       165
        
       

     

       165
       166
        
       (* Main function *)

     

       166
       167
        
       let () =

     

       168
       168
       +
         (* Parse command line arguments *)

     

       169
       169
       +
         let transport_type = ref Stdio in

     

       170
       170
       +
         let args = [

     

       171
       171
       +
           ("--http", Arg.Unit (fun () -> transport_type := Http), 

     

       172
       172
       +
            "Start server with HTTP transport (default is stdio)");

     

       173
       173
       +
         ] in

     

       174
       174
       +
         let usage_msg = "Usage: resource_template_example [--http]" in

     

       175
       175
       +
         Arg.parse args (fun _ -> ()) usage_msg;

     

       176
       176
       +
       

     

       167
       177
        
         (* Instead of printing directly to stdout which messes up the JSON-RPC protocol,

     

       168
       178
        
            use the logging system which sends output to stderr *)

     

       169
       179
        
         Log.info "Starting ResourceTemplateServer...";

     
···

       171
       181
        
         (* Configure the server with appropriate capabilities *)

     

       172
       182
        
         ignore (configure_server server ());

     

       173
       183
        
         

     

       174
       174
       -
         (* Run the server *)

     

       175
       175
       -
         run_server server
     

       184
       184
       +
         (* Create and start MCP server with the selected transport *)

     

       185
       185
       +
         let mcp_server = create ~server ~transport:!transport_type () in

     

       186
       186
       +
         start mcp_server

lib/dune

···

       8
       8
        
         (libraries mcp jsonrpc unix yojson)

     

       9
       9
        
         (modules mcp_sdk)

     

       10
       10
        
         (flags (:standard -w -67 -w -27 -w -32)))

     

       11
       11
       +
       

     

       12
       12
       +
       (library

     

       13
       13
       +
         (name mcp_server)

     

       14
       14
       +
         (libraries mcp mcp_sdk jsonrpc unix yojson)

     

       15
       15
       +
         (modules mcp_server)

     

       16
       16
       +
         (flags (:standard -w -67 -w -27 -w -32 -w -8 -w -11 -w -33)))

+78 -7

lib/mcp_sdk.ml

···

       16
       16
        
       

     

       17
       17
        
         let log level msg =

     

       18
       18
        
           Printf.eprintf "[%s] %s\n" (string_of_level level) msg;

     

       19
       19
       -
           flush stderr;

     

       20
       20
       -
           Printf.printf "[%s] %s\n" (string_of_level level) msg;

     

       21
       21
       -
           flush stdout

     

       19
       19
       +
           flush stderr

     

       22
       20
        
       

     

       23
       21
        
         let debug = log Debug

     

       24
       22
        
         let info = log Info

     
···

       135
       133
        
         let create_argument ~name ?description ?(required=false) () =

     

       136
       134
        
           { name; description; required }

     

       137
       135
        
       

     

       136
       136
       +
         let yojson_of_message { role; content } =

     

       137
       137
       +
           `Assoc [

     

       138
       138
       +
             ("role", Role.yojson_of_t role);

     

       139
       139
       +
             ("content", yojson_of_content content);

     

       140
       140
       +
           ]

     

       141
       141
       +
       

     

       142
       142
       +
         (* This function must match the structure expected by the PromptMessage module in mcp.ml *)

     

       143
       143
       +
         let message_of_yojson json =

     

       144
       144
       +
           match json with

     

       145
       145
       +
           | `Assoc fields -> begin

     

       146
       146
       +
               let role = match List.assoc_opt "role" fields with

     

       147
       147
       +
                 | Some json -> begin

     

       148
       148
       +
                     Role.t_of_yojson json

     

       149
       149
       +
                   end

     

       150
       150
       +
                 | None -> begin

     

       151
       151
       +
                     raise (Json.Of_json ("Missing role field", `Assoc fields))

     

       152
       152
       +
                   end

     

       153
       153
       +
               in

     

       154
       154
       +
               let content = match List.assoc_opt "content" fields with

     

       155
       155
       +
                 | Some json -> begin

     

       156
       156
       +
                     content_of_yojson json

     

       157
       157
       +
                   end

     

       158
       158
       +
                 | None -> begin

     

       159
       159
       +
                     raise (Json.Of_json ("Missing content field", `Assoc fields))

     

       160
       160
       +
                   end

     

       161
       161
       +
               in

     

       162
       162
       +
               { role; content }

     

       163
       163
       +
             end

     

       164
       164
       +
           | j -> begin

     

       165
       165
       +
               raise (Json.Of_json ("Expected object for PromptMessage", j))

     

       166
       166
       +
             end

     

       167
       167
       +
       

     

       138
       168
        
         let to_json prompt =

     

       139
       169
        
           let assoc = [

     

       140
       170
        
             ("name", `String prompt.name);

     
···

       171
       201
        
       let make_text_content text =

     

       172
       202
        
         Text (TextContent.{ text; annotations = None })

     

       173
       203
        
       

     

       204
       204
       +
       let make_text_content_with_annotations text annotations =

     

       205
       205
       +
         Text (TextContent.{ text; annotations = Some annotations })

     

       206
       206
       +
       

     

       207
       207
       +
       let make_image_content data mime_type =

     

       208
       208
       +
         Image (ImageContent.{ data; mime_type; annotations = None })

     

       209
       209
       +
       

     

       210
       210
       +
       let make_image_content_with_annotations data mime_type annotations =

     

       211
       211
       +
         Image (ImageContent.{ data; mime_type; annotations = Some annotations })

     

       212
       212
       +
       

     

       213
       213
       +
       let make_audio_content data mime_type =

     

       214
       214
       +
         Audio (AudioContent.{ data; mime_type; annotations = None })

     

       215
       215
       +
       

     

       216
       216
       +
       let make_audio_content_with_annotations data mime_type annotations =

     

       217
       217
       +
         Audio (AudioContent.{ data; mime_type; annotations = Some annotations })

     

       218
       218
       +
       

     

       219
       219
       +
       let make_text_resource_content uri text ?mime_type () =

     

       220
       220
       +
         Resource (EmbeddedResource.{

     

       221
       221
       +
           resource = `Text TextResourceContents.{ uri; text; mime_type };

     

       222
       222
       +
           annotations = None

     

       223
       223
       +
         })

     

       224
       224
       +
       

     

       225
       225
       +
       let make_blob_resource_content uri blob ?mime_type () =

     

       226
       226
       +
         Resource (EmbeddedResource.{

     

       227
       227
       +
           resource = `Blob BlobResourceContents.{ uri; blob; mime_type };

     

       228
       228
       +
           annotations = None

     

       229
       229
       +
         })

     

       230
       230
       +
       

     

       174
       231
        
       let make_tool_schema properties required =

     

       175
       232
        
         let props = List.map (fun (name, schema_type, description) ->

     

       176
       233
        
           (name, `Assoc [

     
···

       352
       409
        
       let set_shutdown_hook server hook =

     

       353
       410
        
         server.shutdown_hook <- Some hook

     

       354
       411
        
       

     

       355
       355
       -
       (* Run the server *)

     

       412
       412
       +
       (* Transport type for server *)

     

       413
       413
       +
       type transport_type = 

     

       414
       414
       +
         | Stdio  (* Read/write to stdin/stdout *)

     

       415
       415
       +
         | Http   (* HTTP server - to be implemented *)

     

       416
       416
       +
       

     

       417
       417
       +
       (* Run server with stdio transport *)

     

       356
       418
        
       let run_server server =

     

       357
       419
        
         (* Setup *)

     

       358
       420
        
         Printexc.record_backtrace true;

     

       359
       359
       -
         set_binary_mode_out stdout false;

     

       360
       421
        
         

     

       361
       361
       -
         Log.info (Printf.sprintf "%s server started" server.name);

     

       422
       422
       +
         Log.info (Printf.sprintf "%s server starting" server.name);

     

       362
       423
        
         Log.debug (Printf.sprintf "Protocol version: %s" server.protocol_version);

     

       363
       424
        
         Log.debug (Printf.sprintf "Server info: %s v%s" server.name server.version);

     

       364
       425
        
         

     
···

       371
       432
        
          | Some hook -> hook ()

     

       372
       433
        
          | None -> ());

     

       373
       434
        
         

     

       374
       374
       -
         Log.info "Server initialized and ready."
     

       435
       435
       +
         (* This function will be replaced by a full implementation in the mcp_server module *)

     

       436
       436
       +
         Log.info "Server initialized and ready."

     

       437
       437
       +
       

     

       438
       438
       +
       (* Placeholder for running server with different transports *)

     

       439
       439
       +
       let run_server_with_transport server transport =

     

       440
       440
       +
         match transport with

     

       441
       441
       +
         | Http ->

     

       442
       442
       +
             Log.info "HTTP server not implemented in this version, using stdio instead";

     

       443
       443
       +
             run_server server

     

       444
       444
       +
         | Stdio ->

     

       445
       445
       +
             run_server server

+18 -1

lib/mcp_sdk.mli

···

       86
       86
        
       

     

       87
       87
        
         val create : name:string -> ?description:string -> ?arguments:argument list -> handler:handler -> unit -> t

     

       88
       88
        
         val create_argument : name:string -> ?description:string -> ?required:bool -> unit -> argument

     

       89
       89
       +
         val yojson_of_message : message -> Json.t

     

       90
       90
       +
         val message_of_yojson : Json.t -> message

     

       89
       91
        
         val to_json : t -> Json.t

     

       90
       92
        
       end

     

       91
       93
        
       

     
···

       139
       141
        
       (** Set shutdown hook *)

     

       140
       142
        
       val set_shutdown_hook : server -> (unit -> unit) -> unit

     

       141
       143
        
       

     

       142
       142
       -
       (** Run the server *)

     

       144
       144
       +
       (** Run the server using stdio transport (legacy method) *)

     

       143
       145
        
       val run_server : server -> unit

     

       144
       146
        
       

     

       147
       147
       +
       (** Transport type for the server *)

     

       148
       148
       +
       type transport_type = 

     

       149
       149
       +
         | Stdio  (** Read/write to stdin/stdout *)

     

       150
       150
       +
         | Http   (** HTTP server - to be implemented *)

     

       151
       151
       +
       

     

       152
       152
       +
       (** Create and start a server with the specified transport *)

     

       153
       153
       +
       val run_server_with_transport : server -> transport_type -> unit

     

       154
       154
       +
       

     

       145
       155
        
       (** Helper functions for creating common objects *)

     

       146
       156
        
       val make_text_content : string -> content

     

       157
       157
       +
       val make_text_content_with_annotations : string -> Annotated.annotation -> content

     

       158
       158
       +
       val make_image_content : string -> string -> content

     

       159
       159
       +
       val make_image_content_with_annotations : string -> string -> Annotated.annotation -> content

     

       160
       160
       +
       val make_audio_content : string -> string -> content

     

       161
       161
       +
       val make_audio_content_with_annotations : string -> string -> Annotated.annotation -> content

     

       162
       162
       +
       val make_text_resource_content : string -> string -> ?mime_type:string -> unit -> content

     

       163
       163
       +
       val make_blob_resource_content : string -> string -> ?mime_type:string -> unit -> content

     

       147
       164
        
       val make_tool_schema : (string * string * string) list -> string list -> Json.t

+533

lib/mcp_server.ml

···

       1
       1
       +
       open Mcp

     

       2
       2
       +
       open Mcp_sdk

     

       3
       3
       +
       

     

       4
       4
       +
       (* MCP Server module for handling JSON-RPC communication *)

     

       5
       5
       +
       

     

       6
       6
       +
       (** Server types *)

     

       7
       7
       +
       type transport_type = 

     

       8
       8
       +
         | Stdio  (* Read/write to stdin/stdout *)

     

       9
       9
       +
         | Http   (* HTTP server - to be implemented *)

     

       10
       10
       +
       

     

       11
       11
       +
       type t = {

     

       12
       12
       +
         server: Mcp_sdk.server;

     

       13
       13
       +
         transport: transport_type;

     

       14
       14
       +
         mutable running: bool;

     

       15
       15
       +
       }

     

       16
       16
       +
       

     

       17
       17
       +
       (** Process a single message *)

     

       18
       18
       +
       let process_message server message =

     

       19
       19
       +
         try

     

       20
       20
       +
           Log.debug "Parsing message as JSONRPC message...";

     

       21
       21
       +
           match JSONRPCMessage.t_of_yojson message with

     

       22
       22
       +
           | JSONRPCMessage.Request req -> begin

     

       23
       23
       +
               Log.debug (Printf.sprintf "Received request with method: %s" req.method_);

     

       24
       24
       +
               match req.method_ with

     

       25
       25
       +
               | "initialize" -> begin

     

       26
       26
       +
                   Log.debug "Processing initialize request";

     

       27
       27
       +
                   let result = match req.params with

     

       28
       28
       +
                   | Some params -> begin

     

       29
       29
       +
                       Log.debug "Parsing initialize request params...";

     

       30
       30
       +
                       let req_params = Initialize.Request.t_of_yojson params in

     

       31
       31
       +
                       Log.debug (Printf.sprintf "Client info: %s v%s" 

     

       32
       32
       +
                                    req_params.client_info.name 

     

       33
       33
       +
                                    req_params.client_info.version);

     

       34
       34
       +
                       Log.debug (Printf.sprintf "Client protocol version: %s" req_params.protocol_version);

     

       35
       35
       +
                       

     

       36
       36
       +
                       (* Check protocol version compatibility *)

     

       37
       37
       +
                       if req_params.protocol_version <> server.protocol_version then begin

     

       38
       38
       +
                         Log.debug (Printf.sprintf "Protocol version mismatch: client=%s server=%s" 

     

       39
       39
       +
                                     req_params.protocol_version server.protocol_version);

     

       40
       40
       +
                       end;

     

       41
       41
       +
                       

     

       42
       42
       +
                       Initialize.Result.create

     

       43
       43
       +
                         ~capabilities:server.capabilities

     

       44
       44
       +
                         ~server_info:Implementation.{ name = server.name; version = server.version }

     

       45
       45
       +
                         ~protocol_version:server.protocol_version

     

       46
       46
       +
                         ?instructions:(Some "MCP Server") (* TODO: Allow customization *)

     

       47
       47
       +
                         ()

     

       48
       48
       +
                     end

     

       49
       49
       +
                   | None -> begin

     

       50
       50
       +
                       Log.error "Missing params for initialize request";

     

       51
       51
       +
                       Initialize.Result.create

     

       52
       52
       +
                         ~capabilities:server.capabilities

     

       53
       53
       +
                         ~server_info:Implementation.{ name = server.name; version = server.version }

     

       54
       54
       +
                         ~protocol_version:server.protocol_version

     

       55
       55
       +
                         ()

     

       56
       56
       +
                     end

     

       57
       57
       +
                   in

     

       58
       58
       +
                   Some (create_response ~id:req.id ~result:(Initialize.Result.yojson_of_t result))

     

       59
       59
       +
                 end

     

       60
       60
       +
                

     

       61
       61
       +
               | "tools/list" -> begin

     

       62
       62
       +
                   Log.debug "Processing tools/list request";

     

       63
       63
       +
                   let tools_json = List.map Mcp_sdk.Tool.to_json server.tools in

     

       64
       64
       +
                   let result = `Assoc [("tools", `List tools_json)] in

     

       65
       65
       +
                   Some (create_response ~id:req.id ~result)

     

       66
       66
       +
                 end

     

       67
       67
       +
                

     

       68
       68
       +
               | "tools/call" -> begin

     

       69
       69
       +
                   Log.debug "Processing tools/call request";

     

       70
       70
       +
                   match req.params with

     

       71
       71
       +
                   | Some (`Assoc params) -> begin

     

       72
       72
       +
                       let name = match List.assoc_opt "name" params with

     

       73
       73
       +
                       | Some (`String name) -> begin

     

       74
       74
       +
                           Log.debug (Printf.sprintf "Tool name: %s" name);

     

       75
       75
       +
                           name

     

       76
       76
       +
                         end

     

       77
       77
       +
                       | _ -> begin

     

       78
       78
       +
                           Log.error "Missing or invalid 'name' parameter in tool call";

     

       79
       79
       +
                           failwith "Missing or invalid 'name' parameter"

     

       80
       80
       +
                         end

     

       81
       81
       +
                       in

     

       82
       82
       +
                       let args = match List.assoc_opt "arguments" params with

     

       83
       83
       +
                       | Some args -> begin

     

       84
       84
       +
                           Log.debug (Printf.sprintf "Tool arguments: %s" (Yojson.Safe.to_string args));

     

       85
       85
       +
                           args

     

       86
       86
       +
                         end

     

       87
       87
       +
                       | _ -> begin

     

       88
       88
       +
                           Log.debug "No arguments provided for tool call, using empty object";

     

       89
       89
       +
                           `Assoc [] (* Empty arguments is valid *)

     

       90
       90
       +
                         end

     

       91
       91
       +
                       in

     

       92
       92
       +
                       let progress_token = req.progress_token in

     

       93
       93
       +
                       

     

       94
       94
       +
                       (* Find the tool *)

     

       95
       95
       +
                       let tool_opt = List.find_opt (fun t -> t.Mcp_sdk.Tool.name = name) server.tools in

     

       96
       96
       +
                       match tool_opt with

     

       97
       97
       +
                       | Some tool -> begin

     

       98
       98
       +
                           Log.debug (Printf.sprintf "Found tool: %s" name);

     

       99
       99
       +
                           let ctx = Mcp_sdk.Context.create 

     

       100
       100
       +
                               ?request_id:(Some req.id) 

     

       101
       101
       +
                               ~lifespan_context:server.lifespan_context 

     

       102
       102
       +
                               () 

     

       103
       103
       +
                           in

     

       104
       104
       +
                           ctx.progress_token <- progress_token;

     

       105
       105
       +
                           

     

       106
       106
       +
                           (* Call the handler *)

     

       107
       107
       +
                           let result = match tool.handler ctx args with

     

       108
       108
       +
                           | Ok json -> begin

     

       109
       109
       +
                               `Assoc [

     

       110
       110
       +
                                 ("content", `List [Mcp.yojson_of_content (Text (TextContent.{ 

     

       111
       111
       +
                                   text = Yojson.Safe.to_string json;

     

       112
       112
       +
                                   annotations = None 

     

       113
       113
       +
                                 }))]);

     

       114
       114
       +
                                 ("isError", `Bool false)

     

       115
       115
       +
                               ]

     

       116
       116
       +
                             end

     

       117
       117
       +
                           | Error err -> begin

     

       118
       118
       +
                               `Assoc [

     

       119
       119
       +
                                 ("content", `List [Mcp.yojson_of_content (Text (TextContent.{ 

     

       120
       120
       +
                                   text = err;

     

       121
       121
       +
                                   annotations = None 

     

       122
       122
       +
                                 }))]);

     

       123
       123
       +
                                 ("isError", `Bool true)

     

       124
       124
       +
                               ]

     

       125
       125
       +
                             end

     

       126
       126
       +
                           in

     

       127
       127
       +
                           Some (create_response ~id:req.id ~result)

     

       128
       128
       +
                         end

     

       129
       129
       +
                       | None -> begin

     

       130
       130
       +
                           Log.error (Printf.sprintf "Tool not found: %s" name);

     

       131
       131
       +
                           let error_content = TextContent.{

     

       132
       132
       +
                             text = Printf.sprintf "Unknown tool: %s" name;

     

       133
       133
       +
                             annotations = None

     

       134
       134
       +
                           } in

     

       135
       135
       +
                           let result = `Assoc [

     

       136
       136
       +
                             ("content", `List [Mcp.yojson_of_content (Text error_content)]);

     

       137
       137
       +
                             ("isError", `Bool true)

     

       138
       138
       +
                           ] in

     

       139
       139
       +
                           Some (create_response ~id:req.id ~result)

     

       140
       140
       +
                         end

     

       141
       141
       +
                     end

     

       142
       142
       +
                   | _ -> begin

     

       143
       143
       +
                       Log.error "Invalid params format for tools/call";

     

       144
       144
       +
                       Some (create_error ~id:req.id ~code:ErrorCode.invalid_params ~message:"Invalid params for tools/call" ())

     

       145
       145
       +
                     end

     

       146
       146
       +
                 end

     

       147
       147
       +
                   

     

       148
       148
       +
               | "resources/list" -> begin

     

       149
       149
       +
                   Log.debug "Processing resources/list request";

     

       150
       150
       +
                   if server.resources <> [] then begin

     

       151
       151
       +
                     let resources_json = List.map Mcp_sdk.Resource.to_json server.resources in

     

       152
       152
       +
                     let result = `Assoc [("resources", `List resources_json)] in

     

       153
       153
       +
                     Some (create_response ~id:req.id ~result)

     

       154
       154
       +
                   end else begin

     

       155
       155
       +
                     Some (create_error ~id:req.id ~code:ErrorCode.method_not_found ~message:"Resources not supported" ())

     

       156
       156
       +
                   end

     

       157
       157
       +
                 end

     

       158
       158
       +
                  

     

       159
       159
       +
               | "prompts/list" -> begin

     

       160
       160
       +
                   Log.debug "Processing prompts/list request";

     

       161
       161
       +
                   if server.prompts <> [] then begin

     

       162
       162
       +
                     let prompts_json = List.map Mcp_sdk.Prompt.to_json server.prompts in

     

       163
       163
       +
                     let result = `Assoc [("prompts", `List prompts_json)] in

     

       164
       164
       +
                     Some (create_response ~id:req.id ~result)

     

       165
       165
       +
                   end else begin

     

       166
       166
       +
                     Some (create_error ~id:req.id ~code:ErrorCode.method_not_found ~message:"Prompts not supported" ())

     

       167
       167
       +
                   end

     

       168
       168
       +
                 end

     

       169
       169
       +
                  

     

       170
       170
       +
               | "prompts/get" -> begin

     

       171
       171
       +
                   Log.debug "Processing prompts/get request";

     

       172
       172
       +
                   if server.prompts <> [] then begin

     

       173
       173
       +
                     match req.params with

     

       174
       174
       +
                     | Some (`Assoc params) -> begin

     

       175
       175
       +
                         (* Extract prompt name *)

     

       176
       176
       +
                         let name = match List.assoc_opt "name" params with

     

       177
       177
       +
                         | Some (`String name) -> begin 

     

       178
       178
       +
                             Log.debug (Printf.sprintf "Prompt name: %s" name);

     

       179
       179
       +
                             name

     

       180
       180
       +
                           end

     

       181
       181
       +
                         | _ -> begin

     

       182
       182
       +
                             Log.error "Missing or invalid 'name' parameter in prompt request";

     

       183
       183
       +
                             failwith "Missing or invalid 'name' parameter"

     

       184
       184
       +
                           end

     

       185
       185
       +
                         in

     

       186
       186
       +
                         

     

       187
       187
       +
                         (* Extract arguments if any *)

     

       188
       188
       +
                         let arguments = match List.assoc_opt "arguments" params with

     

       189
       189
       +
                         | Some (`Assoc args) -> begin

     

       190
       190
       +
                             Log.debug (Printf.sprintf "Prompt arguments: %s" (Yojson.Safe.to_string (`Assoc args)));

     

       191
       191
       +
                             List.map (fun (k, v) -> 

     

       192
       192
       +
                               match v with 

     

       193
       193
       +
                               | `String s -> begin (k, s) end

     

       194
       194
       +
                               | _ -> begin (k, Yojson.Safe.to_string v) end

     

       195
       195
       +
                             ) args

     

       196
       196
       +
                           end

     

       197
       197
       +
                         | _ -> begin

     

       198
       198
       +
                             []

     

       199
       199
       +
                           end

     

       200
       200
       +
                         in

     

       201
       201
       +
                         

     

       202
       202
       +
                         (* Find the prompt *)

     

       203
       203
       +
                         let prompt_opt = List.find_opt (fun p -> p.Mcp_sdk.Prompt.name = name) server.prompts in

     

       204
       204
       +
                         match prompt_opt with

     

       205
       205
       +
                         | Some prompt -> begin

     

       206
       206
       +
                             Log.debug (Printf.sprintf "Found prompt: %s" name);

     

       207
       207
       +
                             let ctx = Mcp_sdk.Context.create 

     

       208
       208
       +
                                 ?request_id:(Some req.id) 

     

       209
       209
       +
                                 ~lifespan_context:server.lifespan_context 

     

       210
       210
       +
                                 () 

     

       211
       211
       +
                             in

     

       212
       212
       +
                             

     

       213
       213
       +
                             (* Call the prompt handler *)

     

       214
       214
       +
                             match prompt.handler ctx arguments with

     

       215
       215
       +
                             | Ok messages -> begin

     

       216
       216
       +
                                 Log.debug (Printf.sprintf "Prompt handler returned %d messages" (List.length messages));

     

       217
       217
       +
                                 

     

       218
       218
       +
                                 (* Important: We need to directly use yojson_of_message which preserves MIME types *)

     

       219
       219
       +
                                 let messages_json = List.map Prompt.yojson_of_message messages in

     

       220
       220
       +
                                 

     

       221
       221
       +
                                 (* Debug output *)

     

       222
       222
       +
                                 Log.debug (Printf.sprintf "Messages JSON: %s" (Yojson.Safe.to_string (`List messages_json)));

     

       223
       223
       +
                                 

     

       224
       224
       +
                                 (* Verify one message if available to check structure *)

     

       225
       225
       +
                                 if List.length messages > 0 then begin

     

       226
       226
       +
                                   let first_msg = List.hd messages in

     

       227
       227
       +
                                   let content_debug = match first_msg.content with

     

       228
       228
       +
                                   | Text t -> begin

     

       229
       229
       +
                                       Printf.sprintf "Text content: %s" t.text

     

       230
       230
       +
                                     end

     

       231
       231
       +
                                   | Image i -> begin

     

       232
       232
       +
                                       Printf.sprintf "Image content (mime: %s)" i.mime_type

     

       233
       233
       +
                                     end

     

       234
       234
       +
                                   | Audio a -> begin

     

       235
       235
       +
                                       Printf.sprintf "Audio content (mime: %s)" a.mime_type

     

       236
       236
       +
                                     end

     

       237
       237
       +
                                   | Resource r -> begin

     

       238
       238
       +
                                       "Resource content"

     

       239
       239
       +
                                     end

     

       240
       240
       +
                                   in

     

       241
       241
       +
                                   Log.debug (Printf.sprintf "First message content type: %s" content_debug);

     

       242
       242
       +
                                 end;

     

       243
       243
       +
                                 

     

       244
       244
       +
                                 let result = `Assoc [

     

       245
       245
       +
                                   ("messages", `List messages_json);

     

       246
       246
       +
                                   ("description", match prompt.description with 

     

       247
       247
       +
                                     | Some d -> begin `String d end 

     

       248
       248
       +
                                     | None -> begin `Null end)

     

       249
       249
       +
                                 ] in

     

       250
       250
       +
                                 Some (create_response ~id:req.id ~result)

     

       251
       251
       +
                               end

     

       252
       252
       +
                             | Error err -> begin

     

       253
       253
       +
                                 Log.error (Printf.sprintf "Error processing prompt: %s" err);

     

       254
       254
       +
                                 Some (create_error ~id:req.id ~code:ErrorCode.internal_error ~message:err ())

     

       255
       255
       +
                               end

     

       256
       256
       +
                           end

     

       257
       257
       +
                         | None -> begin

     

       258
       258
       +
                             Log.error (Printf.sprintf "Prompt not found: %s" name);

     

       259
       259
       +
                             Some (create_error ~id:req.id ~code:ErrorCode.invalid_params ~message:(Printf.sprintf "Prompt not found: %s" name) ())

     

       260
       260
       +
                           end

     

       261
       261
       +
                       end

     

       262
       262
       +
                     | _ -> begin

     

       263
       263
       +
                         Log.error "Invalid params format for prompts/get";

     

       264
       264
       +
                         Some (create_error ~id:req.id ~code:ErrorCode.invalid_params ~message:"Invalid params format" ())

     

       265
       265
       +
                       end

     

       266
       266
       +
                   end else begin

     

       267
       267
       +
                     Some (create_error ~id:req.id ~code:ErrorCode.method_not_found ~message:"Prompts not supported" ())

     

       268
       268
       +
                   end

     

       269
       269
       +
                 end

     

       270
       270
       +
                  

     

       271
       271
       +
               | "ping" -> begin

     

       272
       272
       +
                   Log.debug "Processing ping request";

     

       273
       273
       +
                   Some (create_response ~id:req.id ~result:(`Assoc []))

     

       274
       274
       +
                 end

     

       275
       275
       +
                

     

       276
       276
       +
               | _ -> begin

     

       277
       277
       +
                   Log.error (Printf.sprintf "Unknown method received: %s" req.method_);

     

       278
       278
       +
                   Some (create_error ~id:req.id ~code:ErrorCode.method_not_found ~message:("Method not found: " ^ req.method_) ())

     

       279
       279
       +
                 end

     

       280
       280
       +
             end

     

       281
       281
       +
                

     

       282
       282
       +
           | JSONRPCMessage.Notification notif -> begin

     

       283
       283
       +
               Log.debug (Printf.sprintf "Received notification with method: %s" notif.method_);

     

       284
       284
       +
               match notif.method_ with

     

       285
       285
       +
               | "notifications/initialized" -> begin

     

       286
       286
       +
                   Log.debug "Client initialization complete - Server is now ready to receive requests";

     

       287
       287
       +
                   None

     

       288
       288
       +
                 end

     

       289
       289
       +
               | _ -> begin

     

       290
       290
       +
                   Log.debug (Printf.sprintf "Ignoring notification: %s" notif.method_);

     

       291
       291
       +
                   None

     

       292
       292
       +
                 end

     

       293
       293
       +
             end

     

       294
       294
       +
                

     

       295
       295
       +
           | JSONRPCMessage.Response _ -> begin

     

       296
       296
       +
               Log.error "Unexpected response message received";

     

       297
       297
       +
               None

     

       298
       298
       +
             end

     

       299
       299
       +
             

     

       300
       300
       +
           | JSONRPCMessage.Error _ -> begin

     

       301
       301
       +
               Log.error "Unexpected error message received";

     

       302
       302
       +
               None

     

       303
       303
       +
             end

     

       304
       304
       +
         with

     

       305
       305
       +
         | Failure msg -> begin

     

       306
       306
       +
             Log.error (Printf.sprintf "JSON error in message processing: %s" msg);

     

       307
       307
       +
             None

     

       308
       308
       +
           end

     

       309
       309
       +
         | exc -> begin

     

       310
       310
       +
             Log.error (Printf.sprintf "Exception during message processing: %s" (Printexc.to_string exc));

     

       311
       311
       +
             Log.error (Printf.sprintf "Backtrace: %s" (Printexc.get_backtrace()));

     

       312
       312
       +
             None

     

       313
       313
       +
           end

     

       314
       314
       +
       

     

       315
       315
       +
       (** Read a single message from stdin *)

     

       316
       316
       +
       let read_stdio_message () =

     

       317
       317
       +
         try

     

       318
       318
       +
           Log.debug "Reading line from stdin...";

     

       319
       319
       +
           let line = read_line () in

     

       320
       320
       +
           if line = "" then begin

     

       321
       321
       +
             Log.debug "Empty line received, ignoring";

     

       322
       322
       +
             None

     

       323
       323
       +
           end else begin

     

       324
       324
       +
             Log.debug (Printf.sprintf "Raw input: %s" (String.sub line 0 (min 100 (String.length line))));

     

       325
       325
       +
             try

     

       326
       326
       +
               let json = Yojson.Safe.from_string line in

     

       327
       327
       +
               Log.debug "Successfully parsed JSON";

     

       328
       328
       +
               Some json

     

       329
       329
       +
             with

     

       330
       330
       +
             | Yojson.Json_error msg -> begin

     

       331
       331
       +
                 Log.error (Printf.sprintf "Error parsing JSON: %s" msg);

     

       332
       332
       +
                 Log.error (Printf.sprintf "Input was: %s" (String.sub line 0 (min 100 (String.length line))));

     

       333
       333
       +
                 None

     

       334
       334
       +
               end

     

       335
       335
       +
           end

     

       336
       336
       +
         with

     

       337
       337
       +
         | End_of_file -> begin

     

       338
       338
       +
             Log.debug "End of file received on stdin";

     

       339
       339
       +
             None

     

       340
       340
       +
           end

     

       341
       341
       +
         | Sys_error msg -> begin

     

       342
       342
       +
             Log.error (Printf.sprintf "System error while reading: %s" msg);

     

       343
       343
       +
             None

     

       344
       344
       +
           end

     

       345
       345
       +
         | exc -> begin

     

       346
       346
       +
             Log.error (Printf.sprintf "Exception while reading: %s" (Printexc.to_string exc));

     

       347
       347
       +
             None

     

       348
       348
       +
           end

     

       349
       349
       +
       

     

       350
       350
       +
       (** Run stdio server with enhanced error handling *)

     

       351
       351
       +
       let rec run_stdio_server mcp_server =

     

       352
       352
       +
         try begin

     

       353
       353
       +
           if not mcp_server.running then begin

     

       354
       354
       +
             Log.debug "Server stopped";

     

       355
       355
       +
             ()

     

       356
       356
       +
           end else begin

     

       357
       357
       +
             match read_stdio_message () with

     

       358
       358
       +
             | Some json -> begin

     

       359
       359
       +
                 Log.debug "Processing message...";

     

       360
       360
       +
                 try begin

     

       361
       361
       +
                   match process_message mcp_server.server json with

     

       362
       362
       +
                   | Some response -> begin

     

       363
       363
       +
                       let response_json = JSONRPCMessage.yojson_of_t response in

     

       364
       364
       +
                       let response_str = Yojson.Safe.to_string response_json in

     

       365
       365
       +
                       Log.debug (Printf.sprintf "Sending response: %s" 

     

       366
       366
       +
                                   (String.sub response_str 0 (min 100 (String.length response_str))));

     

       367
       367
       +
                       Printf.printf "%s\n" response_str;

     

       368
       368
       +
                       flush stdout;

     

       369
       369
       +
                       (* Give client time to process *)

     

       370
       370
       +
                       Unix.sleepf 0.01;

     

       371
       371
       +
                     end

     

       372
       372
       +
                   | None -> begin

     

       373
       373
       +
                       Log.debug "No response needed"

     

       374
       374
       +
                     end

     

       375
       375
       +
                   end with

     

       376
       376
       +
                   | exc -> begin

     

       377
       377
       +
                       Log.error (Printf.sprintf "ERROR in message processing: %s" (Printexc.to_string exc));

     

       378
       378
       +
                       Log.error (Printf.sprintf "Backtrace: %s" (Printexc.get_backtrace()));

     

       379
       379
       +
                       (* Try to extract ID and send an error response *)

     

       380
       380
       +
                       try begin

     

       381
       381
       +
                         let id_opt = match Yojson.Safe.Util.member "id" json with

     

       382
       382
       +
                           | `Int i -> Some (`Int i)

     

       383
       383
       +
                           | `String s -> Some (`String s)

     

       384
       384
       +
                           | _ -> None

     

       385
       385
       +
                         in

     

       386
       386
       +
                         match id_opt with

     

       387
       387
       +
                         | Some id -> begin

     

       388
       388
       +
                             let error_resp = create_error ~id ~code:ErrorCode.internal_error ~message:(Printexc.to_string exc) () in

     

       389
       389
       +
                             let error_json = JSONRPCMessage.yojson_of_t error_resp in

     

       390
       390
       +
                             let error_str = Yojson.Safe.to_string error_json in

     

       391
       391
       +
                             Printf.printf "%s\n" error_str;

     

       392
       392
       +
                             flush stdout;

     

       393
       393
       +
                           end

     

       394
       394
       +
                         | None -> begin

     

       395
       395
       +
                             Log.error "Could not extract request ID to send error response"

     

       396
       396
       +
                           end

     

       397
       397
       +
                         end with

     

       398
       398
       +
                         | e -> begin

     

       399
       399
       +
                             Log.error (Printf.sprintf "Failed to send error response: %s" (Printexc.to_string e))

     

       400
       400
       +
                           end

     

       401
       401
       +
                     end;

     

       402
       402
       +
                 run_stdio_server mcp_server

     

       403
       403
       +
               end

     

       404
       404
       +
             | None -> begin

     

       405
       405
       +
                 if mcp_server.running then begin

     

       406
       406
       +
                   (* No message received, but server is still running *)

     

       407
       407
       +
                   Unix.sleepf 0.1;  (* Small sleep to prevent CPU spinning *)

     

       408
       408
       +
                   run_stdio_server mcp_server

     

       409
       409
       +
                 end else begin

     

       410
       410
       +
                   Log.debug "Server stopped during message processing"

     

       411
       411
       +
                 end

     

       412
       412
       +
               end

     

       413
       413
       +
           end

     

       414
       414
       +
         end with

     

       415
       415
       +
         | exc -> begin

     

       416
       416
       +
             Log.error (Printf.sprintf "FATAL ERROR in server main loop: %s" (Printexc.to_string exc));

     

       417
       417
       +
             Log.error (Printf.sprintf "Backtrace: %s" (Printexc.get_backtrace()));

     

       418
       418
       +
             (* Try to continue anyway *)

     

       419
       419
       +
             if mcp_server.running then begin

     

       420
       420
       +
               Unix.sleepf 0.1;

     

       421
       421
       +
               run_stdio_server mcp_server

     

       422
       422
       +
             end

     

       423
       423
       +
           end

     

       424
       424
       +
       

     

       425
       425
       +
       (** Create an MCP server *)

     

       426
       426
       +
       let create ~server ~transport () =

     

       427
       427
       +
         { server; transport; running = false }

     

       428
       428
       +
       

     

       429
       429
       +
       (** HTTP server placeholder (to be fully implemented) *)

     

       430
       430
       +
       let run_http_server mcp_server port =

     

       431
       431
       +
         Log.info (Printf.sprintf "%s HTTP server starting on port %d" mcp_server.server.name port);

     

       432
       432
       +
         Log.info "HTTP transport is a placeholder and not fully implemented yet";

     

       433
       433
       +
         

     

       434
       434
       +
         (* This would be where we'd set up cohttp server *)

     

       435
       435
       +
         (*

     

       436
       436
       +
         let callback _conn req body =

     

       437
       437
       +
           let uri = req |> Cohttp.Request.uri in

     

       438
       438
       +
           let meth = req |> Cohttp.Request.meth |> Cohttp.Code.string_of_method in

     

       439
       439
       +
           

     

       440
       440
       +
           (* Handle only POST /jsonrpc endpoint *)

     

       441
       441
       +
           match (meth, Uri.path uri) with

     

       442
       442
       +
           | "POST", "/jsonrpc" ->

     

       443
       443
       +
             (* Read the body *)

     

       444
       444
       +
             Cohttp_lwt.Body.to_string body >>= fun body_str ->

     

       445
       445
       +
             

     

       446
       446
       +
             (* Parse JSON *)

     

       447
       447
       +
             let json = try Some (Yojson.Safe.from_string body_str) with _ -> None in

     

       448
       448
       +
             match json with

     

       449
       449
       +
             | Some json_msg ->

     

       450
       450
       +
               (* Process the message *)

     

       451
       451
       +
               let response_opt = process_message mcp_server.server json_msg in

     

       452
       452
       +
               (match response_opt with

     

       453
       453
       +
               | Some response ->

     

       454
       454
       +
                 let response_json = JSONRPCMessage.yojson_of_t response in

     

       455
       455
       +
                 let response_str = Yojson.Safe.to_string response_json in

     

       456
       456
       +
                 Cohttp_lwt_unix.Server.respond_string

     

       457
       457
       +
                   ~status:`OK

     

       458
       458
       +
                   ~body:response_str

     

       459
       459
       +
                   ~headers:(Cohttp.Header.init_with "Content-Type" "application/json")

     

       460
       460
       +
                   ()

     

       461
       461
       +
               | None ->

     

       462
       462
       +
                 Cohttp_lwt_unix.Server.respond_string

     

       463
       463
       +
                   ~status:`OK

     

       464
       464
       +
                   ~body:"{}"

     

       465
       465
       +
                   ~headers:(Cohttp.Header.init_with "Content-Type" "application/json")

     

       466
       466
       +
                   ())

     

       467
       467
       +
             | None ->

     

       468
       468
       +
               Cohttp_lwt_unix.Server.respond_string

     

       469
       469
       +
                 ~status:`Bad_request

     

       470
       470
       +
                 ~body:"{\"error\":\"Invalid JSON\"}"

     

       471
       471
       +
                 ~headers:(Cohttp.Header.init_with "Content-Type" "application/json")

     

       472
       472
       +
                 ()

     

       473
       473
       +
           | _ ->

     

       474
       474
       +
             (* Return 404 for any other routes *)

     

       475
       475
       +
             Cohttp_lwt_unix.Server.respond_string

     

       476
       476
       +
               ~status:`Not_found

     

       477
       477
       +
               ~body:"Not found"

     

       478
       478
       +
               ()

     

       479
       479
       +
         in

     

       480
       480
       +
         

     

       481
       481
       +
         (* Create and start the server *)

     

       482
       482
       +
         let server = Cohttp_lwt_unix.Server.create

     

       483
       483
       +
           ~mode:(`TCP (`Port port))

     

       484
       484
       +
           (Cohttp_lwt_unix.Server.make ~callback ())

     

       485
       485
       +
         in

     

       486
       486
       +
         

     

       487
       487
       +
         (* Run the server *)

     

       488
       488
       +
         Lwt_main.run server

     

       489
       489
       +
         *)

     

       490
       490
       +
         

     

       491
       491
       +
         (* For now, just wait until the server is stopped *)

     

       492
       492
       +
         while mcp_server.running do

     

       493
       493
       +
           Unix.sleep 1

     

       494
       494
       +
         done

     

       495
       495
       +
       

     

       496
       496
       +
       (** Start the server based on transport type *)

     

       497
       497
       +
       let start server =

     

       498
       498
       +
         server.running <- true;

     

       499
       499
       +
         

     

       500
       500
       +
         (* Run startup hook if provided *)

     

       501
       501
       +
         (match server.server.startup_hook with

     

       502
       502
       +
         | Some hook -> begin hook () end

     

       503
       503
       +
         | None -> begin () end);

     

       504
       504
       +
         

     

       505
       505
       +
         (* Install signal handler *)

     

       506
       506
       +
         Sys.(set_signal sigint (Signal_handle (fun _ -> 

     

       507
       507
       +
           Log.debug "Received interrupt signal, stopping server...";

     

       508
       508
       +
           server.running <- false

     

       509
       509
       +
         )));

     

       510
       510
       +
         

     

       511
       511
       +
         match server.transport with

     

       512
       512
       +
         | Stdio -> begin

     

       513
       513
       +
             (* Setup stdout and stderr *)

     

       514
       514
       +
             set_binary_mode_out stdout false;

     

       515
       515
       +
             Log.info (Printf.sprintf "%s server started with stdio transport" server.server.name);

     

       516
       516
       +
             

     

       517
       517
       +
             (* Run the server loop *)

     

       518
       518
       +
             run_stdio_server server

     

       519
       519
       +
           end

     

       520
       520
       +
         | Http -> begin

     

       521
       521
       +
             (* HTTP server placeholder *)

     

       522
       522
       +
             run_http_server server 8080

     

       523
       523
       +
           end

     

       524
       524
       +
       

     

       525
       525
       +
       (** Stop the server *)

     

       526
       526
       +
       let stop server =

     

       527
       527
       +
         Log.info "Stopping server...";

     

       528
       528
       +
         server.running <- false;

     

       529
       529
       +
         

     

       530
       530
       +
         (* Run shutdown hook if provided *)

     

       531
       531
       +
         match server.server.shutdown_hook with

     

       532
       532
       +
         | Some hook -> begin hook () end

     

       533
       533
       +
         | None -> begin () end

+54

lib/mcp_server.mli

···

       1
       1
       +
       (** MCP Server module - full implementation *)

     

       2
       2
       +
       

     

       3
       3
       +
       (** Transport type for server *)

     

       4
       4
       +
       type transport_type = 

     

       5
       5
       +
         | Stdio  (** Read/write to stdin/stdout *)

     

       6
       6
       +
         | Http   (** HTTP server - to be implemented *)

     

       7
       7
       +
       

     

       8
       8
       +
       (** Server type *)

     

       9
       9
       +
       type t = {

     

       10
       10
       +
         server: Mcp_sdk.server;

     

       11
       11
       +
         transport: transport_type;

     

       12
       12
       +
         mutable running: bool;

     

       13
       13
       +
       }

     

       14
       14
       +
       

     

       15
       15
       +
       (** Create an MCP server 

     

       16
       16
       +
           @param server The Mcp_sdk server to use

     

       17
       17
       +
           @param transport The transport type to use

     

       18
       18
       +
       *)

     

       19
       19
       +
       val create : server:Mcp_sdk.server -> transport:transport_type -> unit -> t

     

       20
       20
       +
       

     

       21
       21
       +
       (** Start the server

     

       22
       22
       +
           This function will block until the server is stopped.

     

       23
       23
       +
           @param server The server to start

     

       24
       24
       +
       *)

     

       25
       25
       +
       val start : t -> unit

     

       26
       26
       +
       

     

       27
       27
       +
       (** Stop the server

     

       28
       28
       +
           This will set the running flag to false and invoke the shutdown hook.

     

       29
       29
       +
           @param server The server to stop

     

       30
       30
       +
       *)

     

       31
       31
       +
       val stop : t -> unit

     

       32
       32
       +
       

     

       33
       33
       +
       (** Process a single message 

     

       34
       34
       +
           @param server The Mcp_sdk server to use

     

       35
       35
       +
           @param message The JSON message to process

     

       36
       36
       +
           @return An optional response message

     

       37
       37
       +
       *)

     

       38
       38
       +
       val process_message : Mcp_sdk.server -> Yojson.Safe.t -> Mcp.JSONRPCMessage.t option

     

       39
       39
       +
       

     

       40
       40
       +
       (** Run stdio server implementation 

     

       41
       41
       +
           @param mcp_server The mcp_server to run

     

       42
       42
       +
       *)

     

       43
       43
       +
       val run_stdio_server : t -> unit

     

       44
       44
       +
       

     

       45
       45
       +
       (** Read a message from stdio

     

       46
       46
       +
           @return An optional JSON message

     

       47
       47
       +
       *)

     

       48
       48
       +
       val read_stdio_message : unit -> Yojson.Safe.t option

     

       49
       49
       +
       

     

       50
       50
       +
       (** Run HTTP server implementation (placeholder)

     

       51
       51
       +
           @param mcp_server The mcp_server to run

     

       52
       52
       +
           @param port The port to listen on

     

       53
       53
       +
       *)

     

       54
       54
       +
       val run_http_server : t -> int -> unit