lib/gen_ai/result.rb in gen-ai-0.1.0 vs lib/gen_ai/result.rb in gen-ai-0.2.0

@@ -31,9 +31,13 @@
     end
 
     private
 
     def usage
-      raw['usage']
+      raw['usage'] || {
+        'prompt_tokens' => nil,
+        'completion_tokens' => nil,
+        'total_tokens' => nil
+      }
     end
   end
 end