Skip to content

Commit 84e17c3

Browse files
committed
Add Cohere models support to Chat and Vision model specs
1 parent 21bfcd4 commit 84e17c3

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

spec/spec_helper.rb

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@
6565
config.filter_sensitive_data('<ANTHROPIC_API_KEY>') { ENV.fetch('ANTHROPIC_API_KEY', nil) }
6666
config.filter_sensitive_data('<GEMINI_API_KEY>') { ENV.fetch('GEMINI_API_KEY', nil) }
6767
config.filter_sensitive_data('<DEEPSEEK_API_KEY>') { ENV.fetch('DEEPSEEK_API_KEY', nil) }
68+
config.filter_sensitive_data('<COHERE_API_KEY>') { ENV.fetch('COHERE_API_KEY', nil) }
6869
config.filter_sensitive_data('<OPENROUTER_API_KEY>') { ENV.fetch('OPENROUTER_API_KEY', nil) }
6970
config.filter_sensitive_data('<OLLAMA_API_BASE>') { ENV.fetch('OLLAMA_API_BASE', 'http://localhost:11434/v1') }
7071

@@ -114,6 +115,7 @@
114115
config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', 'test')
115116
config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', 'test')
116117
config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', 'test')
118+
config.cohere_api_key = ENV.fetch('COHERE_API_KEY', 'test')
117119
config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', 'test')
118120
config.ollama_api_base = ENV.fetch('OLLAMA_API_BASE', 'http://localhost:11434/v1')
119121

@@ -138,6 +140,7 @@
138140
{ provider: :deepseek, model: 'deepseek-chat' },
139141
{ provider: :openai, model: 'gpt-4.1-nano' },
140142
{ provider: :openrouter, model: 'anthropic/claude-3.5-haiku' },
143+
{ provider: :cohere, model: 'command-a-03-2025' },
141144
{ provider: :ollama, model: 'qwen3' }
142145
].freeze
143146

@@ -154,6 +157,7 @@
154157
{ provider: :gemini, model: 'gemini-2.0-flash' },
155158
{ provider: :openai, model: 'gpt-4.1-nano' },
156159
{ provider: :openrouter, model: 'anthropic/claude-3.5-haiku' },
160+
{ provider: :cohere, model: 'c4ai-aya-vision-32b' },
157161
{ provider: :ollama, model: 'qwen3' }
158162
].freeze
159163

0 commit comments

Comments
 (0)