commitgpt 0.3.7 → 0.3.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0f0375fced0367f258ad0a60fdda99b10e11e42c2919b87b1a4e3b7d34c64a39
-  data.tar.gz: b900f3eb0ad5fe040bbcdc1c928bb1e3843b94a3251141911e3ba28816775fde
+  metadata.gz: d812f02e181a923e72486f7ffc5ac1d2e7614bd5eee7194bca7e3a764df7afda
+  data.tar.gz: 7195e0edac02853ef4d1dcc7179664beffd023dacdb36c1133c5a01463a0175c
 SHA512:
-  metadata.gz: '079b9b9c114a763033df5a4cddd958e51a9c07b0ed9d09216499d8e60ca9f0f1d3f264a3eb24b1566f215f928e91c73d1cd0318bf1ad812705da02a24bddce94'
-  data.tar.gz: c95285ec608aaa8e5b534d8cc5e6ed2bd3682f460f5b0df7744408521bef5e71f6662b38974e4abf56a7ee4086bb1aed2cf6396ff3d7df61e69f69b16c234e1d
+  metadata.gz: 61f5d8a81bc519c7341dde8808656d729c8c3c6c826aa4ecb27779a5b59b3866f8f4fc671424139aa1fa801ce0d1bee14cac903363babafb082f5e3eb354904d
+  data.tar.gz: a32a40428daef8ecfe051e39fc0f4cf4040755190c01f0e92eb747aed232b8cc4acabe9ba88b6a9d27b0b7edf666d2da96a07f7d0e75e80ca47adc33b1eb9227

data/README.md

@@ -121,10 +121,13 @@ $ aicm setup
 ```
 
 You'll be guided to:
-1. Choose an AI provider (Presets: Cerebras, OpenAI, Ollama, Groq, etc.)
-2. Enter your API Key (stored securely in `config.local.yml`)
-3. Select a model interactively
-4. Set maximum diff length
+1. Set the model list fetch timeout in seconds (defaults to 30; raise it if a provider is slow to respond)
+2. Choose an AI provider (Presets: Cerebras, OpenAI, Ollama, Groq, etc.)
+3. Enter your API Key (stored securely in `config.local.yml`)
+4. Select a model interactively
+5. Set maximum diff length
+
+The timeout is saved as a top-level `model_fetch_timeout` key in `config.yml` and applies to every provider, including `aicm -p` and `aicm -m`. You can also edit it there directly.
 
 **Note:** Please add `~/.config/commitgpt/config.local.yml` to your `.gitignore` if you are syncing your home directory, as it contains your API keys.
 

data/lib/commitgpt/cli.rb

@@ -63,6 +63,7 @@ module CommitGpt
           shell.say "  Format:    #{format.capitalize.yellow}"
           shell.say "  Base URL:  #{config['base_url']}"
           shell.say "  Diff Len:  #{config['diff_len']}"
+          shell.say "  Timeout:   #{CommitGpt::ConfigManager.get_model_fetch_timeout}s"
           shell.say ''
         end
       rescue StandardError

data/lib/commitgpt/config_manager.rb

@@ -8,6 +8,9 @@ require_relative 'string'
 module CommitGpt
   # Manages configuration files for CommitGPT
   class ConfigManager
+    # Seconds to wait when fetching a provider's model list
+    DEFAULT_MODEL_FETCH_TIMEOUT = 30
+
     class << self
       # Get the config directory path
       def config_dir
@@ -165,6 +168,23 @@ module CommitGpt
         save_main_config(main_config)
       end
 
+      # Get the model list fetch timeout (seconds)
+      def get_model_fetch_timeout
+        return DEFAULT_MODEL_FETCH_TIMEOUT unless config_exists?
+
+        main_config = YAML.load_file(main_config_path)
+        timeout = main_config['model_fetch_timeout'].to_i
+        timeout.positive? ? timeout : DEFAULT_MODEL_FETCH_TIMEOUT
+      end
+
+      # Set the model list fetch timeout (seconds)
+      def set_model_fetch_timeout(seconds)
+        ensure_config_dir
+        main_config = config_exists? ? YAML.load_file(main_config_path) : { 'providers' => [], 'active_provider' => '' }
+        main_config['model_fetch_timeout'] = seconds
+        save_main_config(main_config)
+      end
+
       private
 
       # Merge main config with local config (local overrides main)

data/lib/commitgpt/setup_wizard.rb

@@ -19,6 +19,7 @@ module CommitGpt
       ConfigManager.ensure_config_dir
       ConfigManager.generate_default_configs unless ConfigManager.config_exists?
 
+      prompt_model_fetch_timeout
       provider_choice = select_provider
       configure_provider(provider_choice)
     end
@@ -245,13 +246,30 @@ module CommitGpt
       end
     end
 
+    # Ask how long to wait when fetching the model list, and persist it
+    def prompt_model_fetch_timeout
+      current = ConfigManager.get_model_fetch_timeout
+
+      seconds = @prompt.ask('Set the timeout (seconds) for fetching the model list:') do |q|
+        q.default current.to_s
+        q.convert :int
+        q.validate ->(input) { input.to_i.positive? }
+        q.messages[:valid?] = 'Timeout must be a positive number of seconds'
+      end
+
+      seconds = current unless seconds.is_a?(Integer) && seconds.positive?
+      ConfigManager.set_model_fetch_timeout(seconds)
+      seconds
+    end
+
     # Fetch models from provider with timeout
     def fetch_models_with_timeout(base_url, api_key)
+      timeout = ConfigManager.get_model_fetch_timeout
       puts 'Fetching available models...'.gray
 
       models = nil
       begin
-        Timeout.timeout(5) do
+        Timeout.timeout(timeout) do
           headers = {
             'Content-Type' => 'application/json',
             'User-Agent' => "Ruby/#{RUBY_VERSION}"
@@ -269,7 +287,8 @@ module CommitGpt
           end
         end
       rescue Timeout::Error
-        puts '✖ Connection timeout (5s). Please check your network, base_url, and api_key.'.red
+        puts "✖ Connection timeout (#{timeout}s). Please check your network, base_url, and api_key.".red
+        puts "  Run 'aicm setup' again to raise the timeout.".gray
         exit(0)
       rescue StandardError => e
         puts "✖ Error fetching models: #{e.message}".red

data/lib/commitgpt/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CommitGpt
-  VERSION = '0.3.7'
+  VERSION = '0.3.9'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: commitgpt
 version: !ruby/object:Gem::Version
-  version: 0.3.7
+  version: 0.3.9
 platform: ruby
 authors:
 - Peng Zhang