From 85a26a53935b6ee57a3f323d3d352cfeef14f37e Mon Sep 17 00:00:00 2001
From: NeuralFlux <40491005+NeuralFlux@users.noreply.github.com>
Date: Tue, 27 Aug 2024 17:03:07 -0400
Subject: [PATCH] fix: add `query_id` to inferred mode results if it exists
 they are already present in TRAPIQueryHandler results after
 biothings/biothings_explorer#815

---
 src/inferred_mode/inferred_mode.ts | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/inferred_mode/inferred_mode.ts b/src/inferred_mode/inferred_mode.ts
index 96f95b40..a235d34e 100644
--- a/src/inferred_mode/inferred_mode.ts
+++ b/src/inferred_mode/inferred_mode.ts
@@ -11,6 +11,7 @@ import {
   TrapiAuxGraphCollection,
   TrapiEdgeBinding,
   TrapiKnowledgeGraph,
+  TrapiNodeBinding,
   TrapiQEdge,
   TrapiQNode,
   TrapiQualifier,
@@ -286,10 +287,20 @@ export default class InferredQueryHandler {
     });
     // add results
     newResponse.message.results.forEach((result) => {
+      // get query_ids populated by TRAPIQueryHandler.appendOriginalCuriesToResults
+      const subjectBinding: TrapiNodeBinding = { id: result.node_bindings.creativeQuerySubject[0].id, attributes: [] };
+      const objectBinding: TrapiNodeBinding = { id: result.node_bindings.creativeQueryObject[0].id, attributes: [] };
+      if (result.node_bindings.creativeQuerySubject[0].query_id !== undefined) {
+        subjectBinding.query_id = result.node_bindings.creativeQuerySubject[0].query_id;
+      }
+      if (result.node_bindings.creativeQueryObject[0].query_id !== undefined) {
+        objectBinding.query_id = result.node_bindings.creativeQueryObject[0].query_id;
+      }
+
       const translatedResult: TrapiResult = {
         node_bindings: {
-          [qEdge.subject]: [{ id: result.node_bindings.creativeQuerySubject[0].id, attributes: [] }],
-          [qEdge.object]: [{ id: result.node_bindings.creativeQueryObject[0].id, attributes: [] }],
+          [qEdge.subject]: [subjectBinding],
+          [qEdge.object]: [objectBinding],
         },
         pfocr: result.pfocr?.length ? result.pfocr : undefined,
         analyses: [