llm-shell 0.9.2 → 0.10.0

data/lib/llm/shell/internal/llm.rb/lib/llm/tool.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+##
+# The {LLM::Tool LLM::Tool} class represents a local tool
+# that can be called by an LLM. Under the hood, it is a wrapper
+# around {LLM::Function LLM::Function} but allows the definition
+# of a function (also known as a tool) as a class.
+# @example
+#   class System < LLM::Tool
+#     name "system"
+#     description "Runs system commands"
+#     params do |schema|
+#       schema.object(command: schema.string.required)
+#     end
+#
+#     def call(command:)
+#       {success: Kernel.system(command)}
+#     end
+#   end
+class LLM::Tool
+  require_relative "tool/param"
+  extend LLM::Tool::Param
+
+  ##
+  # Registers the tool as a function when inherited
+  # @param [Class] klass The subclass
+  # @return [void]
+  def self.inherited(klass)
+    LLM.lock(:inherited) do
+      klass.instance_eval { @__monitor ||= Monitor.new }
+      klass.function.register(klass)
+    end
+  end
+
+  ##
+  # Returns (or sets) the tool name
+  # @param [String, nil] name The tool name
+  # @return [String]
+  def self.name(name = nil)
+    lock do
+      name ? function.name(name) : function.name
+    end
+  end
+
+  ##
+  # Returns (or sets) the tool description
+  # @param [String, nil] desc The tool description
+  # @return [String]
+  def self.description(desc = nil)
+    lock do
+      desc ? function.description(desc) : function.description
+    end
+  end
+
+  ##
+  # Returns (or sets) tool parameters
+  # @yieldparam [LLM::Schema] schema The schema object to define parameters
+  # @return [LLM::Schema]
+  def self.params(&)
+    lock do
+      function.tap { _1.params(&) }
+    end
+  end
+
+  ##
+  # @api private
+  def self.function
+    lock do
+      @function ||= LLM::Function.new(self)
+    end
+  end
+
+  ##
+  # @api private
+  def self.lock(&)
+    @__monitor.synchronize(&)
+  end
+end

data/lib/llm/shell/internal/llm.rb/lib/llm/utils.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+##
+# @private
+module LLM::Utils
+  def camelcase(key)
+    key.to_s
+      .split("_")
+      .map.with_index { (_2 > 0) ? _1.capitalize : _1 }
+      .join
+  end
+
+  def snakecase(key)
+    key
+      .split(/([A-Z])/)
+      .map { (_1.size == 1) ? "_#{_1.downcase}" : _1 }
+      .join
+  end
+end

data/lib/llm/shell/internal/llm.rb/lib/llm/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module LLM
+  VERSION = "1.0.1"
+end

data/lib/llm/shell/internal/llm.rb/lib/llm.rb

