Merge pull request #25 from biothings/fix_issue_23

Add type node attributes in TRAPI response
biothings · Feb 1, 2021 · 5eb3fef · 5eb3fef
2 parents 7fecd84 + 898c59e
commit 5eb3fef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/__test__/integration/KnowledgeGraph.test.js b/__test__/integration/KnowledgeGraph.test.js
@@ -29,6 +29,7 @@ describe("Testing KnowledgeGraph Module", () => {
             expect(res).toHaveProperty("NCBIGene:1017");
             expect(res["NCBIGene:1017"]).toHaveProperty("name", "CDK2");
             expect(res["NCBIGene:1017"]).toHaveProperty("category", "biolink:Gene");
+            expect(res["NCBIGene:1017"].attributes[0]).toHaveProperty("type", "biolink:id");
         })
     })
 

diff --git a/src/controllers/QueryGraphHandler/knowledge_graph.js b/src/controllers/QueryGraphHandler/knowledge_graph.js
@@ -27,7 +27,8 @@ module.exports = class KnowledgeGraph {
                 attributes: [
                     {
                         name: "equivalent_identifiers",
-                        value: helper._getInputEquivalentIds(record)
+                        value: helper._getInputEquivalentIds(record),
+                        type: "biolink:id"
                     }
                 ]
             }
@@ -42,7 +43,8 @@ module.exports = class KnowledgeGraph {
                 attributes: [
                     {
                         name: "equivalent_identifiers",
-                        value: helper._getOutputEquivalentIds(record)
+                        value: helper._getOutputEquivalentIds(record),
+                        type: "biolink:id"
                     }
                 ]
             }