omniai 1.6.6 → 1.8.0

data/lib/omniai/chat/response/tool_call.rb

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A tool-call returned by the API.
-      class ToolCall < Resource
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} id=#{id.inspect} type=#{type.inspect}>"
-        end
-
-        # @return [String]
-        def id
-          @data['id']
-        end
-
-        # @return [String]
-        def type
-          @data['type']
-        end
-
-        # @return [Function]
-        def function
-          @function ||= Function.new(data: @data['function']) if @data['function']
-        end
-      end
-    end
-  end
-end

data/lib/omniai/chat/response/usage.rb

@@ -1,35 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A usage returned by the API.
-      class Usage < Resource
-        # @return [String]
-        def inspect
-          properties = [
-            "input_tokens=#{input_tokens}",
-            "output_tokens=#{output_tokens}",
-            "total_tokens=#{total_tokens}",
-          ]
-          "#<#{self.class.name} #{properties.join(' ')}>"
-        end
-
-        # @return [Integer]
-        def input_tokens
-          @data['input_tokens'] || @data['prompt_tokens']
-        end
-
-        # @return [Integer]
-        def output_tokens
-          @data['output_tokens'] || @data['completion_tokens']
-        end
-
-        # @return [Integer]
-        def total_tokens
-          @data['total_tokens'] || (input_tokens + output_tokens)
-        end
-      end
-    end
-  end
-end