@tiangong-ai/cli 0.0.13 → 0.0.15

package/dist/http.d.ts

@@ -0,0 +1,28 @@
+export type ResponseLike = {
+    ok: boolean;
+    status: number;
+    headers: {
+        get(name: string): string | null;
+    };
+    text(): Promise<string>;
+};
+export type FetchLike = (input: string, init?: RequestInit) => Promise<ResponseLike>;
+export interface HttpJsonConfig {
+    apiBaseUrl: string;
+    apiPathPrefix: string;
+    apiKey: string;
+    timeoutSeconds: number;
+}
+export declare function jsonRequest(config: HttpJsonConfig, path: string, init?: RequestInit & {
+    headers?: Record<string, string>;
+}): Promise<unknown>;
+export declare function jsonRequestUrl(url: string, init?: RequestInit & {
+    headers?: HeadersInit;
+}, timeoutSeconds?: number): Promise<unknown>;
+export declare function postJson(options: {
+    url: string;
+    headers: Record<string, string>;
+    body: unknown;
+    timeoutMs: number;
+    fetchImpl: FetchLike;
+}): Promise<unknown>;

package/dist/http.js

@@ -0,0 +1,80 @@
+import { CliError, HttpError } from "./errors.js";
+const RETRYABLE_HTTP_STATUSES = new Set([408, 425, 429, 500, 502, 503, 504]);
+export async function jsonRequest(config, path, init = {}) {
+    const url = `${config.apiBaseUrl}${config.apiPathPrefix}/${path.replace(/^\/+/, "")}`;
+    const headers = new Headers(init.headers);
+    headers.set("Authorization", `Bearer ${config.apiKey}`);
+    return jsonRequestUrl(url, { ...init, headers }, config.timeoutSeconds);
+}
+export async function jsonRequestUrl(url, init = {}, timeoutSeconds = 120) {
+    const headers = new Headers(init.headers);
+    headers.set("Accept", "application/json");
+    if (init.body && !(init.body instanceof FormData)) {
+        headers.set("Content-Type", "application/json");
+    }
+    let response;
+    try {
+        response = await fetch(url, {
+            ...init,
+            headers,
+            signal: AbortSignal.timeout(timeoutSeconds * 1000),
+        });
+    }
+    catch (error) {
+        throw new HttpError(`Request failed for ${url}: ${error instanceof Error ? error.message : String(error)}`, undefined, undefined, true);
+    }
+    const text = await response.text();
+    if (!response.ok) {
+        throw new HttpError(`HTTP ${response.status} from ${url}: ${text}`, response.status, retryAfterSeconds(response.headers), RETRYABLE_HTTP_STATUSES.has(response.status));
+    }
+    if (!text.trim())
+        return {};
+    try {
+        return JSON.parse(text);
+    }
+    catch {
+        throw new CliError(`Expected JSON from ${url}, got: ${text.slice(0, 200)}`);
+    }
+}
+async function parseResponse(response, url) {
+    const rawText = await response.text();
+    const contentType = response.headers.get("content-type") ?? "";
+    if (!response.ok) {
+        throw new CliError(`HTTP ${response.status} returned from ${url}`, {
+            code: "REMOTE_REQUEST_FAILED",
+            exitCode: 1,
+            details: rawText,
+        });
+    }
+    if (contentType.includes("application/json")) {
+        try {
+            return JSON.parse(rawText);
+        }
+        catch (error) {
+            throw new CliError(`Remote response was not valid JSON for ${url}`, {
+                code: "REMOTE_INVALID_JSON",
+                exitCode: 1,
+                details: String(error),
+            });
+        }
+    }
+    return rawText;
+}
+export async function postJson(options) {
+    const signal = AbortSignal.timeout(options.timeoutMs);
+    const response = await options.fetchImpl(options.url, {
+        method: "POST",
+        headers: options.headers,
+        body: JSON.stringify(options.body),
+        signal,
+    });
+    return parseResponse(response, options.url);
+}
+function retryAfterSeconds(headers) {
+    const value = headers.get("Retry-After");
+    if (!value)
+        return undefined;
+    const parsed = Number(value);
+    return Number.isFinite(parsed) && parsed > 0 ? parsed : undefined;
+}
+//# sourceMappingURL=http.js.map

