llm-shell 0.10.0 → 0.11.0

data/lib/llm/shell/commands/dir_import.rb

@@ -1,33 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::DirImport < Command
-    name "dir-import"
-    description "Import the contents of a directory"
-
-    ##
-    # Completes a path with a wildcard.
-    # @param path [String]
-    #  The path to complete.
-    # @return [Array<String>]
-    #  Returns the completed path(s)
-    def self.complete(path)
-      Dir["#{path}*"]
-    end
-
-    ##
-    # Recursively imports all files in a directory.
-    # @return [void]
-    def call(dir)
-      Dir.entries(dir).each do |file|
-        if file == "." || file == ".."
-          next
-        elsif File.directory? File.join(dir, file)
-          call File.join(dir, file)
-        else
-          import File.join(dir, file)
-        end
-      end
-    end
-  end
-end

data/lib/llm/shell/commands/disable_tool.rb

@@ -1,33 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::DisableTool < Command
-    Error = Class.new(RuntimeError)
-
-    name "disable-tool"
-    description "Disable a tool"
-
-    ##
-    # Completes a tool name.
-    # @param name [String]
-    #  The tool name to complete.
-    # @return [Array<String>]
-    #  Returns the completed tool name(s)
-    def self.complete(name)
-      LLM::Shell.tools.map(&:name).select { _1.start_with?(name) }
-    end
-
-    ##
-    # Disables the given tool.
-    # @return [void]
-    def call(name)
-      tool = LLM::Shell.tools.find { _1.name == name }
-      if tool
-        tool.disable!
-        io.rewind.print("tool disabled").end
-      else
-        raise Error, "unknown tool: #{name}"
-      end
-    end
-  end
-end

data/lib/llm/shell/commands/enable_tool.rb

@@ -1,33 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::EnableTool < Command
-    Error = Class.new(RuntimeError)
-
-    name "enable-tool"
-    description "Enable a tool"
-
-    ##
-    # Completes a tool name.
-    # @param name [String]
-    #  The tool name to complete.
-    # @return [Array<String>]
-    #  Returns the completed tool name(s)
-    def self.complete(name)
-      LLM::Shell.tools.map(&:name).select { _1.start_with?(name) }
-    end
-
-    ##
-    # Enables the given tool.
-    # @return [void]
-    def call(name)
-      tool = LLM::Shell.tools.find { _1.name == name }
-      if tool
-        tool.enable!
-        io.rewind.print("tool enabled").end
-      else
-        raise Error, "unknown tool: #{name}"
-      end
-    end
-  end
-end

data/lib/llm/shell/commands/file_import.rb

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::FileImport < Command
-    name "file-import"
-    description "Import a file"
-
-    ##
-    # Completes a path with a wildcard.
-    # @param path [String]
-    #  The path to complete.
-    # @return [Array<String>]
-    #  Returns the completed path(s)
-    def self.complete(path)
-      Dir["#{path}*"]
-    end
-
-    ##
-    # Imports one or more globbed files.
-    # @return [void]
-    def call(*files)
-      Dir[*files].each { import(_1) }
-    end
-  end
-end

data/lib/llm/shell/commands/help.rb

@@ -1,53 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::Help < Command
-    name "help"
-    description "Show help"
-
-    ##
-    # Prints help
-    # @return [void]
-    def call
-      pager do |io|
-        io.print(Paint["Help", :bold, :underline], "\n\n")
-        render_commands(io)
-        io.print("\n")
-        render_functions(io)
-      end
-    end
-
-    private
-
-    def render_commands(io)
-      io.print(Paint["Commands", :bold, :underline], "\n\n")
-      render_group(io, "Builtin", commands.select(&:builtin?), :cyan)
-      io.print("\n")
-      render_group(io, "User", commands.reject(&:builtin?), :cyan)
-    end
-
-    def render_functions(io)
-      io.print(Paint["Functions", :bold, :underline], "\n\n")
-      render_group(io, "Builtin (Enabled)", functions.select { |f|  f.builtin? && f.enabled? }, :blue)
-      render_group(io, "Builtin (Disabled)", functions.select { |f|  f.builtin? && !f.enabled? }, :blue)
-      io.print("\n")
-      render_group(io, "User (Enabled)", functions.select { |f| !f.builtin? && f.enabled? }, :blue)
-      render_group(io, "User (Disabled)", functions.select { |f| !f.builtin? && !f.enabled? }, :blue)
-    end
-
-    def render_group(io, title, items, color)
-      io.print(Paint[title, :bold], "\n")
-      if items.empty?
-        io.print(Paint["  None available", :yellow], "\n")
-      else
-        items.each do |item|
-          status = item.enabled? ? Paint["enabled", :green] : Paint["disabled", :red]
-          io.print("  ", Paint[item.name, color, :bold], " - ", item.description || "No description", " (", status, ")\n")
-        end
-      end
-    end
-
-    def commands = LLM::Shell.commands.sort_by(&:name)
-    def functions = LLM::Shell.tools.sort_by(&:name)
-  end
-end

