rename llamafile to llamafile_cli

Author: wwwwhiterabbit

lib/rspec/llama.rb

@@ -12,20 +12,20 @@ module Llama
       # Model configurations
       OpenaiModelConfiguration: 'rspec/llama/model_configurations/openai_model_configuration',
       LlamaCppModelConfiguration: 'rspec/llama/model_configurations/llama_cpp_model_configuration',
-      LlamafileModelConfiguration: 'rspec/llama/model_configurations/llamafile_model_configuration',
+      LlamafileCliModelConfiguration: 'rspec/llama/model_configurations/llamafile_cli_model_configuration',
       OllamaModelConfiguration: 'rspec/llama/model_configurations/ollama_model_configuration',
 
       # Model runners
       OpenaiModelRunner: 'rspec/llama/model_runners/openai_model_runner',
       LlamaCppModelRunner: 'rspec/llama/model_runners/llama_cpp_model_runner',
-      LlamafileModelRunner: 'rspec/llama/model_runners/llamafile_model_runner',
+      LlamafileCliModelRunner: 'rspec/llama/model_runners/llamafile_cli_model_runner',
       OllamaModelRunner: 'rspec/llama/model_runners/ollama_model_runner',
 
       # Model runner results
       BaseModelRunnerResult: 'rspec/llama/model_runner_results/base_model_runner_result',
       OpenaiModelRunnerResult: 'rspec/llama/model_runner_results/openai_model_runner_result',
       LlamaCppModelRunnerResult: 'rspec/llama/model_runner_results/llama_cpp_model_runner_result',
-      LlamafileModelRunnerResult: 'rspec/llama/model_runner_results/llamafile_model_runner_result',
+      LlamafileCliModelRunnerResult: 'rspec/llama/model_runner_results/llamafile_cli_model_runner_result',
       OllamaModelRunnerResult: 'rspec/llama/model_runner_results/ollama_model_runner_result'
     }.each { |class_name, path| autoload class_name, path }
   end

lib/rspec/llama/helpers.rb

@@ -7,14 +7,14 @@ module Helpers
         openai: 'OpenaiModelRunner',
         ollama: 'OllamaModelRunner',
         llama_cpp: 'LlamaCppModelRunner',
