relax the test message shape to support incoming refusal

leafo · Aug 7, 2024 · f648419 · f648419
1 parent cf62d07
commit f648419
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/openai/init.lua b/openai/init.lua
@@ -19,7 +19,7 @@ local content_format = types.string + types.array_of(types.one_of({
   })
 }))
 local test_message = types.one_of({
-  types.shape({
+  types.partial({
     role = types.one_of({
       "system",
       "user",
@@ -29,7 +29,7 @@ local test_message = types.one_of({
     name = empty + types.string,
     function_call = empty + types.table
   }),
-  types.shape({
+  types.partial({
     role = types.one_of({
       "function"
     }),

diff --git a/openai/init.moon b/openai/init.moon
@@ -19,15 +19,15 @@ content_format = types.string + types.array_of types.one_of {
 }
 
 test_message = types.one_of {
-  types.shape {
+  types.partial {
     role: types.one_of {"system", "user", "assistant"}
     content: empty + content_format -- this can be empty when function_call is set
     name: empty + types.string
     function_call: empty + types.table
   }
 
   -- this message type is for sending a function call response back
-  types.shape {
+  types.partial {
     role: types.one_of {"function"}
     name: types.string
     content: empty + types.string