omniai 1.7.0 → 1.8.1

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.7.0
+  version: 1.8.1
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-02 00:00:00.000000000 Z
+date: 2024-08-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -68,27 +68,23 @@ files:
 - exe/omniai
 - lib/omniai.rb
 - lib/omniai/chat.rb
+- lib/omniai/chat/choice.rb
 - lib/omniai/chat/content.rb
 - lib/omniai/chat/file.rb
+- lib/omniai/chat/function.rb
 - lib/omniai/chat/media.rb
 - lib/omniai/chat/message.rb
+- lib/omniai/chat/message/builder.rb
+- lib/omniai/chat/payload.rb
 - lib/omniai/chat/prompt.rb
-- lib/omniai/chat/response/choice.rb
-- lib/omniai/chat/response/chunk.rb
-- lib/omniai/chat/response/completion.rb
-- lib/omniai/chat/response/delta.rb
-- lib/omniai/chat/response/delta_choice.rb
-- lib/omniai/chat/response/function.rb
-- lib/omniai/chat/response/message.rb
-- lib/omniai/chat/response/message_choice.rb
-- lib/omniai/chat/response/part.rb
-- lib/omniai/chat/response/payload.rb
-- lib/omniai/chat/response/resource.rb
-- lib/omniai/chat/response/stream.rb
-- lib/omniai/chat/response/tool_call.rb
-- lib/omniai/chat/response/usage.rb
+- lib/omniai/chat/response.rb
+- lib/omniai/chat/stream.rb
 - lib/omniai/chat/text.rb
+- lib/omniai/chat/tool_call.rb
+- lib/omniai/chat/tool_call_message.rb
+- lib/omniai/chat/tool_call_result.rb
 - lib/omniai/chat/url.rb
+- lib/omniai/chat/usage.rb
 - lib/omniai/cli.rb
 - lib/omniai/cli/base_handler.rb
 - lib/omniai/cli/chat_handler.rb

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

@@ -1,35 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # For use with MessageChoice or DeltaChoice.
-      class Choice < Resource
-        # @return [Integer]
-        def index
-          @data['index']
-        end
-
-        # @return [Part]
-        def part
-          raise NotImplementedError, "#{self.class.name}#part undefined"
-        end
-
-        # @return [ToolCallList]
-        def tool_call_list
-          part.tool_call_list
-        end
-
-        # @return [String, nil]
-        def content
-          part.content
-        end
-
-        # @return [Boolean]
-        def content?
-          !content.nil?
-        end
-      end
-    end
-  end
-end

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

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A chunk returned by the API.
-      class Chunk < Payload
-        # @return [Array<DeltaChoice>]
-        def choices
-          @choices ||= @data['choices'].map { |data| DeltaChoice.new(data:) }
-        end
-      end
-    end
-  end
-end

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

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A completion returned by the API.
-      class Completion < Payload
-        # @return [Array<MessageChoice>]
-        def choices
-          @choices ||= @data['choices'].map { |data| MessageChoice.new(data:) }
-        end
-      end
-    end
-  end
-end

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

@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A delta returned by the API.
-      class Delta < Part
-      end
-    end
-  end
-end

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

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A delta choice returned by the API.
-      class DeltaChoice < Choice
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} index=#{index} delta=#{delta.inspect}>"
-        end
-
-        # @return [Delta]
-        def delta
-          @delta ||= Delta.new(data: @data['delta'])
-        end
-
-        # @return [Delta]
-        def part
-          delta
-        end
-      end
-    end
-  end
-end

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

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A function returned by the API.
-      class Function < Resource
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} name=#{name.inspect} arguments=#{arguments.inspect}>"
-        end
-
-        # @return [String]
-        def name
-          @data['name']
-        end
-
-        # @return [Hash, nil]
-        def arguments
-          JSON.parse(@data['arguments']) if @data['arguments']
-        end
-      end
-    end
-  end
-end

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

@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A message returned by the API.
-      class Message < Part
-      end
-    end
-  end
-end

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

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A choice returned by the API.
-      class MessageChoice < Choice
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} index=#{index} message=#{message.inspect}>"
-        end
-
-        # @return [Message]
-        def message
-          @message ||= Message.new(data: @data['message'])
-        end
-
-        # @return [Message]
-        def part
-          message
-        end
-      end
-    end
-  end
-end

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

@@ -1,38 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # Either a delta or message.
-      class Part < Resource
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} role=#{role.inspect} content=#{content.inspect}>"
-        end
-
-        # @return [String]
-        def role
-          @data['role'] || Role::USER
-        end
-
-        # @return [String, nil]
-        def content
-          @data['content']
-        end
-
-        # @return [Array<ToolCall>]
-        def tool_call_list
-          return [] unless @data['tool_calls']
-
-          @tool_call_list ||= @data['tool_calls'].map { |tool_call_data| ToolCall.new(data: tool_call_data) }
-        end
-
-        # @param index [Integer]
-        # @return [ToolCall, nil]
-        def tool_call(index: 0)
-          tool_call_list[index]
-        end
-      end
-    end
-  end
-end

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

@@ -1,72 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A chunk or completion.
-      class Payload < Resource
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} id=#{id.inspect} choices=#{choices.inspect}>"
-        end
-
-        # @return [String]
-        def id
-          @data['id']
-        end
-
-        # @return [Time]
-        def created
-          Time.at(@data['created']) if @data['created']
-        end
-
-        # @return [Time]
-        def updated
-          Time.at(@data['updated']) if @data['updated']
-        end
-
-        # @return [String]
-        def model
-          @data['model']
-        end
-
-        # @return [Array<Choice>]
-        def choices
-          raise NotImplementedError, "#{self.class.name}#choices undefined"
-        end
-
-        # @param index [Integer]
-        # @return [DeltaChoice]
-        def choice(index: 0)
-          choices[index]
-        end
-
-        # @param index [Integer]
-        # @return [Part]
-        def part(index: 0)
-          choice(index:).part
-        end
-
-        # @return [Usage]
-        def usage
-          @usage ||= Usage.new(data: @data['usage']) if @data['usage']
-        end
-
-        # @return [String, nil]
-        def content
-          choice.content
-        end
-
-        # @return [Boolean]
-        def content?
-          choice.content?
-        end
-
-        # @return [Array<ToolCall>]
-        def tool_call_list
-          choice.tool_call_list
-        end
-      end
-    end
-  end
-end

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

@@ -1,22 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A generic data to handle response.
-      class Resource
-        attr_accessor :data
-
-        # @param data [Hash]
-        def initialize(data:)
-          @data = data
-        end
-
-        # @return [String]
-        def inspect
-          "#<#{self.class.name} data=#{@data.inspect}>"
-        end
-      end
-    end
-  end
-end

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

@@ -1,27 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    module Response
-      # A stream given when streaming.
-      class Stream
-        # @param response [HTTP::Response]
-        def initialize(response:)
-          @response = response
-          @parser = EventStreamParser::Parser.new
-        end
-
-        # @yield [OmniAI::Chat::Chunk]
-        def stream!
-          @response.body.each do |chunk|
-            @parser.feed(chunk) do |_, data|
-              next if data.eql?('[DONE]')
-
-              yield(Chunk.new(data: JSON.parse(data)))
-            end
-          end
-        end
-      end
-    end
-  end
-end

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