|
|
| import { apiFetch } from "../api.mjs";
|
| import { emit } from "../output.mjs";
|
| import { readFileSync } from "node:fs";
|
|
|
| export function register_rerank(parent) {
|
| const tag = parent.command("rerank").description("Rerank endpoints");
|
| tag
|
| .command("post-api-v1-rerank")
|
| .description("Rerank documents")
|
| .option("--body <jsonOrPath>", "JSON body or @path/to/file.json")
|
| .action(async (opts, cmd) => {
|
| const gOpts = cmd.optsWithGlobals();
|
| let url = "/api/v1/rerank";
|
| let body;
|
| if (opts.body) {
|
| body = opts.body.startsWith("@")
|
| ? JSON.parse(readFileSync(opts.body.slice(1), "utf8"))
|
| : JSON.parse(opts.body);
|
| }
|
| const res = await apiFetch(url, {
|
| method: "POST",
|
| body,
|
| baseUrl: gOpts.baseUrl,
|
| apiKey: gOpts.apiKey,
|
| });
|
| const data = res.ok ? await res.json() : await res.text();
|
| emit(data, gOpts);
|
| });
|
| }
|
|
|