@@ -0,0 +1,121 @@
+# frozen_string_literal: true
+
+module LLM
+  require "stringio"
+  require_relative "llm/schema"
+  require_relative "llm/object"
+  require_relative "llm/version"
+  require_relative "llm/utils"
+  require_relative "llm/error"
+  require_relative "llm/message"
+  require_relative "llm/response"
+  require_relative "llm/mime"
+  require_relative "llm/multipart"
+  require_relative "llm/file"
+  require_relative "llm/provider"
+  require_relative "llm/bot"
+  require_relative "llm/buffer"
+  require_relative "llm/function"
+  require_relative "llm/eventstream"
+  require_relative "llm/eventhandler"
+  require_relative "llm/tool"
+  require_relative "llm/server_tool"
+
+  ##
+  # Thread-safe monitors for different contexts
+  @monitors = {require: Monitor.new, clients: Monitor.new, inherited: Monitor.new}
+
+  module_function
+
+  ##
+  # @param (see LLM::Provider#initialize)
+  # @return (see LLM::Anthropic#initialize)
+  def anthropic(**)
+    lock(:require) { require_relative "llm/providers/anthropic" unless defined?(LLM::Anthropic) }
+    LLM::Anthropic.new(**)
+  end
+
+  ##
+  # @param (see LLM::Provider#initialize)
+  # @return (see LLM::Gemini#initialize)
+  def gemini(**)
+    lock(:require) { require_relative "llm/providers/gemini" unless defined?(LLM::Gemini) }
+    LLM::Gemini.new(**)
+  end
+
+  ##
+  # @param key (see LLM::Provider#initialize)
+  # @return (see LLM::Ollama#initialize)
+  def ollama(key: nil, **)
+    lock(:require) { require_relative "llm/providers/ollama" unless defined?(LLM::Ollama) }
+    LLM::Ollama.new(key:, **)
+  end
+
+  ##
+  # @param key (see LLM::Provider#initialize)
+  # @return (see LLM::LlamaCpp#initialize)
+  def llamacpp(key: nil, **)
+    lock(:require) { require_relative "llm/providers/llamacpp" unless defined?(LLM::LlamaCpp) }
+    LLM::LlamaCpp.new(key:, **)
+  end
+
+  ##
+  # @param key (see LLM::Provider#initialize)
+  # @return (see LLM::DeepSeek#initialize)
+  def deepseek(**)
+    lock(:require) { require_relative "llm/providers/deepseek" unless defined?(LLM::DeepSeek) }
+    LLM::DeepSeek.new(**)
+  end
+
+  ##
+  # @param key (see LLM::Provider#initialize)
+  # @return (see LLM::OpenAI#initialize)
+  def openai(**)
+    lock(:require) { require_relative "llm/providers/openai" unless defined?(LLM::OpenAI) }
+    LLM::OpenAI.new(**)
+  end
+
+  ##
+  # @param key (see LLM::XAI#initialize)
+  # @param host (see LLM::XAI#initialize)
+  # @return (see LLM::XAI#initialize)
+  def xai(**)
+    lock(:require) { require_relative "llm/providers/xai" unless defined?(LLM::XAI) }
+    LLM::XAI.new(**)
+  end
+
+  ##
+  # @param key (see LLM::ZAI#initialize)
+  # @param host (see LLM::ZAI#initialize)
+  # @return (see LLM::ZAI#initialize)
+  def zai(**)
+    lock(:require) { require_relative "llm/providers/zai" unless defined?(LLM::ZAI) }
+    LLM::ZAI.new(**)
+  end
+
+  ##
+  # Define a function
+  # @example
+  #   LLM.function(:system) do |fn|
+  #     fn.description "Run system command"
+  #     fn.params do |schema|
+  #       schema.object(command: schema.string.required)
+  #     end
+  #     fn.define do |command:|
+  #       system(command)
+  #     end
+  #   end
+  # @param [Symbol] key The function name / key
+  # @param [Proc] b The block to define the function
+  # @return [LLM::Function] The function object
+  def function(key, &b)
+    LLM::Function.new(key, &b)
+  end
+
+  ##
+  # Provides a thread-safe lock
+  # @param [Symbol] name The name of the lock
+  # @param [Proc] & The block to execute within the lock
+  # @return [void]
+  def lock(name, &) = @monitors[name].synchronize(&)
+end

data/lib/llm/shell/internal/optparse/lib/optionparser.rb

@@ -0,0 +1,2 @@
+# frozen_string_literal: false
+require_relative 'optparse'

data/lib/llm/shell/internal/optparse/lib/optparse/ac.rb

@@ -0,0 +1,70 @@
+# frozen_string_literal: false
+require_relative '../optparse'
+
+#
+# autoconf-like options.
+#
+class OptionParser::AC < OptionParser
+  # :stopdoc:
+  private
+
+  def _check_ac_args(name, block)
+    unless /\A\w[-\w]*\z/ =~ name
+      raise ArgumentError, name
+    end
+    unless block
+      raise ArgumentError, "no block given", ParseError.filter_backtrace(caller)
+    end
+  end
+
+  ARG_CONV = proc {|val| val.nil? ? true : val}
+  private_constant :ARG_CONV
+
+  def _ac_arg_enable(prefix, name, help_string, block)
+    _check_ac_args(name, block)
+
+    sdesc = []
+    ldesc = ["--#{prefix}-#{name}"]
+    desc = [help_string]
+    q = name.downcase
+    ac_block = proc {|val| block.call(ARG_CONV.call(val))}
+    enable = Switch::PlacedArgument.new(nil, ARG_CONV, sdesc, ldesc, nil, desc, ac_block)
+    disable = Switch::NoArgument.new(nil, proc {false}, sdesc, ldesc, nil, desc, ac_block)
+    top.append(enable, [], ["enable-" + q], disable, ['disable-' + q])
+    enable
+  end
+
+  # :startdoc:
+
+  public
+
+  # Define <tt>--enable</tt> / <tt>--disable</tt> style option
+  #
+  # Appears as <tt>--enable-<i>name</i></tt> in help message.
+  def ac_arg_enable(name, help_string, &block)
+    _ac_arg_enable("enable", name, help_string, block)
+  end
+
+  # Define <tt>--enable</tt> / <tt>--disable</tt> style option
+  #
+  # Appears as <tt>--disable-<i>name</i></tt> in help message.
+  def ac_arg_disable(name, help_string, &block)
+    _ac_arg_enable("disable", name, help_string, block)
+  end
+
+  # Define <tt>--with</tt> / <tt>--without</tt> style option
+  #
+  # Appears as <tt>--with-<i>name</i></tt> in help message.
+  def ac_arg_with(name, help_string, &block)
+    _check_ac_args(name, block)
+
+    sdesc = []
+    ldesc = ["--with-#{name}"]
+    desc = [help_string]
+    q = name.downcase
+    with = Switch::PlacedArgument.new(*search(:atype, String), sdesc, ldesc, nil, desc, block)
+    without = Switch::NoArgument.new(nil, proc {}, sdesc, ldesc, nil, desc, block)
+    top.append(with, [], ["with-" + q], without, ['without-' + q])
+    with
+  end
+end

