llm.rb 0.2.0 → 0.3.0

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm.rb
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Antar Azri
@@ -9,50 +9,8 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-04-01 00:00:00.000000000 Z
+date: 2025-04-25 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: net-http
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.6.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.6.0
-- !ruby/object:Gem::Dependency
-  name: json
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: yaml
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: webmock
   requirement: !ruby/object:Gem::Requirement
@@ -189,41 +147,57 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- LICENSE.txt
 - README.md
 - lib/llm.rb
-- lib/llm/conversation.rb
+- lib/llm/buffer.rb
+- lib/llm/chat.rb
 - lib/llm/core_ext/ostruct.rb
 - lib/llm/error.rb
 - lib/llm/file.rb
 - lib/llm/http_client.rb
-- lib/llm/lazy_conversation.rb
 - lib/llm/message.rb
-- lib/llm/message_queue.rb
+- lib/llm/mime.rb
 - lib/llm/model.rb
+- lib/llm/multipart.rb
 - lib/llm/provider.rb
 - lib/llm/providers/anthropic.rb
 - lib/llm/providers/anthropic/error_handler.rb
 - lib/llm/providers/anthropic/format.rb
 - lib/llm/providers/anthropic/response_parser.rb
 - lib/llm/providers/gemini.rb
+- lib/llm/providers/gemini/audio.rb
 - lib/llm/providers/gemini/error_handler.rb
+- lib/llm/providers/gemini/files.rb
 - lib/llm/providers/gemini/format.rb
+- lib/llm/providers/gemini/images.rb
 - lib/llm/providers/gemini/response_parser.rb
 - lib/llm/providers/ollama.rb
 - lib/llm/providers/ollama/error_handler.rb
 - lib/llm/providers/ollama/format.rb
 - lib/llm/providers/ollama/response_parser.rb
 - lib/llm/providers/openai.rb
+- lib/llm/providers/openai/audio.rb
 - lib/llm/providers/openai/error_handler.rb
+- lib/llm/providers/openai/files.rb
 - lib/llm/providers/openai/format.rb
+- lib/llm/providers/openai/images.rb
 - lib/llm/providers/openai/response_parser.rb
+- lib/llm/providers/openai/responses.rb
 - lib/llm/providers/voyageai.rb
 - lib/llm/providers/voyageai/error_handler.rb
 - lib/llm/providers/voyageai/response_parser.rb
 - lib/llm/response.rb
+- lib/llm/response/audio.rb
+- lib/llm/response/audio_transcription.rb
+- lib/llm/response/audio_translation.rb
 - lib/llm/response/completion.rb
+- lib/llm/response/download_file.rb
 - lib/llm/response/embedding.rb
+- lib/llm/response/file.rb
+- lib/llm/response/filelist.rb
+- lib/llm/response/image.rb
+- lib/llm/response/output.rb
+- lib/llm/utils.rb
 - lib/llm/version.rb
 - llm.gemspec
 - share/llm/models/anthropic.yml
@@ -233,22 +207,28 @@ files:
 - spec/anthropic/completion_spec.rb
 - spec/anthropic/embedding_spec.rb
 - spec/gemini/completion_spec.rb
+- spec/gemini/conversation_spec.rb
 - spec/gemini/embedding_spec.rb
+- spec/gemini/files_spec.rb
+- spec/gemini/images_spec.rb
 - spec/llm/conversation_spec.rb
-- spec/llm/lazy_conversation_spec.rb
 - spec/ollama/completion_spec.rb
+- spec/ollama/conversation_spec.rb
 - spec/ollama/embedding_spec.rb
+- spec/openai/audio_spec.rb
 - spec/openai/completion_spec.rb
 - spec/openai/embedding_spec.rb
+- spec/openai/files_spec.rb
+- spec/openai/images_spec.rb
+- spec/openai/responses_spec.rb
 - spec/readme_spec.rb
 - spec/setup.rb
 homepage: https://github.com/llmrb/llm
 licenses:
-- MIT
+- 0BSDL
 metadata:
   homepage_uri: https://github.com/llmrb/llm
   source_code_uri: https://github.com/llmrb/llm