package/dist/http.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../src/http.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAElD,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;AAoB7E,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,MAAsB,EACtB,IAAY,EACZ,OAA2D,EAAE;IAE7D,MAAM,GAAG,GAAG,GAAG,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;IACtF,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACxD,OAAO,cAAc,CAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,GAAW,EACX,OAAgD,EAAE,EAClD,cAAc,GAAG,GAAG;IAEpB,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;IAC1C,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,YAAY,QAAQ,CAAC,EAAE,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,IAAI,QAAkB,CAAC;IACvB,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAC1B,GAAG,IAAI;YACP,OAAO;YACP,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;SACnD,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,SAAS,CACjB,sBAAsB,GAAG,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACtF,SAAS,EACT,SAAS,EACT,IAAI,CACL,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,SAAS,CACjB,QAAQ,QAAQ,CAAC,MAAM,SAAS,GAAG,KAAK,IAAI,EAAE,EAC9C,QAAQ,CAAC,MAAM,EACf,iBAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,EACnC,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAC7C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC;IAC5B,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,QAAQ,CAAC,sBAAsB,GAAG,UAAU,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IAC9E,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,QAAsB,EAAE,GAAW;IAC9D,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACtC,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAE/D,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,QAAQ,CAAC,QAAQ,QAAQ,CAAC,MAAM,kBAAkB,GAAG,EAAE,EAAE;YACjE,IAAI,EAAE,uBAAuB;YAC7B,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;IACL,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,QAAQ,CAAC,0CAA0C,GAAG,EAAE,EAAE;gBAClE,IAAI,EAAE,qBAAqB;gBAC3B,QAAQ,EAAE,CAAC;gBACX,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC;aACvB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,OAM9B;IACC,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE;QACpD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC;QAClC,MAAM;KACP,CAAC,CAAC;IAEH,OAAO,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAgB;IACzC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACzC,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACpE,CAAC"}

package/dist/io.d.ts

@@ -0,0 +1,11 @@
+export interface Output {
+    write(text: string): unknown;
+}
+export interface CliIO {
+    env: NodeJS.ProcessEnv;
+    stdout: Output;
+    stderr: Output;
+}
+export declare function write(output: Output, text: string): void;
+export declare function readJsonInput(inputPath: string): unknown;
+export declare function stringifyJson(value: unknown, compact: boolean): string;

package/dist/io.js

@@ -0,0 +1,33 @@
+import { existsSync, readFileSync } from "node:fs";
+import { CliError } from "./errors.js";
+export function write(output, text) {
+    output.write(text);
+}
+export function readJsonInput(inputPath) {
+    if (!inputPath) {
+        throw new CliError("Missing required --input value.", {
+            code: "INPUT_REQUIRED",
+            exitCode: 2,
+        });
+    }
+    if (!existsSync(inputPath)) {
+        throw new CliError(`Input file not found: ${inputPath}`, {
+            code: "INPUT_NOT_FOUND",
+            exitCode: 2,
+        });
+    }
+    try {
+        return JSON.parse(readFileSync(inputPath, "utf8"));
+    }
+    catch (error) {
+        throw new CliError(`Input file is not valid JSON: ${inputPath}`, {
+            code: "INPUT_INVALID_JSON",
+            exitCode: 2,
+            details: String(error),
+        });
+    }
+}
+export function stringifyJson(value, compact) {
+    return `${JSON.stringify(value, null, compact ? undefined : 2)}\n`;
+}
+//# sourceMappingURL=io.js.map

package/dist/io.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"io.js","sourceRoot":"","sources":["../src/io.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEnD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAYvC,MAAM,UAAU,KAAK,CAAC,MAAc,EAAE,IAAY;IAChD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,SAAiB;IAC7C,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,QAAQ,CAAC,iCAAiC,EAAE;YACpD,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,QAAQ,CAAC,yBAAyB,SAAS,EAAE,EAAE;YACvD,IAAI,EAAE,iBAAiB;YACvB,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAY,CAAC;IAChE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,QAAQ,CAAC,iCAAiC,SAAS,EAAE,EAAE;YAC/D,IAAI,EAAE,oBAAoB;YAC1B,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC;SACvB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAc,EAAE,OAAgB;IAC5D,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AACrE,CAAC"}

package/dist/kb/client.d.ts

@@ -0,0 +1,19 @@
+import type { KbConfig } from "./config.js";
+export interface CollectionSelector {
+    field: "primary_collection_id" | "collection_path" | "collection_key" | "collection_name";
+    value: string;
+}
+export interface CollectionItem {
+    id?: string;
+    key?: string;
+    path?: string;
+    name?: string;
+    [key: string]: unknown;
+}
+export declare function resolveCollection(config: KbConfig, selector: CollectionSelector, options: {
+    includeSchema: boolean;
+}): Promise<unknown>;
+export declare function resolveSelectorFields(config: KbConfig, selector: CollectionSelector): Promise<Record<string, string>>;
+export declare function listCollections(config: KbConfig, capability: string, limit: number): Promise<CollectionItem[]>;
+export declare function collectionKey(item: CollectionItem): string | undefined;
+export declare function collectionPath(item: CollectionItem): string | undefined;

package/dist/kb/client.js

