llm-shell 0.4.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6aec859f4eb4fd9bfc722da0f0ccbaada65932b56bbd5d118f8131be32097e5
-  data.tar.gz: 0156acf8d3efbec33ce61aab82e5cc90f2618888b7cce6fa181c5de5b11b591c
+  metadata.gz: c767e6b583094aec1b80bc31d20de4da454d7ec70f93186118e247ef00deb8b3
+  data.tar.gz: 1a85b31965a9a33eedab86ffcfa6812f001d9b2941fc7c7f0b3c857a94172ff9
 SHA512:
-  metadata.gz: bdc7e09ec212ee04005bf636caa64abbfdbb087a6d20506402aaf81703806d4c3ddf7611f0c4dc61eedaa7711f6fe50880be0691a07181cb9d582e19fbcf2ba8
-  data.tar.gz: 7ce76e58b455c55ee94d52b2913396e514d2b808c0fa00ac57c5dff34b499c4cf1def4f3acc398ddfd602cb9f202a3261f8c51ea6012c89853c3a9d79991f9d5
+  metadata.gz: 8d3b433cf8aa6c01185baa03682171fc98a57d22e0edc211898db47e6a44190eabfa7fa16d65be37317c1ba6a9d2b95e8f61568bfe877067b2b0a01d53b88176
+  data.tar.gz: 6b7c997d7f512614dc3d84c6801d09d59b9bdec0d1fb8a97b2039c4d9ddba92ecceed95d9b745d75bdfa0c483f4c3cddc2ccf2e674e8d0d1dff4fe5d33261ef9

data/README.md

@@ -12,7 +12,7 @@ it in action!
 #### General
 
 - 🌟 Unified interface for multiple Large Language Models (LLMs)
-- 🤝 Supports Gemini, OpenAI, Anthropic, LlamaCpp and Ollama
+- 🤝 Supports Gemini, OpenAI, Anthropic, DeepSeek, LlamaCpp and Ollama
 
 #### Customize
 
@@ -35,13 +35,13 @@ it in action!
 </details>
 
 <details>
-  <summary><b>2. Files: import at boot time</b></summary>
-  <img src="share/llm-shell/examples/files-boottime.gif">
+  <summary><b>2. Files: import at runtime</b></summary>
+  <img src="share/llm-shell/examples/files-runtime.gif">
 </details>
 
 <details>
-<summary><b>3. Files: import at runtime</b></summary>
-  <img src="share/llm-shell/examples/files-runtime.gif">
+  <summary><b>3. Files: import at boot time</b></summary>
+  <img src="share/llm-shell/examples/files-boottime.gif">
 </details>
 
 ## Customization
@@ -80,8 +80,10 @@ end
 llm-shell can be extended with your own console commands. This can be
 done by creating a Ruby file in the `~/.llm-shell/commands/` directory &ndash;
 with one file per command. The commands are loaded at boot time. See the
