export async function predictText()

in packages/cli/src/utils/serve-predict.ts [57:74]

• 18 lines of code
• 5 McCabe index (conditional complexity)

export async function predictText(
  predictCallback: PredictCallBack,
  req: Request,
  res: Response
): Promise<void> {
  if (req.query && req.query['input']) {
    let inputs: string[];
    if (Array.isArray(req.query['input'])) {
      inputs = req.query['input'] as string[];
    } else if (typeof req.query['input'] === 'string') {
      inputs = [ req.query['input'] ];
    }
    const result = await predictCallback(inputs);
    res.json({ success: true, data: result });
  } else {
    res.json({ success: false, message: 'no input available' });
  }
}