Merge pull request #3 from julep-ai/creatorrr/restrict-role-values

Restrict role values in chatml-schema
julep-ai · May 27, 2024 · dbb15e6 · dbb15e6
2 parents d0c914c + 12bdbd6
commit dbb15e6
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/specs/chatml-schema.json b/specs/chatml-schema.json
@@ -4,7 +4,14 @@
   "type": "object",
   "properties": {
     "role": {
-      "type": "string"
+      "type": "string",
+      "enum": [
+        "system",
+        "user",
+        "assistant",
+        "tool_calls",
+        "tool_response"
+      ]
     },
     "name": {
       "type": "string"
@@ -20,5 +27,5 @@
       ]
     }
   },
-  "required": ["role", "name", "content"]
+  "required": ["role", "content"]
 }