pfocr enrichment for pathfinder

biothings · Sep 16, 2024 · 066a273 · 066a273
1 parent cbb8497
commit 066a273
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/inferred_mode/inferred_mode.ts b/src/inferred_mode/inferred_mode.ts
@@ -713,7 +713,9 @@ export default class InferredQueryHandler {
     this.pruneKnowledgeGraph(response);
 
     // add pfocr figures
-    this.logs = [...this.logs, ...(await enrichTrapiResultsWithPfocrFigures(response, true))];
+    if (!this.pathfinder) {
+      this.logs = [...this.logs, ...(await enrichTrapiResultsWithPfocrFigures(response, true))];
+    }
 
     // get the final summary log
     if (successfulQueries) {

diff --git a/src/inferred_mode/pathfinder.ts b/src/inferred_mode/pathfinder.ts
@@ -15,6 +15,7 @@ import Debug from 'debug';
 import generateTemplates from './pf_template_generator';
 import biolink from '../biolink';
 import { removeBioLinkPrefix } from '../utils';
+import { enrichTrapiResultsWithPfocrFigures } from '../results_assembly/pfocr';
 const debug = Debug('bte:biothings-explorer-trapi:pathfinder');
 
 interface ResultAuxObject {
@@ -116,6 +117,9 @@ export default class PathfinderQueryHandler {
     this.parse(creativeResponse);
     this._pruneKg(creativeResponse);
 
+    // pfocr
+    this.logs = [...this.logs, ...(await enrichTrapiResultsWithPfocrFigures(creativeResponse, true))];
+
     // logs
     creativeResponse.logs = this.logs.map((log) => log.toJSON());