Merge 3cce435 into d6e4b66

biothings · May 23, 2022 · e69f7f6 · e69f7f6
2 parents d6e4b66 + 3cce435
commit e69f7f6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/src/transformers/biothings_transformer.ts b/src/transformers/biothings_transformer.ts
@@ -18,8 +18,7 @@ export default class BioThingsTransformer extends BaseTransformer {
             });
             return res;
         } else {
-            let _input = generateCurie(this.edge.association.input_id, this.edge.input);
-            return { [_input]: [this.data["response"]] }
+            return super.pairCurieWithAPIResponse();
         }
 
     }

diff --git a/src/transformers/transformer.ts b/src/transformers/transformer.ts
@@ -24,7 +24,10 @@ export default class BaseTransformer {
      * Create an object with key representing input, and value representing the output of API
      */
     pairCurieWithAPIResponse() {
-        let input = generateCurie(this.edge.association.input_id, this.edge.input);
+        let input = generateCurie(
+            this.edge.association.input_id,
+            this.edge.input.hasOwnProperty('queryInputs') ? this.edge.input["queryInputs"] : this.edge.input as string
+        );
         return {
             [input]: [this.data.response],
         };

diff --git a/src/types.ts b/src/types.ts
@@ -35,8 +35,12 @@ interface SmartAPIKGOperationObject {
     tags?: string[];
 }
 
+interface templatedInput {
+    queryInputs: any;
+}
+
 export interface BTEKGOperationObject extends SmartAPIKGOperationObject {
-    input: string | string[];
+    input: string | string[] | templatedInput;
     reasoner_edge?: any;
     filter?: string;
     original_input?: object;