@@ -0,0 +1,71 @@
+import { isObject, responseData, stringField } from "../data.js";
+import { CliError } from "../errors.js";
+import { jsonRequest } from "../http.js";
+export async function resolveCollection(config, selector, options) {
+    const params = new URLSearchParams({ action: "upload" });
+    params.set(selector.field, selector.value);
+    if (options.includeSchema)
+        params.set("include_schema", "true");
+    return jsonRequest(config, `collections/resolve?${params.toString()}`);
+}
+export async function resolveSelectorFields(config, selector) {
+    if (selector.field !== "collection_name") {
+        return { [selector.field]: selector.value };
+    }
+    const matches = (await listCollections(config, "upload", 100)).filter((item) => item.name === selector.value);
+    if (matches.length === 0)
+        throw new CliError(`No uploadable collection matched name: ${selector.value}`);
+    if (matches.length > 1) {
+        const choices = matches.map((item) => item.key ?? item.path ?? item.id ?? "").join(", ");
+        throw new CliError(`Collection name is not unique: ${selector.value}. Use --collection-key or --collection-path. Matches: ${choices}`);
+    }
+    const match = matches[0];
+    const key = collectionKey(match);
+    if (key)
+        return { collection_key: key };
+    const path = collectionPath(match);
+    if (path)
+        return { collection_path: path };
+    if (typeof match.id === "string" && match.id)
+        return { primary_collection_id: match.id };
+    throw new CliError(`Collection matched name ${selector.value}, but response had no key, path, or id.`);
+}
+export async function listCollections(config, capability, limit) {
+    const collections = [];
+    let offset = 0;
+    while (true) {
+        const payload = await jsonRequest(config, `collections?${new URLSearchParams({
+            capability,
+            limit: String(limit),
+            offset: String(offset),
+        }).toString()}`);
+        const page = collectionItems(payload);
+        collections.push(...page);
+        if (page.length < limit)
+            return collections;
+        offset += limit;
+    }
+}
+export function collectionKey(item) {
+    return (stringField(item, "key") ??
+        stringField(item, "collectionKey") ??
+        stringField(item, "collection_key"));
+}
+export function collectionPath(item) {
+    return (stringField(item, "path") ??
+        stringField(item, "collectionPath") ??
+        stringField(item, "collection_path"));
+}
+function collectionItems(payload) {
+    const data = responseData(payload);
+    if (Array.isArray(data))
+        return data.filter(isObject);
+    if (isObject(data) && Array.isArray(data.collections))
+        return data.collections.filter(isObject);
+    if (isObject(data) && Array.isArray(data.data))
+        return data.data.filter(isObject);
+    if (isObject(payload) && Array.isArray(payload.data))
+        return payload.data.filter(isObject);
+    throw new CliError("Collection list response did not contain a data array.");
+}
+//# sourceMappingURL=client.js.map

package/dist/kb/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/kb/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAgBzC,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,MAAgB,EAChB,QAA4B,EAC5B,OAAmC;IAEnC,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IACzD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC3C,IAAI,OAAO,CAAC,aAAa;QAAE,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;IAChE,OAAO,WAAW,CAAC,MAAM,EAAE,uBAAuB,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAAgB,EAChB,QAA4B;IAE5B,IAAI,QAAQ,CAAC,KAAK,KAAK,iBAAiB,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC;IAC9C,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,MAAM,eAAe,CAAC,MAAM,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,MAAM,CACnE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,KAAK,CACvC,CAAC;IACF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QACtB,MAAM,IAAI,QAAQ,CAAC,0CAA0C,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;IACjF,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzF,MAAM,IAAI,QAAQ,CAChB,kCAAkC,QAAQ,CAAC,KAAK,yDAAyD,OAAO,EAAE,CACnH,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAmB,CAAC;IAC3C,MAAM,GAAG,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,GAAG;QAAE,OAAO,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC;IACxC,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,IAAI;QAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;IAC3C,IAAI,OAAO,KAAK,CAAC,EAAE,KAAK,QAAQ,IAAI,KAAK,CAAC,EAAE;QAAE,OAAO,EAAE,qBAAqB,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;IACzF,MAAM,IAAI,QAAQ,CAChB,2BAA2B,QAAQ,CAAC,KAAK,yCAAyC,CACnF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAAgB,EAChB,UAAkB,EAClB,KAAa;IAEb,MAAM,WAAW,GAAqB,EAAE,CAAC;IACzC,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,OAAO,GAAG,MAAM,WAAW,CAC/B,MAAM,EACN,eAAe,IAAI,eAAe,CAAC;YACjC,UAAU;YACV,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;YACpB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;SACvB,CAAC,CAAC,QAAQ,EAAE,EAAE,CAChB,CAAC;QACF,MAAM,IAAI,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QACtC,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;QAC1B,IAAI,IAAI,CAAC,MAAM,GAAG,KAAK;YAAE,OAAO,WAAW,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC;IAClB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAoB;IAChD,OAAO,CACL,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC;QACxB,WAAW,CAAC,IAAI,EAAE,eAAe,CAAC;QAClC,WAAW,CAAC,IAAI,EAAE,gBAAgB,CAAC,CACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAoB;IACjD,OAAO,CACL,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC;QACzB,WAAW,CAAC,IAAI,EAAE,gBAAgB,CAAC;QACnC,WAAW,CAAC,IAAI,EAAE,iBAAiB,CAAC,CACrC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAgB;IACvC,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACnC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAqB,CAAC;IAC1E,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC;QACnD,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAqB,CAAC;IAC/D,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;QAC5C,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAqB,CAAC;IACxD,IAAI,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;QAClD,OAAO,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAqB,CAAC;IAC3D,MAAM,IAAI,QAAQ,CAAC,wDAAwD,CAAC,CAAC;AAC/E,CAAC"}

