commitgpt 0.3.8 → 0.3.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5087467607c5fa8e321e66d9515ac5ebd4a65ef513a1cb0dc9a5819db4e72601
-  data.tar.gz: 3e33525a45b3892e4811671dde772df60e230d22b1b011019f3906ab4752ff1e
+  metadata.gz: d812f02e181a923e72486f7ffc5ac1d2e7614bd5eee7194bca7e3a764df7afda
+  data.tar.gz: 7195e0edac02853ef4d1dcc7179664beffd023dacdb36c1133c5a01463a0175c
 SHA512:
-  metadata.gz: c78638efe575fd73035cffee4bd8779eace5552315beb8c5400a6223d99410dd4981fef26c755156920fd65206f99b9c4bdb3530b638afdf2dfbb3bd76e5190c
-  data.tar.gz: f40cc753a3b6849df45d2b8a11b19d008ad20661f4461d5ba61f54a20ce8234592fa6432b64354a7e27bbfd0ffda1dde82fb2f5385bd34dec76fab6bc3012f75
+  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

@@ -10,9 +10,6 @@ require_relative 'string'
 module CommitGpt
   # Interactive setup wizard for configuring AI providers
   class SetupWizard
-    # Seconds to wait when fetching the provider's model list
-    MODEL_FETCH_TIMEOUT = 30
-
     def initialize
       @prompt = TTY::Prompt.new
     end
@@ -22,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
@@ -248,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(MODEL_FETCH_TIMEOUT) do
+        Timeout.timeout(timeout) do
           headers = {
             'Content-Type' => 'application/json',
             'User-Agent' => "Ruby/#{RUBY_VERSION}"
@@ -272,7 +287,8 @@ module CommitGpt
           end
         end
       rescue Timeout::Error
-        puts "✖ Connection timeout (#{MODEL_FETCH_TIMEOUT}s). 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.8'
+  VERSION = '0.3.9'
 end

metadata

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