ruby_llm-providers-groq 2.0.0.pre.1

data/spec/ruby_llm/moderation_spec.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Moderation, :live do
+  include_context 'with configured RubyLLM'
+
+  each_model(MODERATION_MODELS) do |provider, model|
+    it "#{provider}/#{model} moderates content" do
+      moderation = RubyLLM.moderate(
+        'This is a safe message',
+        model: model,
+        provider: provider,
+        assume_model_exists: true
+      )
+
+      expect(moderation).to be_a(described_class)
+      expect(moderation.results).to all(be_a(RubyLLM::Moderation::Result))
+      expect(moderation.flagged?).to be_in([true, false])
+    end
+  end
+end

data/spec/ruby_llm/providers/groq_spec.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Providers::Groq do
+  subject(:provider) { described_class.new(config) }
+
+  let(:config) do
+    RubyLLM::Configuration.new.tap do |provider_config|
+      provider_config.groq_api_key = 'test-key'
+      provider_config.groq_api_base = 'https://example.test/v1'
+    end
+  end
+
+  it 'is registered with RubyLLM' do
+    expect(RubyLLM::Provider.resolve(:groq)).to eq(described_class)
+  end
+
+  it 'registers a default protocol' do
+    expect(described_class.protocols).to include(responses: described_class::Responses)
+  end
+
+  it 'declares provider configuration' do
+    expect(described_class.configuration_options).to eq(%i[groq_api_key groq_api_base])
+    expect(described_class.configuration_requirements).to eq(%i[groq_api_key])
+  end
+
+  it 'uses configured API base and bearer token' do
+    expect(provider.api_base).to eq('https://example.test/v1')
+    expect(provider.headers).to eq('Authorization' => 'Bearer test-key')
+  end
+end

data/spec/ruby_llm/rerank_spec.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Rerank, :live do
+  include_context 'with configured RubyLLM'
+
+  each_model(RERANK_MODELS) do |provider, model|
+    it "#{provider}/#{model} orders documents by relevance" do
+      rerank = RubyLLM.rerank(
+        'What is the capital of the United States?',
+        ['Carson City is the capital of Nevada.', 'Washington, D.C. is the capital of the United States.'],
+        model: model,
+        provider: provider,
+        assume_model_exists: true
+      )
+
+      expect(rerank.results.first.document).to include('Washington')
+      expect(rerank.results.first.score).to be > rerank.results.last.score
+      expect(rerank.results.map(&:index)).to contain_exactly(0, 1)
+    end
+  end
+end

data/spec/ruby_llm/speech_spec.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Speech, :live do
+  include_context 'with configured RubyLLM'
+
+  each_model(SPEECH_MODELS) do |provider, model, model_info|
+    it "#{provider}/#{model} speaks" do
+      speech = RubyLLM.speak(
+        'Ruby is a programming language designed for developer happiness.',
+        model: model,
+        provider: provider,
+        assume_model_exists: true,
+        voice: model_info[:voice],
+        format: model_info[:format]
+      )
+
+      expect(speech.data).to be_a(String)
+      expect(speech.data.bytesize).to be > 1000
+      expect(speech.model).to eq(model)
+      expect(speech.mime_type).to start_with('audio/')
+    end
+  end
+end

data/spec/ruby_llm/video_spec.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Video, :live do
+  include_context 'with configured RubyLLM'
+
+  before do
+    RubyLLM.config.video_generation_poll_interval = VCR.current_cassette&.recording? ? 5 : 0
+  end
+
+  each_model(VIDEO_GENERATION_MODELS) do |provider, model, model_info|
+    it "#{provider}/#{model} animates a video" do
+      video = RubyLLM.animate(
+        'a calm ocean wave at sunset',
+        model: model,
+        provider: provider,
+        assume_model_exists: true,
+        provider_options: model_info.fetch(:provider_options, {})
+      )
+
+      expect(video.mime_type).to include('video')
+      expect(video.url || video.data).not_to be_nil
+      expect(video.to_blob.bytesize).to be > 10_000
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'dotenv/load'
+require 'fileutils'
+
+require 'vcr'
+require 'ruby_llm/providers/groq'
+require 'webmock/rspec'
+
+Dir[File.expand_path('support/**/*.rb', __dir__)].each { |file| require file }
+
+RSpec.configure do |config|
+  config.disable_monkey_patching!
+  config.expect_with(:rspec) { |expectations| expectations.syntax = :expect }
+  config.order = :random
+  Kernel.srand config.seed
+
+  config.around(:each, :live) do |example|
+    cassette_name = example.full_description.downcase.gsub(/[^a-z0-9]+/, '_').delete_prefix('_').delete_suffix('_')
+    cassette_path = File.join(VCR.configuration.cassette_library_dir, "#{cassette_name}.yml")
+
+    VCR.use_cassette(cassette_name) { example.run }
+    FileUtils.rm_f(cassette_path) if example.exception
+  end
+end