package/dist/kb/config.d.ts

@@ -0,0 +1,12 @@
+import { type ParsedArgs } from "../args.js";
+export declare const DEFAULT_API_BASE_URL = "https://thuenv.tiangong.world:7300";
+export declare const DEFAULT_API_PATH_PREFIX = "/api/v1/kb";
+export interface KbConfig {
+    apiBaseUrl: string;
+    apiPathPrefix: string;
+    apiKey: string;
+    timeoutSeconds: number;
+}
+export declare function resolveKbConfig(args: ParsedArgs, env: NodeJS.ProcessEnv, options?: {
+    requireKey?: boolean;
+}): KbConfig;

package/dist/kb/config.js

@@ -0,0 +1,26 @@
+import { getString, positiveNumberValue } from "../args.js";
+import { firstEnv } from "../env.js";
+import { CliError } from "../errors.js";
+export const DEFAULT_API_BASE_URL = "https://thuenv.tiangong.world:7300";
+export const DEFAULT_API_PATH_PREFIX = "/api/v1/kb";
+export function resolveKbConfig(args, env, options = {}) {
+    const requireKey = options.requireKey ?? true;
+    const apiKey = getString(args, "api-key") ?? firstEnv(env, "TIANGONG_AI_API_KEY", "TIANGONG_KB_API_KEY") ?? "";
+    if (requireKey && !apiKey) {
+        throw new CliError("Missing API key. Provide --api-key or set TIANGONG_AI_API_KEY.");
+    }
+    return {
+        apiBaseUrl: (getString(args, "api-base-url") ??
+            firstEnv(env, "TIANGONG_KB_API_BASE_URL") ??
+            DEFAULT_API_BASE_URL).replace(/\/+$/, ""),
+        apiPathPrefix: normalizePrefix(getString(args, "api-path-prefix") ??
+            firstEnv(env, "TIANGONG_KB_API_PATH_PREFIX") ??
+            DEFAULT_API_PATH_PREFIX),
+        apiKey,
+        timeoutSeconds: positiveNumberValue(getString(args, "timeout") ?? firstEnv(env, "TIANGONG_KB_TIMEOUT"), 120, "--timeout"),
+    };
+}
+function normalizePrefix(value) {
+    return `/${value.replace(/^\/+|\/+$/g, "")}`;
+}
+//# sourceMappingURL=config.js.map

package/dist/kb/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/kb/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,mBAAmB,EAAmB,MAAM,YAAY,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAExC,MAAM,CAAC,MAAM,oBAAoB,GAAG,oCAAoC,CAAC;AACzE,MAAM,CAAC,MAAM,uBAAuB,GAAG,YAAY,CAAC;AASpD,MAAM,UAAU,eAAe,CAC7B,IAAgB,EAChB,GAAsB,EACtB,UAAoC,EAAE;IAEtC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,IAAI,CAAC;IAC9C,MAAM,MAAM,GACV,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,QAAQ,CAAC,GAAG,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,IAAI,EAAE,CAAC;IAClG,IAAI,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC;QAC1B,MAAM,IAAI,QAAQ,CAAC,gEAAgE,CAAC,CAAC;IACvF,CAAC;IAED,OAAO;QACL,UAAU,EAAE,CACV,SAAS,CAAC,IAAI,EAAE,cAAc,CAAC;YAC/B,QAAQ,CAAC,GAAG,EAAE,0BAA0B,CAAC;YACzC,oBAAoB,CACrB,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;QACrB,aAAa,EAAE,eAAe,CAC5B,SAAS,CAAC,IAAI,EAAE,iBAAiB,CAAC;YAChC,QAAQ,CAAC,GAAG,EAAE,6BAA6B,CAAC;YAC5C,uBAAuB,CAC1B;QACD,MAAM;QACN,cAAc,EAAE,mBAAmB,CACjC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,QAAQ,CAAC,GAAG,EAAE,qBAAqB,CAAC,EAClE,GAAG,EACH,WAAW,CACZ;KACF,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAa;IACpC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC;AAC/C,CAAC"}

