hyrum 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2c973fd6cbfc5ee63d43e581024c7fc9720501d6e544aa90f524fc6ce8e3b641
-  data.tar.gz: a080770b267980f238d35e0d7ea86613ad37b306075a887d8ec83eaf52cd19db
+  metadata.gz: b7177d332d4a6a3e7a9527a906e39d4f141af5444aedbb3acff837b07b93f619
+  data.tar.gz: 9b0d67445326a57b5aa4f1bc2926cf06455609f491c75c08e65889a8d019b481
 SHA512:
-  metadata.gz: 95833489cffa0a13282d7c294e43346d6aff1aa762629f03a49b2b65dc663f3e9ec722861d4f2b5325fb5351ee3ad7444995f3981a43ee541fb9583f003579a8
-  data.tar.gz: 7b198b8dce4536f38f1e37b07fa7dad5c83bbcd4121e5a9ff76d16029aba45f5b60ba012b8cb153b3c5e9601834b6853daf6beee109321a4c67a6efeb41ffde2
+  metadata.gz: aa204a19e6e0080792381c09904cc0fcf013b05490875b4f793f4135b3cec0d605d544261266a6b00f54be2a30ddfef4d3fd44e2f8be17060f49924fabe528b3
+  data.tar.gz: 17ceb22c5ea1b7a3837ce9195cc4b0f8a9b4276d502e3a9b4f4450aec61f5bcb780eaf135f2d9ac92717c060a81cf1423af751ef34de54faf532850c67b024f0

data/CHANGELOG.md

@@ -5,12 +5,19 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [0.0.1] - 2024-12-01
+## [0.1.0] - 2024-12-16
+
+### Fixed
+
+- Minor bug fixes and spec updates
+- Option defaults added in help where appropriate
 
 ### Added
 
-- This CHANGELOG file
-- Initial version of the project
+- New -n | --number option to specify number of messages to produce
+- New ScriptOption specs
+- New MessageGenerator specs
+- New FakeGenerator specs
 
 ## [0.0.2] - 2024-12-06
 
@@ -22,3 +29,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - New OpenAI generator specs
+
+## [0.0.1] - 2024-12-01
+
+### Added
+
+- This CHANGELOG file
+- Initial version of the project

data/lib/hyrum/generators/fake_generator.rb

@@ -40,7 +40,7 @@ module Hyrum
 
       def initialize(options)
         @options = options
-        @ai_service = options[:ai_service]
+        # @ai_service = options[:ai_service]
       end
 
       def generate

data/lib/hyrum/generators/message_generator.rb

@@ -18,9 +18,11 @@ module Hyrum
 
     class MessageGenerator
       def self.create(options)
-        generator_class = GENERATOR_CLASSES[options[:ai_service].to_sym]
+        unless GENERATOR_CLASSES.key?(options[:ai_service].to_sym)
+          raise ArgumentError, "Invalid AI service: #{options[:ai_service]}"
+        end
 
-        # Add error handling for invalid format
+        generator_class = GENERATOR_CLASSES[options[:ai_service].to_sym]
         generator_class.new(options)
       end
     end

data/lib/hyrum/generators/openai_generator.rb

@@ -26,12 +26,12 @@ module Hyrum
 
       def prompt
         prompt = <<~PROMPT
-          Please provide 5 alternative status messages for the following message:
+          Please provide <%= number %> alternative status messages for the following message:
           `<%= message %>`. The messages should be unique and informative. The messages
           should be returned as json in the format: `{ "<%= key %>": ['list', 'of', 'messages']}`
           The key should be `"<%= key %>"` followed by the list of messages.
         PROMPT
-        erb_hash = { key: options[:key], message: options[:message] }
+        erb_hash = { key: options[:key], message: options[:message], number: options[:number] }
         template = ERB.new(prompt, trim_mode: '-')
         template.result_with_hash(erb_hash)
       end

data/lib/hyrum/script_options.rb

@@ -3,6 +3,8 @@
 require 'optparse'
 
 module Hyrum
+  class ScriptOptionsError < StandardError; end
+
   class ScriptOptions
     MANDATORY_OPTIONS = %i[message].freeze
 
@@ -22,16 +24,11 @@ module Hyrum
       set_dynamic_defaults
       options
     rescue OptionParser::InvalidOption => e
-      err = "Invalid option: #{e.message}"
+      raise ScriptOptionsError.new("Invalid option: #{e.message}")
     rescue OptionParser::MissingArgument => e
-      err = "Missing argument for option: #{e.message}"
+      raise ScriptOptionsError.new("Missing argument for option: #{e.message}")
     rescue OptionParser::InvalidArgument => e
-      err = "Invalid argument for option: #{e.message}"
-    ensure
-      if err
-        puts err
-        exit
-      end
+      raise ScriptOptionsError.new("Invalid argument for option: #{e.message}")
     end
 
     private
@@ -74,7 +71,7 @@ module Hyrum
     def ai_service_options(parser)
       options[:ai_service] = :fake
 
-      description = "AI service: one of #{Generators::AI_SERVICES.join(', ')}"
+      description = "AI service: one of #{Generators::AI_SERVICES.join(', ')} (default: fake)"
       parser.on('-s SERVICE', '--service SERVICE', Generators::AI_SERVICES, description) do |service|
         options[:ai_service] = service.to_sym
       end
@@ -88,15 +85,21 @@ module Hyrum
     def message_key_options(parser)
       options[:key] = :status
 
-      parser.on('-k KEY', '--key KEY', 'Message key') do |key|
+      parser.on('-k KEY', '--key KEY', 'Message key (default: status)') do |key|
         options[:key] = key.to_sym
       end
     end
 
     def message_options(parser)
-      parser.on('-m MESSAGE', '--message MESSAGE', 'Status message') do |message|
+      parser.on('-m MESSAGE', '--message MESSAGE', 'Status message (required)') do |message|
         options[:message] = message
       end
+
+      options[:number] = 5
+
+      parser.on('-n NUMBER', '--number NUMBER', Integer, 'Number of messages to generate (default: 5)',) do |number|
+        options[:number] = number.to_i
+      end
     end
 
     def verbosity_options(parser)
@@ -111,7 +114,7 @@ module Hyrum
       formats = Formats::FORMATS
       description = 'Output format. Supported formats are:'
       supported   = formats.join(', ')
-      parser.on('-f FORMAT', '--format FORMAT', formats, description, supported) do |format|
+      parser.on('-f FORMAT', '--format FORMAT', formats, description, supported, "(default: text)") do |format|
         options[:format] = format
       end
     end

data/lib/hyrum/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Hyrum
-  VERSION = '0.0.2'
-end
+  VERSION = '0.1.0'
+end

data/lib/hyrum.rb

@@ -8,7 +8,7 @@ loader.setup
 module Hyrum
   def self.run(args)
     options = ScriptOptions.new(args).parse
-    generator_opts = options.slice(:message, :key, :ai_service, :ai_model, :verbose)
+    generator_opts = options.slice(:message, :key, :ai_service, :ai_model, :number, :verbose)
     formatter_opts = options.slice(:format, :verbose)
 
     puts "Options: #{options.inspect}" if options[:verbose]
@@ -17,6 +17,9 @@ module Hyrum
     messages = message_generator.generate
     output = formatter.format(messages)
     puts output
+  rescue ScriptOptionsError => e
+    puts e.message
+    exit
   end
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hyrum
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - Tracy Atteberry
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-12-07 00:00:00.000000000 Z
+date: 2024-12-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai