diff options
Diffstat (limited to 'src/server/ApiManagers')
-rw-r--r-- | src/server/ApiManagers/UploadManager.ts | 1 | ||||
-rw-r--r-- | src/server/ApiManagers/UtilManager.ts | 26 |
2 files changed, 27 insertions, 0 deletions
diff --git a/src/server/ApiManagers/UploadManager.ts b/src/server/ApiManagers/UploadManager.ts index 8f2a5ea3e..f872bdf94 100644 --- a/src/server/ApiManagers/UploadManager.ts +++ b/src/server/ApiManagers/UploadManager.ts @@ -42,6 +42,7 @@ export default class UploadManager extends ApiManager { method: Method.POST, subscription: "/uploadFormData", secureHandler: async ({ req, res }) => { + console.log("/upload register"); const form = new formidable.IncomingForm(); form.uploadDir = pathToDirectory(Directory.parsed_files); form.keepExtensions = true; diff --git a/src/server/ApiManagers/UtilManager.ts b/src/server/ApiManagers/UtilManager.ts index 8adc3da81..d18529cf2 100644 --- a/src/server/ApiManagers/UtilManager.ts +++ b/src/server/ApiManagers/UtilManager.ts @@ -3,6 +3,11 @@ import { Method } from "../RouteManager"; import { exec } from 'child_process'; import RouteSubscriber from "../RouteSubscriber"; import { red } from "colors"; +import { IBM_Recommender } from "../../client/apis/IBM_Recommender"; +import { Recommender } from "../Recommender"; + +const recommender = new Recommender(); +recommender.testModel(); import executeImport from "../../scraping/buxton/final/BuxtonImporter"; export default class UtilManager extends ApiManager { @@ -23,6 +28,27 @@ export default class UtilManager extends ApiManager { }); register({ + method: Method.POST, + subscription: "/IBMAnalysis", + secureHandler: async ({ req, res }) => res.send(await IBM_Recommender.analyze(req.body)) + }); + + register({ + method: Method.POST, + subscription: "/recommender", + secureHandler: async ({ req, res }) => { + const keyphrases = req.body.keyphrases; + const wordvecs = await recommender.vectorize(keyphrases); + let embedding: Float32Array = new Float32Array(); + if (wordvecs && wordvecs.dataSync()) { + embedding = wordvecs.dataSync() as Float32Array; + } + res.send(embedding); + } + }); + + + register({ method: Method.GET, subscription: "/pull", secureHandler: async ({ res }) => { |