feat: handling for non-array values

biothings · Apr 12, 2024 · ac88bfc · ac88bfc
1 parent 46e8298
commit ac88bfc
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/graph/knowledge_graph.ts b/src/graph/knowledge_graph.ts
@@ -17,6 +17,8 @@ import { APIDefinition } from '@biothings-explorer/types';
 
 const debug = Debug('bte:biothings-explorer-trapi:KnowledgeGraph');
 
+const NON_ARRAY_ATTRIBUTES = ['biolink:knowledge_level', 'biolink:agent_type'];
+
 export default class KnowledgeGraph {
   nodes: {
     [nodePrimaryID: string]: TrapiKGNode;
@@ -114,11 +116,14 @@ export default class KnowledgeGraph {
     }
 
     Object.entries(kgEdge.attributes).forEach(([key, value]) => {
-      if (key == 'edge-attributes') return;
+      if (key === 'edge-attributes') return;
       // if (key == 'edge-attributes') return;
       attributes.push({
         attribute_type_id: key,
-        value: Array.from(value as Set<string>),
+        value: // technically works for numbers as well
+          NON_ARRAY_ATTRIBUTES.includes(key)
+            ? [...(value as Set<string>)].reduce((acc, val) => acc + val)
+            : Array.from(value as Set<string>),
         //value_type_id: 'bts:' + key,
       });
     });