data/lib/llm/shell/commands/show_chat.rb

@@ -1,31 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::ShowChat < Command
-    name "show-chat"
-    description "Show the chat"
-
-    ##
-    # Emits the full chat history to standard output
-    # @return [void]
-    def call
-      clear_screen
-      emit
-    end
-
-    private
-
-    def emit
-      pager do |io|
-        messages.each.with_index do |message, index|
-          io << render(message) << "\n"
-        end
-      end
-    end
-
-    def console = IO.console
-    def clear_screen = console.clear_screen
-    def messages = bot.messages
-    def render(message) = LLM::Shell::Renderer.new(message).render
-  end
-end

data/lib/llm/shell/commands/show_version.rb

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::ShowVersion < Command
-    name "show-version"
-    description "Show the version"
-
-    ##
-    # Shows the current llm-shell version
-    # @return [void]
-    def call
-      pager do |io|
-        io.write("llm-shell version: #{LLM::Shell::VERSION}\n")
-      end
-    end
-  end
-end

data/lib/llm/shell/commands/system_prompt.rb

@@ -1,21 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Command::SystemPrompt < Command
-    name "system-prompt"
-    description "Show the system prompt"
-
-    ##
-    # Emits the system prompt to standard output
-    # @return [void]
-    def call
-      pager do |io|
-        io.write render(bot.messages[0])
-      end
-    end
-
-    private
-
-    def render(message) = LLM::Shell::Renderer.new(message).render
-  end
-end

