feat(chat.cr): add helper for tracking token usage

spider-gazelle · Nov 3, 2023 · f5d57e6 · f5d57e6
1 parent b6c669a
commit f5d57e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/openai/api/chat.cr b/src/openai/api/chat.cr
@@ -107,7 +107,10 @@ module OpenAI
     # The name and arguments of a function that should be called, as generated by the model.
     getter function_call : ChatFunctionCall?
 
-    def initialize(@role, @content = nil, @name = nil, @function_call = nil)
+    @[JSON::Field(ignore: true)]
+    property tokens : Int32 = 0
+
+    def initialize(@role, @content = nil, @name = nil, @function_call = nil, @tokens = 0)
     end
   end