-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
98 additions
and
67 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,97 +1,120 @@ | ||
| let api_key = "" | ||
| let base_url = "https://api.openai.com/v1" | ||
| let default_headers = Cohttp.Header.init_with "Content-Type" "application/json" | ||
| module type Auth = sig | ||
| val api_key : string | ||
| val org_id : string option | ||
| end | ||
|
|
||
| let option_fold f default o = | ||
| match o with | ||
| | Some v -> f v | ||
| | None -> default | ||
|
|
||
| let build_uri operation_path = Uri.of_string (base_url ^ operation_path) | ||
| module Make (Config: Auth) = struct | ||
|
|
||
| let add_string_header headers key value = | ||
| Cohttp.Header.add headers key value | ||
| let api_key = "" | ||
| let base_url = "https://api.openai.com/v1" | ||
|
|
||
| let add_string_header_multi headers key values = | ||
| Cohttp.Header.add_multi headers key values | ||
| (* Authentication was not added by the OpenAPI generator, | ||
| but a bearer token is required and an organization id is | ||
| optionally supported. *) | ||
| let default_headers = | ||
| let org_id_header = match Config.org_id with | ||
| | None -> [] | ||
| | Some id -> ["OpenAI-Organization", id] | ||
| in | ||
| Cohttp.Header.of_list ( | ||
| [ "Content-Type", "application/json" | ||
| ; "Authorization", "Bearer "^Config.api_key | ||
| ] @ org_id_header | ||
| ) | ||
|
|
||
| let add_header headers key to_string value = | ||
| Cohttp.Header.add headers key (to_string value) | ||
| let option_fold f default o = | ||
| match o with | ||
| | Some v -> f v | ||
| | None -> default | ||
|
|
||
| let add_header_multi headers key to_string value = | ||
| Cohttp.Header.add_multi headers key (to_string value) | ||
| let build_uri operation_path = Uri.of_string (base_url ^ operation_path) | ||
|
|
||
| let maybe_add_header headers key to_string value = | ||
| option_fold (add_header headers key to_string) headers value | ||
| let add_string_header headers key value = | ||
| Cohttp.Header.add headers key value | ||
|
|
||
| let maybe_add_header_multi headers key to_string value = | ||
| option_fold (add_header_multi headers key to_string) headers value | ||
| let add_string_header_multi headers key values = | ||
| Cohttp.Header.add_multi headers key values | ||
|
|
||
| let write_string_body s = Cohttp_lwt.Body.of_string s | ||
| let add_header headers key to_string value = | ||
| Cohttp.Header.add headers key (to_string value) | ||
|
|
||
| let write_json_body payload = | ||
| Cohttp_lwt.Body.of_string (Yojson.Safe.to_string payload ~std:true) | ||
| let add_header_multi headers key to_string value = | ||
| Cohttp.Header.add_multi headers key (to_string value) | ||
|
|
||
| let write_as_json_body to_json payload = write_json_body (to_json payload) | ||
| let maybe_add_header headers key to_string value = | ||
| option_fold (add_header headers key to_string) headers value | ||
|
|
||
| let handle_response resp on_success_handler = | ||
| match Cohttp_lwt.Response.status resp with | ||
| | #Cohttp.Code.success_status -> on_success_handler () | ||
| | s -> failwith ("Server responded with status " ^ Cohttp.Code.(reason_phrase_of_code (code_of_status s))) | ||
| let maybe_add_header_multi headers key to_string value = | ||
| option_fold (add_header_multi headers key to_string) headers value | ||
|
|
||
| let handle_unit_response resp = handle_response resp (fun () -> Lwt.return ()) | ||
| let write_string_body s = Cohttp_lwt.Body.of_string s | ||
|
|
||
| let read_json_body resp body = | ||
| handle_response resp (fun () -> | ||
| (Lwt.(Cohttp_lwt.Body.to_string body >|= Yojson.Safe.from_string))) | ||
| let write_json_body payload = | ||
| Cohttp_lwt.Body.of_string (Yojson.Safe.to_string payload ~std:true) | ||
|
|
||
| let read_json_body_as of_json resp body = | ||
| Lwt.(read_json_body resp body >|= of_json) | ||
| let write_as_json_body to_json payload = write_json_body (to_json payload) | ||
|
|
||
| let read_json_body_as_list resp body = | ||
| Lwt.(read_json_body resp body >|= Yojson.Safe.Util.to_list) | ||
| let handle_response resp on_success_handler = | ||
| match Cohttp_lwt.Response.status resp with | ||
| | #Cohttp.Code.success_status -> on_success_handler () | ||
| | s -> failwith ("Server responded with status " ^ Cohttp.Code.(reason_phrase_of_code (code_of_status s))) | ||
|
|
||
| let read_json_body_as_list_of of_json resp body = | ||
| Lwt.(read_json_body_as_list resp body >|= List.map of_json) | ||
| let handle_unit_response resp = handle_response resp (fun () -> Lwt.return ()) | ||
|
|
||
| let read_json_body_as_map resp body = | ||
| Lwt.(read_json_body resp body >|= Yojson.Safe.Util.to_assoc) | ||
| let read_json_body resp body = | ||
| handle_response resp (fun () -> | ||
| (Lwt.(Cohttp_lwt.Body.to_string body >|= Yojson.Safe.from_string))) | ||
|
|
||
| let read_json_body_as_map_of of_json resp body = | ||
| Lwt.(read_json_body_as_map resp body >|= List.map (fun (s, v) -> (s, of_json v))) | ||
| let read_json_body_as of_json resp body = | ||
| Lwt.(read_json_body resp body >|= of_json) | ||
|
|
||
| let replace_string_path_param uri param_name param_value = | ||
| let regexp = Str.regexp (Str.quote ("{" ^ param_name ^ "}")) in | ||
| let path = Str.global_replace regexp param_value (Uri.pct_decode (Uri.path uri)) in | ||
| Uri.with_path uri path | ||
| let read_json_body_as_list resp body = | ||
| Lwt.(read_json_body resp body >|= Yojson.Safe.Util.to_list) | ||
|
|
||
| let replace_path_param uri param_name to_string param_value = | ||
| replace_string_path_param uri param_name (to_string param_value) | ||
| let read_json_body_as_list_of of_json resp body = | ||
| Lwt.(read_json_body_as_list resp body >|= List.map of_json) | ||
|
|
||
| let maybe_replace_path_param uri param_name to_string param_value = | ||
| option_fold (replace_path_param uri param_name to_string) uri param_value | ||
| let read_json_body_as_map resp body = | ||
| Lwt.(read_json_body resp body >|= Yojson.Safe.Util.to_assoc) | ||
|
|
||
| let add_query_param uri param_name to_string param_value = | ||
| Uri.add_query_param' uri (param_name, to_string param_value) | ||
| let read_json_body_as_map_of of_json resp body = | ||
| Lwt.(read_json_body_as_map resp body >|= List.map (fun (s, v) -> (s, of_json v))) | ||
|
|
||
| let add_query_param_list uri param_name to_string param_value = | ||
| Uri.add_query_param uri (param_name, to_string param_value) | ||
| let replace_string_path_param uri param_name param_value = | ||
| let regexp = Str.regexp (Str.quote ("{" ^ param_name ^ "}")) in | ||
| let path = Str.global_replace regexp param_value (Uri.pct_decode (Uri.path uri)) in | ||
| Uri.with_path uri path | ||
|
|
||
| let maybe_add_query_param uri param_name to_string param_value = | ||
| option_fold (add_query_param uri param_name to_string) uri param_value | ||
| let replace_path_param uri param_name to_string param_value = | ||
| replace_string_path_param uri param_name (to_string param_value) | ||
|
|
||
| let init_form_encoded_body () = "" | ||
| let maybe_replace_path_param uri param_name to_string param_value = | ||
| option_fold (replace_path_param uri param_name to_string) uri param_value | ||
|
|
||
| let add_form_encoded_body_param params param_name to_string param_value = | ||
| let new_param_enc = Printf.sprintf {|%s=%s|} (Uri.pct_encode param_name) (Uri.pct_encode (to_string param_value)) in | ||
| if params = "" | ||
| then new_param_enc | ||
| else Printf.sprintf {|%s&%s|} params new_param_enc | ||
| let add_query_param uri param_name to_string param_value = | ||
| Uri.add_query_param' uri (param_name, to_string param_value) | ||
|
|
||
| let add_form_encoded_body_param_list params param_name to_string new_params = | ||
| add_form_encoded_body_param params param_name (String.concat ",") (to_string new_params) | ||
| let add_query_param_list uri param_name to_string param_value = | ||
| Uri.add_query_param uri (param_name, to_string param_value) | ||
|
|
||
| let maybe_add_form_encoded_body_param params param_name to_string param_value = | ||
| option_fold (add_form_encoded_body_param params param_name to_string) params param_value | ||
| let maybe_add_query_param uri param_name to_string param_value = | ||
| option_fold (add_query_param uri param_name to_string) uri param_value | ||
|
|
||
| let finalize_form_encoded_body body = Cohttp_lwt.Body.of_string body | ||
| let init_form_encoded_body () = "" | ||
|
|
||
| let add_form_encoded_body_param params param_name to_string param_value = | ||
| let new_param_enc = Printf.sprintf {|%s=%s|} (Uri.pct_encode param_name) (Uri.pct_encode (to_string param_value)) in | ||
| if params = "" | ||
| then new_param_enc | ||
| else Printf.sprintf {|%s&%s|} params new_param_enc | ||
|
|
||
| let add_form_encoded_body_param_list params param_name to_string new_params = | ||
| add_form_encoded_body_param params param_name (String.concat ",") (to_string new_params) | ||
|
|
||
| let maybe_add_form_encoded_body_param params param_name to_string param_value = | ||
| option_fold (add_form_encoded_body_param params param_name to_string) params param_value | ||
|
|
||
| let finalize_form_encoded_body body = Cohttp_lwt.Body.of_string body | ||
|
|
||
| end |