data/lib/llm/shell/completion.rb

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Completion
-    ##
-    # Returns a proc suitable for Reline completion.
-    # @return [Proc]
-    def self.to_proc
-      new.to_proc
-    end
-
-    ##
-    # @return [LLM::Shell::Completion]
-    def initialize
-      @commands = LLM::Shell.commands.map { [_1.name, _1] }.to_h
-    end
-
-    ##
-    # Returns a proc suitable for Reline completion.
-    # @return [Proc]
-    def to_proc
-      method(:complete).to_proc
-    end
-
-    private
-
-    def complete(input)
-      words = Reline.line_buffer.split(" ")
-      command = words[0]
-      if commands[command]
-        object = commands[command]
-        object.respond_to?(:complete) ? object.complete(input) : []
-      else
-        commands.keys.grep(/\A#{command}/)
-      end
-    end
-
-    attr_reader :commands
-  end
-end

data/lib/llm/shell/config.rb

@@ -1,34 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Config
-    ##
-    # @param [String] provider
-    # @return [LLM::Shell::Config]
-    def initialize(provider)
-      @provider = provider
-    end
-
-    ##
-    # @return [Hash]
-    def merge(other)
-      to_h.merge(other)
-    end
-
-    ##
-    # @return [Hash]
-    def to_h
-      toml[@provider] || {}
-    end
-
-    private
-
-    def toml
-      @toml ||= Tomlrb.load_file(path)
-    end
-
-    def path
-      File.join LLM::Shell.home_config, "llm-shell.toml"
-    end
-  end
-end

data/lib/llm/shell/default.rb

@@ -1,23 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Default
-    def initialize(provider)
-      @provider = provider
-    end
-
-    def prompt
-      File.join(SHAREDIR, "prompts", "default.txt")
-    end
-
-    def role
-      case @provider
-      when "openai", "ollama" then :system
-      else :user
-      end
-    end
-
-    SHAREDIR = File.join(__dir__, "..", "..", "..", "share", "llm-shell")
-    private_constant :SHAREDIR
-  end
-end

data/lib/llm/shell/formatter.rb

@@ -1,41 +0,0 @@
-# frozen_string_literal: true
-
-class LLM::Shell
-  class Formatter
-    FormatError = Class.new(RuntimeError)
-
-    def initialize(messages)
-      @messages = messages.reject(&:tool_call?)
-    end
-
-    def format!(role)
-      case role
-      when :user then format_user(messages)
-      when :assistant then format_assistant(messages)
-      else raise FormatError.new("#{role} is not known")
-      end
-    end
-
-    private
-
-    def format_user(messages)
-      messages.filter_map do |message|
-        next unless message.user?
-        next unless String === message.content
-        next unless message.content !~ LLM::Shell.pattern
-        render(message.tap(&:read!))
-      end.join("\n")
-    end
-
-    def format_assistant(messages)
-      messages.filter_map do |message|
-        next unless message.assistant?
-        next unless String === message.content
-        render(message.tap(&:read!))
-      end.join("\n")
-    end
-
-    attr_reader :messages
-    def render(message) = Renderer.new(message).render
-  end
-end

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

@@ -1,81 +0,0 @@
-module CodeRay
-  
-  # = Duo
-  #
-  # A Duo is a convenient way to use CodeRay. You just create a Duo,
-  # giving it a lang (language of the input code) and a format (desired
-  # output format), and call Duo#highlight with the code.
-  # 
-  # Duo makes it easy to re-use both scanner and encoder for a repetitive
-  # task. It also provides a very easy interface syntax:
-  # 
-  #   require 'coderay'
-  #   CodeRay::Duo[:python, :div].highlight 'import this'
-  # 
-  # Until you want to do uncommon things with CodeRay, I recommend to use
-  # this method, since it takes care of everything.
-  class Duo
-
-    attr_accessor :lang, :format, :options
-    
-    # Create a new Duo, holding a lang and a format to highlight code.
-    # 
-    # simple:
-    #   CodeRay::Duo[:ruby, :html].highlight 'bla 42'
-    # 
-    # with options:
-    #   CodeRay::Duo[:ruby, :html, :hint => :debug].highlight '????::??'
-    # 
-    # alternative syntax without options:
-    #   CodeRay::Duo[:ruby => :statistic].encode 'class << self; end'
-    # 
-    # alternative syntax with options:
-    #   CodeRay::Duo[{ :ruby => :statistic }, :do => :something].encode 'abc'
-    # 
-    # The options are forwarded to scanner and encoder
-    # (see CodeRay.get_scanner_options).
-    def initialize lang = nil, format = nil, options = {}
-      if format.nil? && lang.is_a?(Hash) && lang.size == 1
-        @lang = lang.keys.first
-        @format = lang[@lang]
-      else
-        @lang = lang
-        @format = format
-      end
-      @options = options
-    end
-    
-    class << self
-      # To allow calls like Duo[:ruby, :html].highlight.
-      alias [] new
-    end
-    
-    # The scanner of the duo. Only created once.
-    def scanner
-      @scanner ||= CodeRay.scanner @lang, CodeRay.get_scanner_options(@options)
-    end
-    
-    # The encoder of the duo. Only created once.
-    def encoder
-      @encoder ||= CodeRay.encoder @format, @options
-    end
-    
-    # Tokenize and highlight the code using +scanner+ and +encoder+.
-    def encode code, options = {}
-      options = @options.merge options
-      encoder.encode(code, @lang, options)
-    end
-    alias highlight encode
-    
-    # Allows to use Duo like a proc object:
-    # 
-    #  CodeRay::Duo[:python => :yaml].call(code)
-    # 
-    # or, in Ruby 1.9 and later:
-    # 
-    #  CodeRay::Duo[:python => :yaml].(code)
-    alias call encode
-    
-  end
-  
-end

data/lib/llm/shell/internal/coderay/lib/coderay/encoders/_map.rb

@@ -1,17 +0,0 @@
-module CodeRay
-module Encoders
-  
-  map \
-    :loc             => :lines_of_code,
-    :plain           => :text,
-    :plaintext       => :text,
-    :remove_comments => :comment_filter,
-    :stats           => :statistic,
-    :term            => :terminal,
-    :tty             => :terminal,
-    :yml             => :yaml
-  
-  # No default because Tokens#nonsense should raise NoMethodError.
-  
-end
-end

data/lib/llm/shell/internal/coderay/lib/coderay/encoders/comment_filter.rb

@@ -1,25 +0,0 @@
-module CodeRay
-module Encoders
-  
-  load :token_kind_filter
-  
-  # A simple Filter that removes all tokens of the :comment kind.
-  # 
-  # Alias: +remove_comments+
-  # 
-  # Usage:
-  #  CodeRay.scan('print # foo', :ruby).comment_filter.text
-  #  #-> "print "
-  # 
-  # See also: TokenKindFilter, LinesOfCode
-  class CommentFilter < TokenKindFilter
-    
-    register_for :comment_filter
-    
-    DEFAULT_OPTIONS = superclass::DEFAULT_OPTIONS.merge \
-      :exclude => [:comment, :docstring]
-    
-  end
-  
-end
-end

data/lib/llm/shell/internal/coderay/lib/coderay/encoders/count.rb

@@ -1,39 +0,0 @@
-module CodeRay
-module Encoders
-  
-  # Returns the number of tokens.
-  # 
-  # Text and block tokens are counted.
-  class Count < Encoder
-    
-    register_for :count
-    
-  protected
-    
-    def setup options
-      super
-      
-      @count = 0
-    end
-    
-    def finish options
-      output @count
-    end
-    
-  public
-    
-    def text_token text, kind
-      @count += 1
-    end
-    
-    def begin_group kind
-      @count += 1
-    end
-    alias end_group begin_group
-    alias begin_line begin_group
-    alias end_line begin_group
-    
-  end
-  
-end
-end

data/lib/llm/shell/internal/coderay/lib/coderay/encoders/debug.rb

@@ -1,49 +0,0 @@
-module CodeRay
-module Encoders
-  
-  # = Debug Encoder
-  #
-  # Fast encoder producing simple debug output.
-  #
-  # It is readable and diff-able and is used for testing.
-  #
-  # You cannot fully restore the tokens information from the
-  # output, because consecutive :space tokens are merged.
-  # 
-  # See also: Scanners::Debug
-  class Debug < Encoder
-    
-    register_for :debug
-    
-    FILE_EXTENSION = 'raydebug'
-    
-    def text_token text, kind
-      if kind == :space
-        @out << text
-      else
-        text = text.gsub('\\', '\\\\\\\\') if text.index('\\')
-        text = text.gsub(')',  '\\\\)')    if text.index(')')
-        @out << "#{kind}(#{text})"
-      end
-    end
-    
-    def begin_group kind
-      @out << "#{kind}<"
-    end
-    
-    def end_group kind
-      @out << '>'
-    end
-    
-    def begin_line kind
-      @out << "#{kind}["
-    end
-    
-    def end_line kind
-      @out << ']'
-    end
-    
-  end
-  
-end
-end

data/lib/llm/shell/internal/coderay/lib/coderay/encoders/debug_lint.rb

@@ -1,63 +0,0 @@
-module CodeRay
-module Encoders
-  
-  load :lint
-  
-  # = Debug Lint Encoder
-  #
-  # Debug encoder with additional checks for:
-  # 
-  # - empty tokens
-  # - incorrect nesting
-  # 
-  # It will raise an InvalidTokenStream exception when any of the above occurs.
-  # 
-  # See also: Encoders::Debug
-  class DebugLint < Debug
-    
-    register_for :debug_lint
-    
-    def text_token text, kind
-      raise Lint::EmptyToken,       'empty token for %p' % [kind] if text.empty?
-      raise Lint::UnknownTokenKind, 'unknown token kind %p (text was %p)' % [kind, text] unless TokenKinds.has_key? kind
-      super
-    end
-    
-    def begin_group kind
-      @opened << kind
-      super
-    end
-    
-    def end_group kind
-      raise Lint::IncorrectTokenGroupNesting, 'We are inside %s, not %p (end_group)' % [@opened.reverse.map(&:inspect).join(' < '), kind] if @opened.last != kind
-      @opened.pop
-      super
-    end
-    
-    def begin_line kind
-      @opened << kind
-      super
-    end
-    
-    def end_line kind
-      raise Lint::IncorrectTokenGroupNesting, 'We are inside %s, not %p (end_line)' % [@opened.reverse.map(&:inspect).join(' < '), kind] if @opened.last != kind
-      @opened.pop
-      super
-    end
-    
-    protected
-    
-    def setup options
-      super
-      @opened = []
-    end
-    
-    def finish options
-      raise 'Some tokens still open at end of token stream: %p' % [@opened] unless @opened.empty?
-      super
-    end
-    
-  end
-  
-end
-end

data/lib/llm/shell/internal/coderay/lib/coderay/encoders/div.rb

@@ -1,23 +0,0 @@
-module CodeRay
-module Encoders
-  
-  load :html
-  
-  # Wraps HTML output into a DIV element, using inline styles by default.
-  # 
-  # See Encoders::HTML for available options.
-  class Div < HTML
-    
-    FILE_EXTENSION = 'div.html'
-    
-    register_for :div
-    
-    DEFAULT_OPTIONS = HTML::DEFAULT_OPTIONS.merge \
-      :css          => :style,
-      :wrap         => :div,
-      :line_numbers => false
-    
-  end
-  
-end
-end