Make keys required field, add schema generator

requirements-dev.txt

@@ -0,0 +1,2 @@
+-r requirements.txt
+pydantic

schema.json

@@ -13,7 +13,7 @@
     },
     "description": {
       "title": "Description",
-      "default": "An optional brief description of the system that will be shown on the website",
+      "default": "An optional brief description of the system that will be shown on the results page",
       "type": "string"
     },
     "tasks": {
@@ -43,10 +43,6 @@
         },
         "keys": {
           "title": "Keys",
-          "default": [
-            "schema_guided_dialog-test-9585",
-            "schema_guided_dialog-test-9585"
-          ],
           "type": "array",
           "items": {
             "type": "string"
@@ -54,7 +50,8 @@
         }
       },
       "required": [
-        "values"
+        "values",
+        "keys"
       ]
     }
   }

schema.py

@@ -0,0 +1,27 @@
+from typing import Dict, List
+
+from pydantic import BaseModel
+
+
+class Task(BaseModel):
+    values: List[str]
+    keys: List[str]
+
+
+class GEMSchema(BaseModel):
+    """The default GEM schema"""
+
+    submission_name: str
+    param_count: int
+    description: str = "An optional brief description of the system that will be shown on the results page"
+    tasks: Dict[str, Task]
+
+
+def main():
+    schema_data = GEMSchema.schema_json(indent=2)
+    with open("schema.json", "w", encoding="utf-8") as f:
+        f.write(schema_data)
+
+
+if __name__ == "__main__":
+    main()