add test suite for functions chat session flow

Author: leafo

spec/openai_spec.moon

@@ -1,5 +1,7 @@
 import OpenAI from require "openai"
 
+cjson = require "cjson"
+
 describe "OpenAI API Client", ->
   before_each ->
     package.loaded["ssl.https"] = {
@@ -35,7 +37,6 @@ describe "OpenAI API Client", ->
           }
 
         status = 200
-        cjson = require "cjson"
         opts.sink cjson.encode(response)
         true, status, {}
     }
@@ -59,10 +60,166 @@ describe "OpenAI API Client", ->
     client = OpenAI "test-api-key"
     status, response = assert client\completion "Tell me a joke."
 
-
     assert.same 200, status
     assert.same "This is a completion response.", response.choices[1].text
 
+  describe "chat session #ddd", ->
+    local snapshot
+    before_each = ->
+      snapshot = assert\snapshot!
+
+    after_each = ->
+      snapshot\revert!
+
+    it "simple exchange", ->
+
+    it "handles error", ->
+
+    it "with functions", ->
+      stub(OpenAI.__base, "chat").invokes (c, args, params) ->
+        assert.same {
+          {
+            role: "system"
+            content: "You are a calculator with access to specified set of functions. All computation should be done with the functions"
+          }
+          {
+            role: "user"
+            content: "Calculate the square root of 23892391"
+          }
+        }, args
+
+        assert.same {
+          model: "gpt-4-0613"
+          functions: {
+            {
+              name: "sqrt"
+              description: "Calculate square root of a number"
+              parameters: {
+                type: "object"
+                properties: {
+                  a: { type: "number" }
+                }
+              }
+            }
+          }
+        }, params
+
+        200, {
+          usage: {}
+          choices: {
+            {
+              message: {
+                role: "assistant"
+                function_call: {
+                  name: "sqrt"
+                  arguments: [[{ "a": 23892391 }]]
+                }
+                content: cjson.null
+              }
+            }
+          }
+        }
+
+      client = OpenAI "test-api-key"
+      chat = client\new_chat_session {
+        model: "gpt-4-0613"
+        messages: {
+          {
+            role: "system"
+            content: "You are a calculator with access to specified set of functions. All computation should be done with the functions"
+          }
+        }
+        functions: {
+          {
+            name: "sqrt"
+            description: "Calculate square root of a number"
+            parameters: {
+              type: "object"
+              properties: {
+                a: { type: "number" }
+              }
+            }
+          }
+        }
+      }
+
+      res = assert chat\send "Calculate the square root of 23892391"
+
+      -- returns message object instead of string result due to
+      -- function call
+      assert.same {
+        role: "assistant"
+        function_call: {
+          name: "sqrt"
+          arguments: [[{ "a": 23892391 }]]
+        }
+        content: cjson.null
+      }, res
+
+
+      stub(OpenAI.__base, "chat").invokes (c, args, params) ->
+        assert.same {
+          {
+            role: "system"
+            content: "You are a calculator with access to specified set of functions. All computation should be done with the functions"
+          }
+          {
+            role: "user"
+            content: "Calculate the square root of 23892391"
+          }
+          {
+            role: "assistant"
+            function_call: {
+              name: "sqrt"
+              arguments: [[{ "a": 23892391 }]]
+            }
+            content: cjson.null -- it must preserve null values
+          }
+          {
+            role: "function"
+            name: "sqrt"
+            content: "99"
+          }
+        }, args
+
+        assert.same {
+          model: "gpt-4-0613"
+          functions: {
+            {
+              name: "sqrt"
+              description: "Calculate square root of a number"
+              parameters: {
+                type: "object"
+                properties: {
+                  a: { type: "number" }
+                }
+              }
+            }
+          }
+        }, params
+
+        200, {
+          usage: {}
+          choices: {
+            {
+              message: {
+                role: "assistant"
+                content: "Good work!"
+              }
+            }
+          }
+        }
+
+      -- send the response
+      res = assert chat\send {
+        role: "function"
+        name: "sqrt"
+        content: "99"
+      }
+
+      assert.same "Good work!", res
+
+
   describe "streaming", ->
     before_each ->
       package.loaded["ssl.https"] = {
@@ -83,9 +240,7 @@ describe "OpenAI API Client", ->
       }
 
     it "processes streaming chunks", ->
-      client = OpenAI "test-api-key", {
-        http_provider: http_stream_stub
-      }
+      client = OpenAI "test-api-key"
       chat = client\new_chat_session {
         messages: {
           {role: "user", content: "tell me a joke"}