-[import-file](lib/llm/shell/commands/import_file.rb)
-command for a realistic example:
+[file-import](lib/llm/shell/commands/file_import.rb)
+and
+[dir-import](lib/llm/shell/commands/dir_import.rb)
+commands for a realistic example:
 
 ```ruby
 LLM.command "say-hello" do |cmd|
@@ -112,6 +114,9 @@ gemini:
 anthropic:
   key: YOURKEY
   model: claude-3-7-sonnet-20250219
+deepseek:
+  key: YOURKEY
+  model: deepseek-chat
 ollama:
   host: localhost
   model: deepseek-coder:6.7b

data/lib/llm/shell/command/extension.rb

@@ -16,7 +16,7 @@ class LLM::Shell::Command
     # @return [void]
     def command(name)
       cmd = LLM::Shell::Command.new
-      cmd.name(name) if name
+      cmd.name(name)
       yield cmd
       commands[cmd.name] = cmd
     end

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

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+class LLM::Shell::Command
+  class DirImport
+    ##
+    # 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
+
+    ##
+    # @param [LLM::Shell::Context] context
+    #  The context of the command
+    # @return [LLM::Shell::Command::DirImport]
+    def initialize(context)
+      @context = context
+    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
+
+    private
+
+    def import(file)
+      return unless File.file?(file)
+      bot.chat [
+        "--- START: #{file} ---",
+        File.read(file),
+        "--- END: #{file} ---"
+      ].join("\n")
+    end
+
+    def bot = @context.bot
+    def io = @context.io
+  end
+
+  LLM.command "dir-import" do |cmd|
+    cmd.description "Share the contents of a directory with the LLM"
+    cmd.register(DirImport)
+  end
+end

data/lib/llm/shell/commands/{import_file.rb → file_import.rb}

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 class LLM::Shell::Command
-  class ImportFile
+  class FileImport
     ##
     # Completes a path with a wildcard.
     # @param path [String]
@@ -15,11 +15,14 @@ class LLM::Shell::Command
     ##
     # @param [LLM::Shell::Context] context
     #  The context of the command
-    # @return [LLM::Shell::Command::ImportFile]
+    # @return [LLM::Shell::Command::FileImport]
     def initialize(context)
       @context = context
     end
 
+    ##
+    # Imports one or more globbed files.
+    # @return [void]
     def call(*files)
       Dir[*files].each { import(_1) }
     end
@@ -27,6 +30,7 @@ class LLM::Shell::Command
     private
 
     def import(file)
+      return unless File.file?(file)
       bot.chat [
         "--- START: #{file} ---",
         File.read(file),
@@ -38,8 +42,8 @@ class LLM::Shell::Command
     def io = @context.io
   end
 
-  LLM.command "import-file" do |cmd|
+  LLM.command "file-import" do |cmd|
     cmd.description "Share one or more files with the LLM"
-    cmd.register ImportFile
+    cmd.register(FileImport)
   end
 end

data/lib/llm/shell/repl.rb

@@ -10,7 +10,7 @@ class LLM::Shell
     # @param [LLM::Chat] bot
     # @param [LLM::Shell::Options] options
     # @return [LLM::Shell::REPL]
-    def initialize(bot, options:)
+    def initialize(bot:, options:)
       @bot = bot
       @console = IO.console
       @options = options

data/lib/llm/shell/version.rb

@@ -4,5 +4,5 @@ module LLM
 end unless defined?(LLM)
 
 class LLM::Shell
-  VERSION = "0.4.1"
+  VERSION = "0.5.0"
 end

data/lib/llm/shell.rb

@@ -48,7 +48,7 @@ class LLM::Shell
     @config  = Config.new(options[:provider])
     @options = Options.new @config.merge(options), Default.new(options[:provider])
     @bot  = LLM::Chat.new(llm, {tools:}.merge(@options.chat)).lazy
-    @repl = REPL.new(@bot, options: @options)
+    @repl = REPL.new(bot: @bot, options: @options)
   end
 
   ##

data/libexec/llm-shell/shell

@@ -4,13 +4,17 @@
 require_relative "../../lib/llm/shell"
 
 def main(argv)
-  options = {tools: []}
-  option_parser.parse(argv, into: options)
-  if argv.empty? || options[:provider].nil?
-    warn option_parser.help
-    throw(:exit, 1)
+  if argv.include?("-v") || argv.include?("--version")
+    puts LLM::Shell::VERSION
   else
-    LLM::Shell.new(options).start
+    options = {tools: []}
+    option_parser.parse(argv, into: options)
+    if argv.empty? || options[:provider].nil?
+      warn option_parser.help
+      throw(:exit, 1)
+    else
+      LLM::Shell.new(options).start
+    end
   end
 end
 
@@ -24,6 +28,7 @@ def option_parser
     o.on("-o [PORT]", "--port [PORT]", "Optional. Sometimes required by ollama.", Integer)
     o.on("-f [GLOB]", "--files [GLOB]", "Optional. Glob pattern(s) separated by a comma.", Array)
     o.on("-t [TOOLS]", "--tools [TOOLS]", "Optional. One or more tool names to load automatically.", Array)
+    o.on("-v", "--version", "Optional. Print the version and exit.")
   end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm-shell
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - Antar Azri
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-05-11 00:00:00.000000000 Z
+date: 2025-05-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: llm.rb
@@ -17,14 +17,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.7'
+        version: '0.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.7'
+        version: '0.8'
 - !ruby/object:Gem::Dependency
   name: paint
   requirement: !ruby/object:Gem::Requirement
@@ -224,7 +224,8 @@ files:
 - lib/llm/shell.rb
 - lib/llm/shell/command.rb
 - lib/llm/shell/command/extension.rb
-- lib/llm/shell/commands/import_file.rb
+- lib/llm/shell/commands/dir_import.rb
+- lib/llm/shell/commands/file_import.rb
 - lib/llm/shell/completion.rb
 - lib/llm/shell/config.rb
 - lib/llm/shell/default.rb
@@ -249,7 +250,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.0.0
+      version: '3.2'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="