Add Cohere model support to model and alias rake tasks

Author: infinityrobot

lib/tasks/aliases.rake

@@ -117,6 +117,18 @@ namespace :aliases do # rubocop:disable Metrics/BlockLength
       }
     end
 
+    # Cohere models
+    models['cohere'].each do |model|
+      openrouter_model = "cohere/#{model}"
+      next unless models['openrouter'].include?(openrouter_model)
+
+      alias_key = model.gsub(/-\d{2}-\d{4}$/, '')
+      aliases[alias_key] = {
+        'cohere' => model,
+        'openrouter' => openrouter_model
+      }
+    end
+
     # Write the result
     sorted_aliases = aliases.sort.to_h
     File.write('lib/ruby_llm/aliases.json', JSON.pretty_generate(sorted_aliases))

lib/tasks/models_update.rake

@@ -23,6 +23,7 @@ def configure_from_env
     config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
     config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
     config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', nil)
+    config.cohere_api_key = ENV.fetch('COHERE_API_KEY', nil)
     configure_bedrock(config)
     config.request_timeout = 30
   end