data/lib/llm/shell/internal/optparse/lib/optparse/date.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: false
+require_relative '../optparse'
+require 'date'
+
+OptionParser.accept(DateTime) do |s,|
+  begin
+    DateTime.parse(s) if s
+  rescue ArgumentError
+    raise OptionParser::InvalidArgument, s
+  end
+end
+OptionParser.accept(Date) do |s,|
+  begin
+    Date.parse(s) if s
+  rescue ArgumentError
+    raise OptionParser::InvalidArgument, s
+  end
+end

data/lib/llm/shell/internal/optparse/lib/optparse/kwargs.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+require_relative '../optparse'
+
+class OptionParser
+  # :call-seq:
+  #   define_by_keywords(options, method, **params)
+  #
+  # :include: ../../doc/optparse/creates_option.rdoc
+  #
+  # Defines options which set in to _options_ for keyword parameters
+  # of _method_.
+  #
+  # Parameters for each keywords are given as elements of _params_.
+  #
+  def define_by_keywords(options, method, **params)
+    method.parameters.each do |type, name|
+      case type
+      when :key, :keyreq
+        op, cl = *(type == :key ? %w"[ ]" : ["", ""])
+        define("--#{name}=#{op}#{name.upcase}#{cl}", *params[name]) do |o|
+          options[name] = o
+        end
+      end
+    end
+    options
+  end
+end

data/lib/llm/shell/internal/optparse/lib/optparse/shellwords.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: false
+# -*- ruby -*-
+
+require 'shellwords'
+require_relative '../optparse'
+
+OptionParser.accept(Shellwords) {|s,| Shellwords.shellwords(s)}

data/lib/llm/shell/internal/optparse/lib/optparse/time.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: false
+require_relative '../optparse'
+require 'time'
+
+OptionParser.accept(Time) do |s,|
+  begin
+    (Time.httpdate(s) rescue Time.parse(s)) if s
+  rescue
+    raise OptionParser::InvalidArgument, s
+  end
+end

data/lib/llm/shell/internal/optparse/lib/optparse/uri.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: false
+# -*- ruby -*-
+
+require_relative '../optparse'
+require 'uri'
+
+OptionParser.accept(URI) {|s,| URI.parse(s) if s}

data/lib/llm/shell/internal/optparse/lib/optparse/version.rb

@@ -0,0 +1,80 @@
+# frozen_string_literal: false
+# OptionParser internal utility
+
+class << OptionParser
+  #
+  # Shows version string in packages if Version is defined.
+  #
+  # +pkgs+:: package list
+  #
+  def show_version(*pkgs)
+    progname = ARGV.options.program_name
+    result = false
+    show = proc do |klass, cname, version|
+      str = "#{progname}"
+      unless klass == ::Object and cname == :VERSION
+        version = version.join(".") if Array === version
+        str << ": #{klass}" unless klass == Object
+        str << " version #{version}"
+      end
+      [:Release, :RELEASE].find do |rel|
+        if klass.const_defined?(rel)
+          str << " (#{klass.const_get(rel)})"
+        end
+      end
+      puts str
+      result = true
+    end
+    if pkgs.size == 1 and pkgs[0] == "all"
+      self.search_const(::Object, /\AV(?:ERSION|ersion)\z/) do |klass, cname, version|
+        unless cname[1] == ?e and klass.const_defined?(:Version)
+          show.call(klass, cname.intern, version)
+        end
+      end
+    else
+      pkgs.each do |pkg|
+        begin
+          pkg = pkg.split(/::|\//).inject(::Object) {|m, c| m.const_get(c)}
+          v = case
+              when pkg.const_defined?(:Version)
+                pkg.const_get(n = :Version)
+              when pkg.const_defined?(:VERSION)
+                pkg.const_get(n = :VERSION)
+              else
+                n = nil
+                "unknown"
+              end
+          show.call(pkg, n, v)
+        rescue NameError
+        end
+      end
+    end
+    result
+  end
+
+  # :stopdoc:
+
+  def each_const(path, base = ::Object)
+    path.split(/::|\//).inject(base) do |klass, name|
+      raise NameError, path unless Module === klass
+      klass.constants.grep(/#{name}/i) do |c|
+        klass.const_defined?(c) or next
+        klass.const_get(c)
+      end
+    end
+  end
+
+  def search_const(klass, name)
+    klasses = [klass]
+    while klass = klasses.shift
+      klass.constants.each do |cname|
+        klass.const_defined?(cname) or next
+        const = klass.const_get(cname)
+        yield klass, cname, const if name === cname
+        klasses << const if Module === const and const != ::Object
+      end
+    end
+  end
+
+  # :startdoc:
+end