package/dist/kb/pipeline-health.d.ts

@@ -0,0 +1,10 @@
+import type { KbConfig } from "./config.js";
+export interface BulkPipelineHealthSnapshot {
+    healthy: boolean;
+    pressure: "ok" | "degraded" | "paused" | "unknown";
+    recommendedAction: "continue" | "slow_down" | "pause_top_up";
+    recommendedPollAfterSeconds: number;
+    checkedAt?: string | undefined;
+    message?: string | undefined;
+}
+export declare function readBulkPipelineHealth(config: KbConfig, fallbackPollAfterSeconds: number, selectorFields?: Record<string, string>): Promise<BulkPipelineHealthSnapshot>;

package/dist/kb/pipeline-health.js

@@ -0,0 +1,68 @@
+import { isObject, responseData, stringField } from "../data.js";
+import { CliError, HttpError } from "../errors.js";
+import { jsonRequest } from "../http.js";
+export async function readBulkPipelineHealth(config, fallbackPollAfterSeconds, selectorFields = {}) {
+    try {
+        const payload = await jsonRequest(config, pipelineHealthPath(selectorFields));
+        return pipelineHealthFromPayload(payload, fallbackPollAfterSeconds);
+    }
+    catch (error) {
+        if (error instanceof HttpError && error.status && [404, 405, 501].includes(error.status)) {
+            return {
+                healthy: true,
+                pressure: "unknown",
+                recommendedAction: "continue",
+                recommendedPollAfterSeconds: fallbackPollAfterSeconds,
+                message: "Pipeline health endpoint is unavailable; continuing without backpressure.",
+            };
+        }
+        if (error instanceof HttpError) {
+            return {
+                healthy: false,
+                pressure: "paused",
+                recommendedAction: "pause_top_up",
+                recommendedPollAfterSeconds: error.retryAfterSeconds ?? Math.max(fallbackPollAfterSeconds, 30),
+                message: error.message,
+            };
+        }
+        return {
+            healthy: false,
+            pressure: "paused",
+            recommendedAction: "pause_top_up",
+            recommendedPollAfterSeconds: Math.max(fallbackPollAfterSeconds, 30),
+            message: error instanceof Error ? error.message : String(error),
+        };
+    }
+}
+function pipelineHealthPath(selectorFields) {
+    const params = new URLSearchParams();
+    for (const key of ["primary_collection_id", "collection_path", "collection_key"]) {
+        const value = selectorFields[key];
+        if (value)
+            params.set(key, value);
+    }
+    const query = params.toString();
+    return query ? `pipeline/health?${query}` : "pipeline/health";
+}
+function pipelineHealthFromPayload(payload, fallbackPollAfterSeconds) {
+    const data = responseData(payload);
+    if (!isObject(data)) {
+        throw new CliError("Pipeline health response did not contain an object payload.");
+    }
+    const action = stringField(data, "recommendedAction");
+    if (action !== "continue" && action !== "slow_down" && action !== "pause_top_up") {
+        throw new CliError("Pipeline health response did not contain a valid recommendedAction.");
+    }
+    const pressure = stringField(data, "pressure");
+    const pollAfter = Number(data.recommendedPollAfterSeconds);
+    return {
+        healthy: typeof data.healthy === "boolean" ? data.healthy : action === "continue",
+        pressure: pressure === "ok" || pressure === "degraded" || pressure === "paused" ? pressure : "unknown",
+        recommendedAction: action,
+        recommendedPollAfterSeconds: Number.isFinite(pollAfter) && pollAfter > 0 ? pollAfter : fallbackPollAfterSeconds,
+        checkedAt: stringField(data, "checkedAt"),
+        message: stringField(data, "message") ??
+            (isObject(data.indexPreflight) ? stringField(data.indexPreflight, "message") : undefined),
+    };
+}
+//# sourceMappingURL=pipeline-health.js.map

