Fix collection JSON keys

RevEngAI · Jul 29, 2024 · 863ba6a · 863ba6a
1 parent 77bd21b
commit 863ba6a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/ai/reveng/toolkit/ghidra/core/services/api/types/Collection.java b/src/main/java/ai/reveng/toolkit/ghidra/core/services/api/types/Collection.java
@@ -16,8 +16,8 @@ public record Collection(
     public static Collection fromJSONObject(JSONObject json){
         return new Collection(
                 json.getInt("collection_id"),
-                json.getString("collection_scope"),
-                json.getString("collection_name"),
+                json.getString("scope"),
+                json.getString("name"),
                 json.getString("last_updated"),
                 json.getString("model_name"),
                 json.getJSONArray("tags").toList().stream().map(Object::toString).toList()
@@ -27,8 +27,8 @@ public static Collection fromJSONObject(JSONObject json){
     public static Collection fromSmallJSONObject(JSONObject json, ModelName modelName){
         return new Collection(
                 -1,
-                json.getString("collection_scope"),
-                json.getString("collection_name"),
+                json.getString("scope"),
+                json.getString("name"),
                 "",
                 modelName.modelName(),
                 List.of()