File size: 1,484 Bytes
9ada4bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import { Client } from "../client";
import type { Dependency, PredictReturn } from "../types";
export async function predict(
this: Client,
endpoint: string | number,
data: unknown[] | Record<string, unknown>
): Promise<PredictReturn> {
let data_returned = false;
let status_complete = false;
let dependency: Dependency;
if (!this.config) {
throw new Error("Could not resolve app config");
}
if (typeof endpoint === "number") {
dependency = this.config.dependencies.find((dep) => dep.id == endpoint)!;
} else {
const trimmed_endpoint = endpoint.replace(/^\//, "");
dependency = this.config.dependencies.find(
(dep) => dep.id == this.api_map[trimmed_endpoint]
)!;
}
if (dependency?.types.continuous) {
throw new Error(
"Cannot call predict on this function as it may run forever. Use submit instead"
);
}
return new Promise(async (resolve, reject) => {
const app = this.submit(endpoint, data, null, null, true);
let result: unknown;
for await (const message of app) {
if (message.type === "data") {
if (status_complete) {
resolve(result as PredictReturn);
}
data_returned = true;
result = message;
}
if (message.type === "status") {
if (message.stage === "error") reject(message);
if (message.stage === "complete") {
status_complete = true;
// if complete message comes after data, resolve here
if (data_returned) {
resolve(result as PredictReturn);
}
}
}
}
});
}
|