package/dist/kb/pipeline-health.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline-health.js","sourceRoot":"","sources":["../../src/kb/pipeline-health.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAYzC,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,MAAgB,EAChB,wBAAgC,EAChC,iBAAyC,EAAE;IAE3C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,kBAAkB,CAAC,cAAc,CAAC,CAAC,CAAC;QAC9E,OAAO,yBAAyB,CAAC,OAAO,EAAE,wBAAwB,CAAC,CAAC;IACtE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,SAAS,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YACzF,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,SAAS;gBACnB,iBAAiB,EAAE,UAAU;gBAC7B,2BAA2B,EAAE,wBAAwB;gBACrD,OAAO,EAAE,2EAA2E;aACrF,CAAC;QACJ,CAAC;QACD,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;YAC/B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,QAAQ,EAAE,QAAQ;gBAClB,iBAAiB,EAAE,cAAc;gBACjC,2BAA2B,EACzB,KAAK,CAAC,iBAAiB,IAAI,IAAI,CAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,CAAC;gBACnE,OAAO,EAAE,KAAK,CAAC,OAAO;aACvB,CAAC;QACJ,CAAC;QACD,OAAO;YACL,OAAO,EAAE,KAAK;YACd,QAAQ,EAAE,QAAQ;YAClB,iBAAiB,EAAE,cAAc;YACjC,2BAA2B,EAAE,IAAI,CAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,CAAC;YACnE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAChE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,cAAsC;IAChE,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;IACrC,KAAK,MAAM,GAAG,IAAI,CAAC,uBAAuB,EAAE,iBAAiB,EAAE,gBAAgB,CAAC,EAAE,CAAC;QACjF,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,KAAK;YAAE,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;IAChC,OAAO,KAAK,CAAC,CAAC,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC;AAChE,CAAC;AAED,SAAS,yBAAyB,CAChC,OAAgB,EAChB,wBAAgC;IAEhC,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACnC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACpB,MAAM,IAAI,QAAQ,CAAC,6DAA6D,CAAC,CAAC;IACpF,CAAC;IACD,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;IACtD,IAAI,MAAM,KAAK,UAAU,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,cAAc,EAAE,CAAC;QACjF,MAAM,IAAI,QAAQ,CAAC,qEAAqE,CAAC,CAAC;IAC5F,CAAC;IACD,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAC3D,OAAO;QACL,OAAO,EAAE,OAAO,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU;QACjF,QAAQ,EACN,QAAQ,KAAK,IAAI,IAAI,QAAQ,KAAK,UAAU,IAAI,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;QAC9F,iBAAiB,EAAE,MAAM;QACzB,2BAA2B,EACzB,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,wBAAwB;QACpF,SAAS,EAAE,WAAW,CAAC,IAAI,EAAE,WAAW,CAAC;QACzC,OAAO,EACL,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC;YAC5B,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;KAC5F,CAAC;AACJ,CAAC"}

package/dist/kb/selector.d.ts

@@ -0,0 +1,3 @@
+import { type ParsedArgs } from "../args.js";
+import type { CollectionSelector } from "./client.js";
+export declare function resolveCollectionSelector(args: ParsedArgs, env: NodeJS.ProcessEnv): CollectionSelector;

package/dist/kb/selector.js

@@ -0,0 +1,40 @@
+import { getString } from "../args.js";
+import { firstEnv } from "../env.js";
+import { CliError } from "../errors.js";
+export function resolveCollectionSelector(args, env) {
+    const selectors = [];
+    addSelector(selectors, "primary_collection_id", getString(args, "collection-id"));
+    addSelector(selectors, "collection_path", getString(args, "collection-path"));
+    addSelector(selectors, "collection_key", getString(args, "collection-key"));
+    addSelector(selectors, "collection_name", getString(args, "collection-name"));
+    if (selectors.length > 1) {
+        throw new CliError("Provide exactly one collection selector.");
+    }
+    if (selectors.length === 1)
+        return selectors[0];
+    const envName = firstEnv(env, "TIANGONG_KB_DEFAULT_COLLECTION_NAME");
+    if (envName)
+        return { field: "collection_name", value: envName };
+    const legacyName = firstEnv(env, "TIANGONG_KB_DEFAULT_COLLECTION_ID");
+    if (legacyName) {
+        if (isUuid(legacyName)) {
+            throw new CliError("TIANGONG_KB_DEFAULT_COLLECTION_ID is treated as a collection name. Use --collection-id for UUID uploads.");
+        }
+        return { field: "collection_name", value: legacyName };
+    }
+    const envPath = firstEnv(env, "TIANGONG_KB_DEFAULT_COLLECTION_PATH");
+    if (envPath)
+        return { field: "collection_path", value: envPath };
+    const envKey = firstEnv(env, "TIANGONG_KB_DEFAULT_COLLECTION_KEY");
+    if (envKey)
+        return { field: "collection_key", value: envKey };
+    throw new CliError("Missing collection selector. Provide --collection-name, --collection-key, --collection-path, --collection-id, or set TIANGONG_KB_DEFAULT_COLLECTION_NAME.");
+}
+function addSelector(selectors, field, value) {
+    if (value)
+        selectors.push({ field, value });
+}
+function isUuid(value) {
+    return /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i.test(value);
+}
+//# sourceMappingURL=selector.js.map

