Contents

# extensions/omniai-ollama.rb
# frozen_string_literal: true

require 'omniai'
require 'omniai/openai'

module OmniAI

  # Create an alias for OmniAI::OpenAI module
  module Ollama
    extend OmniAI::OpenAI

    # Alias classes from OmniAI::OpenAI
    class Client < OmniAI::OpenAI::Client
      def initialize(**options)
        options[:host] = 'http://localhost:11434' unless options.has_key?(:host)
        super(**options)
      end
    end    

    Config = OmniAI::OpenAI::Config

    # Alias the Thread class and its nested classes
    Thread      = OmniAI::OpenAI::Thread
    Annotation  = OmniAI::OpenAI::Thread::Annotation
    Attachment  = OmniAI::OpenAI::Thread::Attachment
    Message     = OmniAI::OpenAI::Thread::Message
    Run         = OmniAI::OpenAI::Thread::Run
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
ai_client-0.2.2	lib/extensions/omniai-ollama.rb
ai_client-0.2.1	lib/extensions/omniai-ollama.rb
ai_client-0.2.0	lib/extensions/omniai-ollama.rb
ai_client-0.1.0	lib/extensions/omniai-ollama.rb