commit 01a8ae346ecff94c1f1901fbee41f73c65be37ef · anil.recoil.org/ocaml-mcp

+3 -3

lib/mcp_sdk.ml

···

       29
       29
        
         type t = {

     

       30
       30
        
           request_id: RequestId.t option;

     

       31
       31
        
           lifespan_context: (string * Json.t) list;

     

       32
       32
       -
           mutable progress_token: ProgressToken.t option;

     

       32
       32
       +
           progress_token: ProgressToken.t option;

     

       33
       33
        
         }

     

       34
       34
        
       

     

       35
       35
       -
         let create ?request_id ?(lifespan_context=[]) () =

     

       36
       36
       -
           { request_id; lifespan_context; progress_token = None }

     

       35
       35
       +
         let create ?request_id ?progress_token ?(lifespan_context=[]) () =

     

       36
       36
       +
           { request_id; lifespan_context; progress_token }

     

       37
       37
        
       

     

       38
       38
        
         let get_context_value ctx key =

     

       39
       39
        
           List.assoc_opt key ctx.lifespan_context

+2 -6

lib/mcp_sdk.mli

···

       21
       21
        
       

     

       22
       22
        
       (** Context for tools and resources *)

     

       23
       23
        
       module Context : sig

     

       24
       24
       -
         type t = {

     

       25
       25
       -
           request_id: RequestId.t option;

     

       26
       26
       -
           lifespan_context: (string * Json.t) list;

     

       27
       27
       -
           mutable progress_token: ProgressToken.t option;

     

       28
       28
       -
         }

     

       24
       24
       +
         type t

     

       29
       25
        
       

     

       30
       30
       -
         val create : ?request_id:RequestId.t -> ?lifespan_context:(string * Json.t) list -> unit -> t

     

       26
       26
       +
         val create : ?request_id:RequestId.t -> ?progress_token:ProgressToken.t -> ?lifespan_context:(string * Json.t) list -> unit -> t

     

       31
       27
        
         val get_context_value : t -> string -> Json.t option

     

       32
       28
        
         val report_progress : t -> float -> float -> JSONRPCMessage.t option

     

       33
       29
        
       end

+4 -12

lib/mcp_server.ml

···

       126
       126
        
               

     

       127
       127
        
               (* Create context for this request *)

     

       128
       128
        
               let ctx = Context.create 

     

       129
       129
       -
                 ?request_id:(Some req.id) (* Store request ID for progress reporting *)

     

       130
       130
       -
                 ~lifespan_context:[("tools/call", `Assoc params)] (* Store params for reference *)

     

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

     

       130
       130
       +
                 ?progress_token:req.progress_token

     

       131
       131
       +
                 ~lifespan_context:[("tools/call", `Assoc params)] 

     

       131
       132
        
                 () 

     

       132
       133
        
               in

     

       133
       133
       -
               

     

       134
       134
       -
               (* Set progress token if present *)

     

       135
       135
       -
               ctx.progress_token <- req.progress_token;

     

       136
       136
       -
               

     

       134
       134
       +
                       

     

       137
       135
        
               (* Execute the tool *)

     

       138
       136
        
               let result = execute_tool server ctx name args in

     

       139
       139
       -
               

     

       140
       140
       -
               (* Process progress messages if any *)

     

       141
       141
       -
               let progress_msg = Context.report_progress ctx 1.0 1.0 in

     

       142
       142
       -
               (match progress_msg with

     

       143
       143
       -
                | Some _msg -> Log.debug "Progress complete notification would be sent here";

     

       144
       144
       -
                | None -> ());

     

       145
       137
        
               

     

       146
       138
        
               Some (create_response ~id:req.id ~result)

     

       147
       139
        
           | None ->