package/dist/kb/selector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"selector.js","sourceRoot":"","sources":["../../src/kb/selector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAmB,MAAM,YAAY,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAGxC,MAAM,UAAU,yBAAyB,CACvC,IAAgB,EAChB,GAAsB;IAEtB,MAAM,SAAS,GAAyB,EAAE,CAAC;IAC3C,WAAW,CAAC,SAAS,EAAE,uBAAuB,EAAE,SAAS,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAClF,WAAW,CAAC,SAAS,EAAE,iBAAiB,EAAE,SAAS,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAC9E,WAAW,CAAC,SAAS,EAAE,gBAAgB,EAAE,SAAS,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAC5E,WAAW,CAAC,SAAS,EAAE,iBAAiB,EAAE,SAAS,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC,CAAC;IAE9E,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,QAAQ,CAAC,0CAA0C,CAAC,CAAC;IACjE,CAAC;IACD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC,CAAC,CAAuB,CAAC;IAEtE,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,EAAE,qCAAqC,CAAC,CAAC;IACrE,IAAI,OAAO;QAAE,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAEjE,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,EAAE,mCAAmC,CAAC,CAAC;IACtE,IAAI,UAAU,EAAE,CAAC;QACf,IAAI,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,QAAQ,CAChB,0GAA0G,CAC3G,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;IACzD,CAAC;IAED,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,EAAE,qCAAqC,CAAC,CAAC;IACrE,IAAI,OAAO;QAAE,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAEjE,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,EAAE,oCAAoC,CAAC,CAAC;IACnE,IAAI,MAAM;QAAE,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;IAE9D,MAAM,IAAI,QAAQ,CAChB,2JAA2J,CAC5J,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAClB,SAA+B,EAC/B,KAAkC,EAClC,KAAyB;IAEzB,IAAI,KAAK;QAAE,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,MAAM,CAAC,KAAa;IAC3B,OAAO,4EAA4E,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAClG,CAAC"}

package/dist/kb/status.d.ts

@@ -0,0 +1,19 @@
+import type { KbConfig } from "./config.js";
+export interface BulkStatusItem {
+    documentId: string;
+    status?: string | undefined;
+    terminal?: boolean | undefined;
+    opensearchIndexed?: boolean | undefined;
+    pineconeIndexed?: boolean | undefined;
+    indexRecordCount?: number | undefined;
+    lastError?: string | undefined;
+    lastErrorStage?: string | undefined;
+    itemError?: {
+        code: string;
+        message: string;
+        retryable: boolean;
+    };
+    raw: unknown;
+}
+export declare function batchDocumentStatuses(config: KbConfig, documentIds: string[]): Promise<BulkStatusItem[]>;
+export declare function getDocumentStatus(config: KbConfig, documentId: string): Promise<unknown>;

package/dist/kb/status.js

@@ -0,0 +1,74 @@
+import { isObject, responseData, stringField } from "../data.js";
+import { HttpError } from "../errors.js";
+import { jsonRequest } from "../http.js";
+const KB_DOCUMENT_STATUS_BATCH_LIMIT = 100;
+export async function batchDocumentStatuses(config, documentIds) {
+    const statuses = [];
+    for (let index = 0; index < documentIds.length; index += KB_DOCUMENT_STATUS_BATCH_LIMIT) {
+        const chunk = documentIds.slice(index, index + KB_DOCUMENT_STATUS_BATCH_LIMIT);
+        try {
+            const payload = await jsonRequest(config, "documents/status:batch", {
+                method: "POST",
+                body: JSON.stringify({ documentIds: chunk }),
+            });
+            statuses.push(...batchStatusItems(payload));
+        }
+        catch (error) {
+            if (error instanceof HttpError && error.status && [404, 405, 501].includes(error.status)) {
+                statuses.push(...(await Promise.all(chunk.map(async (documentId) => statusItemFromPayload(await getDocumentStatus(config, documentId), documentId)))));
+                continue;
+            }
+            throw error;
+        }
+    }
+    return statuses;
+}
+export async function getDocumentStatus(config, documentId) {
+    return jsonRequest(config, `documents/${encodeURIComponent(documentId)}/status`);
+}
+function batchStatusItems(payload) {
+    const data = responseData(payload);
+    const items = isObject(data) && Array.isArray(data.documents)
+        ? data.documents
+        : isObject(data) && Array.isArray(data.results)
+            ? data.results
+            : Array.isArray(data)
+                ? data
+                : [];
+    return items.filter(isObject).map((item) => batchStatusItemFromPayload(item));
+}
+function statusItemFromPayload(payload, fallbackDocumentId = "") {
+    const data = responseData(payload);
+    const item = isObject(data) ? data : {};
+    return {
+        documentId: stringField(item, "documentId") ?? stringField(item, "document_id") ?? fallbackDocumentId,
+        status: stringField(item, "status"),
+        terminal: typeof item.terminal === "boolean" ? item.terminal : undefined,
+        opensearchIndexed: typeof item.opensearchIndexed === "boolean" ? item.opensearchIndexed : undefined,
+        pineconeIndexed: typeof item.pineconeIndexed === "boolean" ? item.pineconeIndexed : undefined,
+        indexRecordCount: typeof item.indexRecordCount === "number" ? item.indexRecordCount : undefined,
+        lastError: typeof item.lastError === "string" ? item.lastError : undefined,
+        lastErrorStage: typeof item.lastErrorStage === "string" ? item.lastErrorStage : undefined,
+        raw: payload,
+    };
+}
+function batchStatusItemFromPayload(item) {
+    const documentId = stringField(item, "documentId") ?? stringField(item, "document_id") ?? "";
+    if (item.ok === true && isObject(item.status)) {
+        return statusItemFromPayload(item.status, documentId);
+    }
+    if (item.ok === false && isObject(item.error)) {
+        const error = item.error;
+        return {
+            documentId,
+            itemError: {
+                code: stringField(error, "code") ?? "STATUS_ITEM_ERROR",
+                message: stringField(error, "message") ?? "Document status lookup failed.",
+                retryable: typeof error.retryable === "boolean" ? error.retryable : false,
+            },
+            raw: item,
+        };
+    }
+    return statusItemFromPayload(item, documentId);
+}
+//# sourceMappingURL=status.js.map