-  changelog_uri: https://github.com/llmrb/llm/blob/main/CHANGELOG.md
 post_install_message:
 rdoc_options: []
 require_paths:

data/LICENSE.txt

@@ -1,21 +0,0 @@
-The MIT License (MIT)
-
-Copyright (c) 2024 Antar Azri
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in
-all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.

data/lib/llm/conversation.rb

@@ -1,50 +0,0 @@
-# frozen_string_literal: true
-
-module LLM
-  ##
-  # {LLM::Conversation LLM::Conversation} provides a conversation
-  # object that maintains a thread of messages that act as the
-  # context of the conversation.
-  #
-  # @example
-  #   llm = LLM.openai(key)
-  #   bot = llm.chat("What is the capital of France?")
-  #   bot.chat("What should we eat in Paris?")
-  #   bot.chat("What is the weather like in Paris?")
-  #   p bot.messages.map { [_1.role, _1.content] }
-  class Conversation
-    ##
-    # @return [Array<LLM::Message>]
-    attr_reader :messages
-
-    ##
-    # @param [LLM::Provider] provider
-    #  A provider
-    def initialize(provider, params = {})
-      @provider = provider
-      @params = params
-      @messages = []
-    end
-
-    ##
-    # @param prompt (see LLM::Provider#prompt)
-    # @return [LLM::Conversation]
-    def chat(prompt, role = :user, **params)
-      tap do
-        completion = @provider.complete(prompt, role, **@params.merge(params.merge(messages:)))
-        @messages.concat [Message.new(role, prompt), completion.choices[0]]
-      end
-    end
-
-    ##
-    # @param [#to_s] role
-    #  The role of the last message.
-    #  Defaults to the LLM's assistant role (eg "assistant" or "model")
-    # @return [LLM::Message]
-    #  The last message for the given role
-    def last_message(role: @provider.assistant_role)
-      messages.reverse_each.find { _1.role == role.to_s }
-    end
-    alias_method :recent_message, :last_message
-  end
-end

data/lib/llm/lazy_conversation.rb

@@ -1,51 +0,0 @@
-# frozen_string_literal: true
-
-module LLM
-  require_relative "message_queue"
-
-  ##
-  # {LLM::LazyConversation LLM::LazyConversation} provides a
-  # conversation object that allows input prompts to be queued
-  # and only sent to the LLM when a response is needed.
-  #
-  # @example
-  #   llm = LLM.openai(key)
-  #   bot = llm.chat("Be a helpful weather assistant", :system)
-  #   bot.chat("What's the weather like in Rio?")
-  #   bot.chat("What's the weather like in Algiers?")
-  #   bot.messages.each do |message|
-  #     # A single request is made at this point
-  #   end
-  class LazyConversation
-    ##
-    # @return [LLM::MessageQueue]
-    attr_reader :messages
-
-    ##
-    # @param [LLM::Provider] provider
-    #  A provider
-    def initialize(provider, params = {})
-      @provider = provider
-      @params = params
-      @messages = LLM::MessageQueue.new(provider)
-    end
-
-    ##
-    # @param prompt (see LLM::Provider#prompt)
-    # @return [LLM::Conversation]
-    def chat(prompt, role = :user, **params)
-      tap { @messages << [prompt, role, @params.merge(params)] }
-    end
-
-    ##
-    # @param [#to_s] role
-    #  The role of the last message.
-    #  Defaults to the LLM's assistant role (eg "assistant" or "model")
-    # @return [LLM::Message]
-    #  The last message for the given role
-    def last_message(role: @provider.assistant_role)
-      messages.reverse_each.find { _1.role == role.to_s }
-    end
-    alias_method :recent_message, :last_message
-  end
-end

data/lib/llm/message_queue.rb