-        llamafile: 'LlamafileModelRunner'
+        llamafile_cli: 'LlamafileCliModelRunner'
       }.freeze
 
       CONFIGURATION_TYPES = {
         openai: 'OpenaiModelConfiguration',
         ollama: 'OllamaModelConfiguration',
         llama_cpp: 'LlamaCppModelConfiguration',
-        llamafile: 'LlamafileModelConfiguration'
+        llamafile_cli: 'LlamafileCliModelConfiguration'
       }.freeze
 
       ASSERTION_TYPES = {

lib/rspec/llama/model_configurations/llamafile_model_configuration.rb renamed to lib/rspec/llama/model_configurations/llamafile_cli_model_configuration.rb

@@ -2,7 +2,7 @@
 
 module RSpec
   module Llama
-    class LlamafileModelConfiguration
+    class LlamafileCliModelConfiguration
       DEFAULT_TEMPERATURE = 0.5
       DEFAULT_PREDICT = 500
 
@@ -18,10 +18,10 @@ class LlamafileModelConfiguration
       #   are either true (for flags) or values for key-value pairs. These options are passed directly to the CLI.
       #
       # @example Basic usage with a specific model path and default parameters
-      #   config = RSpec::Llama::LlamafileModelConfiguration.new
+      #   config = RSpec::Llama::LlamafileCliModelConfiguration.new
       #
       # @example Custom parameters and additional CLI options
-      #   config = RSpec::Llama::LlamafileModelConfiguration.new(
+      #   config = RSpec::Llama::LlamafileCliModelConfiguration.new(
       #     temperature: 0.7, predict: 300, threads: 8
       #   )
       def initialize(

lib/rspec/llama/model_runner_results/llamafile_model_runner_result.rb renamed to lib/rspec/llama/model_runner_results/llamafile_cli_model_runner_result.rb

@@ -2,7 +2,7 @@
 
 module RSpec
   module Llama
-    class LlamafileModelRunnerResult
+    class LlamafileCliModelRunnerResult
       attr_reader :result
 
       def initialize(result)

lib/rspec/llama/model_runners/llamafile_cli_model_runner.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module RSpec
+  module Llama
+    class LlamafileCliModelRunner
+      attr_reader :path
+
+      def initialize(path:)
+        @path = path
+      end
+
+      # Runs the model with the given configuration and prompt.
+      #
+      # @param [RSpec::Llama::LlamafileCliModelConfiguration] configuration
+      # @param [RSpec::Llama::ModelPrompt] prompt
+      #
+      # @return [RSpec::Llama::LlamafileCliModelRunnerResult]
+      def call(configuration, prompt)
+        command = [path, '--prompt', prompt.message] + configuration.to_a
+
+        IO.popen(command, 'r+') do |io|
+          LlamafileCliModelRunnerResult.new(io.read)
+        end
+      end
+    end
+  end
+end

lib/rspec/llama/model_runners/llamafile_model_runner.rb

@@ -60,24 +60,24 @@
       end
     end
 
-    context 'with llamafile model configuration' do
-      it 'builds a LlamafileModelConfiguration with default options' do
-        config = helpers.build_model_configuration(:llamafile)
+    context 'with llamafile_cli model configuration' do
+      it 'builds a LlamafileCliModelConfiguration with default options' do
+        config = helpers.build_model_configuration(:llamafile_cli)
 
         expect(config).to have_attributes(
-          class: RSpec::Llama::LlamafileModelConfiguration,
+          class: RSpec::Llama::LlamafileCliModelConfiguration,
           temperature: 0.5,
           predict: 500
         )
       end
 
       it 'builds a LlamaCppModelConfiguration with given options' do
         config = helpers.build_model_configuration(
-          :llamafile, temperature: 0.7, threads: 8, predict: 100
+          :llamafile_cli, temperature: 0.7, threads: 8, predict: 100
         )
 
         expect(config).to have_attributes(
-          class: RSpec::Llama::LlamafileModelConfiguration,
+          class: RSpec::Llama::LlamafileCliModelConfiguration,
           temperature: 0.7,
           predict: 100,
           additional_options: { threads: 8 }
@@ -121,12 +121,12 @@
       )
     end
 
-    it 'builds a LlamafileModelRunner' do
-      runner = helpers.build_model_runner(:llamafile, cli_path: '/path/to/llamafile')
+    it 'builds a LlamafileCliModelRunner' do
+      runner = helpers.build_model_runner(:llamafile_cli, path: '/path/to/llamafile')
 
       expect(runner).to have_attributes(
-        class: RSpec::Llama::LlamafileModelRunner,
-        cli_path: '/path/to/llamafile'
+        class: RSpec::Llama::LlamafileCliModelRunner,
+        path: '/path/to/llamafile'
       )
     end
   end

spec/rspec/llama/helpers_spec.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-RSpec.describe RSpec::Llama::LlamafileModelConfiguration do
+RSpec.describe RSpec::Llama::LlamafileCliModelConfiguration do
   describe '#initialize' do
     it 'initializes with default options' do
       config = described_class.new

spec/rspec/llama/model_configurations/llamafile_model_configuration_spec.rb renamed to spec/rspec/llama/model_configurations/llamafile_cli_model_configuration_spec.rb

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 
-RSpec.describe RSpec::Llama::LlamafileModelRunner do
-  subject(:call_runner!) { described_class.new(cli_path:).call(model_configuration, model_prompt) }
+RSpec.describe RSpec::Llama::LlamafileCliModelRunner do
+  subject(:call_runner!) { described_class.new(path: cli_path).call(model_configuration, model_prompt) }
 
   let(:cli_path) { './llava-v1.5-7b-q4.llamafile' }
   let(:cli_options) { ['--cli', '--silent-prompt', '--log-disable', '--temp', '0.1', '--seed', '20'] }
-  let(:model_configuration) { instance_double(RSpec::Llama::LlamafileModelConfiguration, to_a: cli_options) }
+  let(:model_configuration) { instance_double(RSpec::Llama::LlamafileCliModelConfiguration, to_a: cli_options) }
   let(:model_prompt) { 'Who created the Ruby language?' }
   let(:response) do
     'Matz, a Japanese computer programmer, created the Ruby programming language. He released the first ' \
@@ -19,8 +19,8 @@
 
       result = call_runner!
 
-      expect(result).to have_attributes(class: RSpec::Llama::LlamafileModelRunnerResult, to_s: response)
-      expect(IO).to have_received(:popen).with([cli_path, '--prompt', model_prompt_message] + cli_options, 'r+')
+      expect(result).to have_attributes(class: RSpec::Llama::LlamafileCliModelRunnerResult, to_s: response)
+      expect(IO).to have_received(:popen).with([cli_path, '--prompt', model_prompt] + cli_options, 'r+')
     end
   end
 end