package/dist/kb/status.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"status.js","sourceRoot":"","sources":["../../src/kb/status.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACjE,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAGzC,MAAM,8BAA8B,GAAG,GAAG,CAAC;AAmB3C,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,MAAgB,EAChB,WAAqB;IAErB,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,WAAW,CAAC,MAAM,EAAE,KAAK,IAAI,8BAA8B,EAAE,CAAC;QACxF,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,8BAA8B,CAAC,CAAC;QAC/E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,wBAAwB,EAAE;gBAClE,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;aAC7C,CAAC,CAAC;YACH,QAAQ,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,SAAS,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzF,QAAQ,CAAC,IAAI,CACX,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CACnB,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CAC7B,qBAAqB,CAAC,MAAM,iBAAiB,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,UAAU,CAAC,CAC/E,CACF,CAAC,CACH,CAAC;gBACF,SAAS;YACX,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAgB,EAAE,UAAkB;IAC1E,OAAO,WAAW,CAAC,MAAM,EAAE,aAAa,kBAAkB,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAgB;IACxC,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACnC,MAAM,KAAK,GACT,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;QAC7C,CAAC,CAAC,IAAI,CAAC,SAAS;QAChB,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;YAC7C,CAAC,CAAC,IAAI,CAAC,OAAO;YACd,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;gBACnB,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,EAAE,CAAC;IACb,OAAO,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAgB,EAAE,kBAAkB,GAAG,EAAE;IACtE,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACnC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IACxC,OAAO;QACL,UAAU,EACR,WAAW,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,WAAW,CAAC,IAAI,EAAE,aAAa,CAAC,IAAI,kBAAkB;QAC3F,MAAM,EAAE,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC;QACnC,QAAQ,EAAE,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;QACxE,iBAAiB,EACf,OAAO,IAAI,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;QAClF,eAAe,EAAE,OAAO,IAAI,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS;QAC7F,gBAAgB,EAAE,OAAO,IAAI,CAAC,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS;QAC/F,SAAS,EAAE,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QAC1E,cAAc,EAAE,OAAO,IAAI,CAAC,cAAc,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;QACzF,GAAG,EAAE,OAAO;KACb,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CAAC,IAA6B;IAC/D,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,WAAW,CAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,CAAC;IAC7F,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,OAAO,qBAAqB,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,IAAI,CAAC,EAAE,KAAK,KAAK,IAAI,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,OAAO;YACL,UAAU;YACV,SAAS,EAAE;gBACT,IAAI,EAAE,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,mBAAmB;gBACvD,OAAO,EAAE,WAAW,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,gCAAgC;gBAC1E,SAAS,EAAE,OAAO,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;aAC1E;YACD,GAAG,EAAE,IAAI;SACV,CAAC;IACJ,CAAC;IACD,OAAO,qBAAqB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;AACjD,CAAC"}

package/dist/research/commands.d.ts

@@ -0,0 +1,2 @@
+import type { CliIO } from "../io.js";
+export declare function runResearchCommand(argv: string[], io: CliIO): Promise<number>;