data/spec/support/models.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+# Remove chat-derived matrices the model does not support and add real ids for the empty operation matrices.
+# See RubyLLM's full live matrix and specs: https://github.com/crmne/ruby_llm/tree/main/spec
+PROVIDER = :groq
+CHAT_MODELS = [
+  { provider: :groq, model: 'openai/gpt-oss-120b' },
+  { provider: :groq, model: 'qwen/qwen3.8-27b' }
+].freeze
+TOOL_MODELS = CHAT_MODELS
+STRUCTURED_OUTPUT_MODELS = CHAT_MODELS
+
+EMBEDDING_MODELS = [].freeze
+IMAGE_GENERATION_MODELS = [].freeze
+SPEECH_MODELS = [].freeze
+VIDEO_GENERATION_MODELS = [].freeze
+MODERATION_MODELS = [].freeze
+RERANK_MODELS = [].freeze
+
+def each_model(models)
+  models.each { |model_info| yield model_info[:provider], model_info[:model], model_info }
+end

data/spec/support/rubyllm_configuration.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+RSpec.shared_context 'with configured RubyLLM' do
+  before do
+    RubyLLM.configure do |config|
+      config.groq_api_key = ENV.fetch('GROQ_API_KEY', 'test')
+      config.groq_api_base = ENV.fetch('GROQ_API_BASE', 'https://api.groq.com/openai/v1')
+      config.max_retries = 0
+      config.retry_backoff_factor = 0
+      config.retry_interval = 0
+      config.retry_interval_randomness = 0
+    end
+  end
+end

data/spec/support/vcr_configuration.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+VCR.configure do |config|
+  config.cassette_library_dir = 'spec/fixtures/vcr_cassettes'
+  config.hook_into :webmock
+  config.default_cassette_options = { record: ENV['CI'] ? :none : :once }
+  config.allow_http_connections_when_no_cassette = true
+  config.filter_sensitive_data('<GROQ_API_KEY>') { ENV.fetch('GROQ_API_KEY', nil) }
+  config.filter_sensitive_data('https://api.groq.com/openai/v1') { ENV.fetch('GROQ_API_BASE', nil) }
+
+  config.before_record do |interaction|
+    next unless interaction.request.headers['Authorization']
+
+    interaction.request.headers['Authorization'] = ['Bearer <AUTH_TOKEN>']
+  end
+end

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification
+name: ruby_llm-providers-groq
+version: !ruby/object:Gem::Version
+  version: 2.0.0.pre.1
+platform: ruby
+authors:
+- Siarhiej Siamaška
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruby_llm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+description: Adds Groq provider support to RubyLLM.
+email:
+- siarhei.siamashka@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".flayignore"
+- ".github/workflows/ci.yml"
+- ".github/workflows/gitleaks.yml"
+- ".github/workflows/release.yml"
+- ".overcommit.yml"
+- ".rspec"
+- ".rubocop.yml"
+- Archspec.rb
+- LICENSE
+- README.md
+- lib/ruby_llm/providers/groq.rb
+- models.json
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_openai_gpt_oss_120b_can_handle_a_multi_turn_conversation.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_openai_gpt_oss_120b_can_have_a_basic_conversation.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_openai_gpt_oss_120b_can_use_tools.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_openai_gpt_oss_120b_returns_structured_output.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_openai_gpt_oss_120b_returns_the_raw_response.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_openai_gpt_oss_120b_supports_streaming_responses.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_qwen_qwen3_8_27b_can_handle_a_multi_turn_conversation.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_qwen_qwen3_8_27b_can_have_a_basic_conversation.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_qwen_qwen3_8_27b_can_use_tools.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_qwen_qwen3_8_27b_returns_structured_output.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_qwen_qwen3_8_27b_returns_the_raw_response.yml
+- spec/fixtures/vcr_cassettes/rubyllm_chat_groq_qwen_qwen3_8_27b_supports_streaming_responses.yml
+- spec/ruby_llm/chat_schema_spec.rb
+- spec/ruby_llm/chat_spec.rb
+- spec/ruby_llm/chat_streaming_spec.rb
+- spec/ruby_llm/chat_tools_spec.rb
+- spec/ruby_llm/embedding_spec.rb
+- spec/ruby_llm/image_spec.rb
+- spec/ruby_llm/models_spec.rb
+- spec/ruby_llm/moderation_spec.rb
+- spec/ruby_llm/providers/groq_spec.rb
+- spec/ruby_llm/rerank_spec.rb
+- spec/ruby_llm/speech_spec.rb
+- spec/ruby_llm/video_spec.rb
+- spec/spec_helper.rb
+- spec/support/models.rb
+- spec/support/rubyllm_configuration.rb
+- spec/support/vcr_configuration.rb
+homepage: https://github.com/ssvb/ruby_llm-providers-groq
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/ssvb/ruby_llm-providers-groq
+  source_code_uri: https://github.com/ssvb/ruby_llm-providers-groq
+  changelog_uri: https://github.com/ssvb/ruby_llm-providers-groq/releases
+  bug_tracker_uri: https://github.com/ssvb/ruby_llm-providers-groq/issues
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.3'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: RubyLLM provider for Groq.
+test_files: []