Log predictions to Aporia from Node.js

log_predictions.js

import fetch from "node-fetch";

const APORIA_TOKEN = "<TOKEN>";

async function logPredictions(data) {
  const query = `
    mutation LogPredict(
      $modelId: String!,
      $modelVersion: String!,
      $environment: String!,
      $predictions: [Prediction]!
      $isSync: Boolean!
    ) {
      logPredictions(
        modelId: $modelId,
        modelVersion: $modelVersion,
        environment: $environment,
        predictions: $predictions
        isSync: $isSync
      ) {
        warnings
      }
    }
  `

  const response = await fetch("https://app.aporia.com/v1/controller/graphql", {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      "Authorization": `Bearer ${APORIA_TOKEN}`,
    },
    body: JSON.stringify({
      query,
      variables: {
        ...data,
        isSync: true,
      },
    }),
  });

  const warnings = (await response.json()).data.logPredictions.warnings;
  if (warnings.length > 0) {
    throw new Error(`Failed to log predictions; ${warnings}`);
  }
}

Usage example:

await logPredictions({
  modelId: "my-model",
  modelVersion: "v1",
  environment: "prod",
  predictions: [{
    id: "some-prediction-id",
    occurredAt: new Date().toISOString(),
    features: {
      x: 0.5,
    },
    predictions: {
      y: true,
    },
  }],
})