@@ -1,47 +0,0 @@
-# frozen_string_literal: true
-
-module LLM
-  ##
-  # {LLM::MessageQueue LLM::MessageQueue} provides an Enumerable
-  # object that yields each message in a conversation on-demand,
-  # and only sends a request to the LLM when a response is needed.
-  class MessageQueue
-    include Enumerable
-
-    ##
-    # @param [LLM::Provider] provider
-    # @return [LLM::MessageQueue]
-    def initialize(provider)
-      @provider = provider
-      @messages = []
-    end
-
-    ##
-    # @yield [LLM::Message]
-    #  Yields each message in the conversation thread
-    # @raise (see LLM::Provider#complete)
-    # @return [void]
-    def each
-      @messages = complete! unless @messages.grep(LLM::Message).size == @messages.size
-      @messages.each { yield(_1) }
-    end
-
-    ##
-    # @param message [Object]
-    #  A message to add to the conversation thread
-    # @return [void]
-    def <<(message)
-      @messages << message
-    end
-    alias_method :push, :<<
-
-    private
-
-    def complete!
-      prompt, role, params = @messages[-1]
-      rest = @messages[0..-2].map { (Array === _1) ? LLM::Message.new(_1[1], _1[0]) : _1 }
-      comp = @provider.complete(prompt, role, **params.merge(messages: rest)).choices.last
-      [*rest, LLM::Message.new(role, prompt), comp]
-    end
-  end
-end

data/spec/llm/lazy_conversation_spec.rb

@@ -1,92 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe LLM::LazyConversation do
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-  let(:prompt) { "Keep your answers short and concise, and provide three answers to the three questions" }
-
-  context "with gemini",
-          vcr: {cassette_name: "gemini/lazy_conversation/successful_response"} do
-    let(:provider) { LLM.gemini(token) }
-    let(:conversation) { described_class.new(provider) }
-
-    context "when given a thread of messages" do
-      subject(:message) { conversation.messages.to_a[-1] }
-
-      before do
-        conversation.chat prompt
-        conversation.chat "What is 3+2 ?"
-        conversation.chat "What is 5+5 ?"
-        conversation.chat "What is 5+7 ?"
-      end
-
-      it "maintains a conversation" do
-        is_expected.to have_attributes(
-          role: "model",
-          content: "5\n10\n12\n"
-        )
-      end
-    end
-  end
-
-  context "with openai"  do
-    let(:provider) { LLM.openai(token) }
-    let(:conversation) { described_class.new(provider) }
-
-    context "when given a thread of messages",
-            vcr: {cassette_name: "openai/lazy_conversation/successful_response"} do
-      subject(:message) { conversation.recent_message }
-
-      before do
-        conversation.chat prompt, :system
-        conversation.chat "What is 3+2 ?"
-        conversation.chat "What is 5+5 ?"
-        conversation.chat "What is 5+7 ?"
-      end
-
-      it "maintains a conversation" do
-        is_expected.to have_attributes(
-          role: "assistant",
-          content: "1. 5  \n2. 10  \n3. 12  "
-        )
-      end
-    end
-
-    context "when given a specific model",
-            vcr: {cassette_name: "openai/lazy_conversation/successful_response_o3_mini"} do
-      let(:conversation) { described_class.new(provider, model: provider.models["o3-mini"]) }
-
-      it "maintains the model throughout a conversation" do
-        conversation.chat(prompt, :system)
-        expect(conversation.recent_message.extra[:completion].model).to eq("o3-mini-2025-01-31")
-        conversation.chat("What is 5+5?")
-        expect(conversation.recent_message.extra[:completion].model).to eq("o3-mini-2025-01-31")
-      end
-    end
-  end
-
-  context "with ollama",
-          vcr: {cassette_name: "ollama/lazy_conversation/successful_response"} do
-    let(:provider) { LLM.ollama(nil, host: "eel.home.network") }
-    let(:conversation) { described_class.new(provider) }
-
-    context "when given a thread of messages" do
-      subject(:message) { conversation.recent_message }
-
-      before do
-        conversation.chat prompt, :system
-        conversation.chat "What is 3+2 ?"
-        conversation.chat "What is 5+5 ?"
-        conversation.chat "What is 5+7 ?"
-      end
-
-      it "maintains a conversation" do
-        is_expected.to have_attributes(
-          role: "assistant",
-          content: "Here are the calculations:\n\n1. 3 + 2 = 5\n2. 5 + 5 = 10\n3. 5 + 7 = 12"
-        )
-      end
-    end
-  end
-end