summarize

labscore/api/obelisk/endpoints.js

@@ -0,0 +1,15 @@
+const ObeliskHosts = Object.freeze({
+  prod: process.env.OBELISK_HOST,
+  local: "http://localhost"
+})
+
+const ObeliskApi = Object.freeze({
+  HOST: ObeliskHosts.prod,
+
+  SUMMARIZE_WEBPAGES: "/flamingo/v1/web:summarize"
+})
+
+module.exports = {
+  ObeliskApi,
+  ObeliskHosts
+}

labscore/api/obelisk/index.js

@@ -0,0 +1,47 @@
+const superagent = require('superagent')
+const { ObeliskApi, ObeliskHosts } = require('./endpoints')
+
+async function request(path, type, headers, args, host) {
+  let timing = Date.now();
+  url = ObeliskApi.HOST + path
+  if(process.env.USE_LOCAL_API) url = ObeliskHosts.local + ":" + process.env.USE_LOCAL_API + path
+  if(host) url = host + path
+
+  // apply default headers
+  if(!headers["Authorization"]) headers["Authorization"] = process.env.OBELISK_API_KEY
+  if(!headers["x-obelisk-client"]) headers["x-obelisk-client"] = process.env.OBELISK_CLIENT_ID || "labscore-production-001"
+
+  if (type === "GET") {
+    if(!args){
+      const response = await superagent.get(url);
+      return {
+        timings: ((Date.now() - timing) / 1000).toFixed(2),
+        response: response
+      };
+    }
+    const response = await superagent.get(url)
+      .query(args)
+      .set(headers);
+      return {
+        timings: ((Date.now() - timing) / 1000).toFixed(2),
+        response: response
+      };
+  }
+  if (type === "POST") {
+    const response = await superagent
+      .post(url)
+      .set(headers)
+      .send(args);
+    return {
+      timings: ((Date.now() - timing) / 1000).toFixed(2),
+      response: response
+    };
+  }
+  throw new Error("unsupported, must either use GET or POST");
+}
+
+module.exports.summarizeWebpage = async function(context, url){
+  return await request(ObeliskApi.SUMMARIZE_WEBPAGES, "POST", {}, {
+    url
+  })
+}