lluminary 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '015093c0cffd9d6e752bfd3bd6b03b16fb3740f33f1e7b11285908a4be4f3317'
-  data.tar.gz: 4120b07419eee36bfe7d9b00cf2224ae35739ac21be9e31f3b45cb531e88b2f8
+  metadata.gz: 187dc38fab8e09fd65ffb2db359e914f16dc79ace8ff9d058de01f547db0921c
+  data.tar.gz: 6a9a08f01d36966529c3c97e1877834df7a59e41347f68ffcba4c37a56597a4e
 SHA512:
-  metadata.gz: d7424d0cdfbace6684373aa11dbc3d73d5ae2e4b7dddcb2f28eff13d5ac07e5a69672fca7da7fed0ad02be92faceb45d80bc5d339906e584e5d2732584e6a0c9
-  data.tar.gz: e529bd7a10437c74240b8849880108268c43b6ff6312c047b0b4809b75c80d0e800cbe2d7dc51b7fea09bdf6acf1e4af716defdd9937bf81cce8eb1b4b3bb63b
+  metadata.gz: 202c4b2f46a3a0f35e978406055781fe3c85d390ca5e520955b8fad80734dbe6434a9034767f149141876a28a69e50e2636c0064d1119ede7e2314df7e49bac1
+  data.tar.gz: f0c17df74edfc0db4b7eb97917355f6c32da8761fa294ddd030f86bdc0540f208ae5220700b2d1a752ad25a1665367f7bfc6f02e51df3fe76b4ed79a51bef46a

data/lib/lluminary/models/google/gemini_20_flash.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Lluminary
+  module Models
+    module Google
+      class Gemini20Flash < Lluminary::Models::Base
+        NAME = "gemini-2.0-flash"
+
+        def compatible_with?(provider_name)
+          provider_name == :google
+        end
+
+        def name
+          NAME
+        end
+      end
+    end
+  end
+end

data/lib/lluminary/providers/google.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require "openai"
+require "json"
+require_relative "../provider_error"
+
+# This is a quick and dirty implementation of a provider that works with Google's AI studio.
+# It does not currently support vertex. Plans are to eventually create a separate gem similar
+# `gemini-ai` that can work with either AI studio or Vertex. For now, this just uses the
+# OpenAI compatible endpoint.
+module Lluminary
+  module Providers
+    class Google < Base
+      NAME = :google
+      DEFAULT_MODEL = Models::Google::Gemini20Flash
+
+      attr_reader :client, :config
+
+      def initialize(**config_overrides)
+        super
+        @config = { model: DEFAULT_MODEL }.merge(config)
+        @client =
+          ::OpenAI::Client.new(
+            access_token: config[:api_key],
+            api_version: "",
+            uri_base: "https://generativelanguage.googleapis.com/v1beta/openai"
+          )
+      end
+
+      def call(prompt, _task)
+        response =
+          client.chat(
+            parameters: {
+              model: model.class::NAME,
+              messages: [{ role: "user", content: prompt }],
+              response_format: {
+                type: "json_object"
+              }
+            }
+          )
+
+        content = response.dig("choices", 0, "message", "content")
+
+        {
+          raw: content,
+          parsed:
+            begin
+              JSON.parse(content) if content
+            rescue JSON::ParserError
+              nil
+            end
+        }
+      end
+
+      def model
+        @model ||= config[:model].new
+      end
+
+      def models
+        response = @client.models.list
+        response["data"].map { |model| model["id"].split("/").last }
+      end
+    end
+  end
+end

data/lib/lluminary/task.rb

@@ -3,9 +3,6 @@ require "ostruct"
 require "json"
 require_relative "schema"
 require_relative "validation_error"
-require_relative "models/base"
-require_relative "models/openai/gpt35_turbo"
-require_relative "models/bedrock/anthropic_claude_instant_v1"
 
 module Lluminary
   # Base class for all Lluminary tasks.
@@ -25,18 +22,21 @@ module Lluminary
       def use_provider(provider_name, **config)
         provider_class =
           case provider_name
+          when :anthropic
+            require_relative "providers/anthropic"
+            Providers::Anthropic
+          when :bedrock
+            require_relative "providers/bedrock"
+            Providers::Bedrock
+          when :google
+            require_relative "providers/google"
+            Providers::Google
           when :openai
             require_relative "providers/openai"
             Providers::OpenAI
           when :test
             require_relative "providers/test"
             Providers::Test
-          when :bedrock
-            require_relative "providers/bedrock"
-            Providers::Bedrock
-          when :anthropic
-            require_relative "providers/anthropic"
-            Providers::Anthropic
           else
             raise ArgumentError, "Unknown provider: #{provider_name}"
           end

data/spec/lluminary/models/google/gemini_20_flash_spec.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+
+RSpec.describe Lluminary::Models::Google::Gemini20Flash do
+  subject(:model) { described_class.new }
+
+  describe "#NAME" do
+    it "has the correct model name" do
+      expect(described_class::NAME).to eq("gemini-2.0-flash")
+    end
+  end
+
+  describe "#compatible_with?" do
+    it "returns true for google provider" do
+      expect(model.compatible_with?(:google)).to be true
+    end
+
+    it "returns false for other providers" do
+      expect(model.compatible_with?(:openai)).to be false
+      expect(model.compatible_with?(:bedrock)).to be false
+      expect(model.compatible_with?(:anthropic)).to be false
+    end
+  end
+
+  describe "#name" do
+    it "returns the model name" do
+      expect(model.name).to eq("gemini-2.0-flash")
+    end
+  end
+end

data/spec/lluminary/providers/google_spec.rb

@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+require "spec_helper"
+
+RSpec.describe Lluminary::Providers::Google do
+  let(:config) { { api_key: "test-key" } }
+  let(:provider) { described_class.new(**config) }
+
+  describe "#client" do
+    it "returns the OpenAI client instance" do
+      expect(provider.client).to be_a(OpenAI::Client)
+    end
+  end
+
+  describe "#models" do
+    let(:mock_models_response) do
+      {
+        "object" => "list",
+        "data" => [
+          {
+            "id" => "models/gemini-2.0-flash",
+            "object" => "model",
+            "owned_by" => "google"
+          },
+          {
+            "id" => "models/gemini-1.5-pro",
+            "object" => "model",
+            "owned_by" => "google"
+          }
+        ]
+      }
+    end
+
+    before do
+      allow_any_instance_of(OpenAI::Client).to receive(:models).and_return(
+        double("ModelsClient", list: mock_models_response)
+      )
+    end
+
+    it "returns an array of model IDs as strings with the 'models/' prefix removed" do
+      expect(provider.models).to eq(%w[gemini-2.0-flash gemini-1.5-pro])
+    end
+  end
+
+  describe "#call" do
+    let(:prompt) { "Test prompt" }
+    let(:task) { "Test task" }
+    let(:mock_response) do
+      {
+        "choices" => [
+          { "message" => { "content" => '{"summary": "Test response"}' } }
+        ]
+      }
+    end
+
+    before do
+      allow_any_instance_of(OpenAI::Client).to receive(:chat).and_return(
+        mock_response
+      )
+    end
+
+    it "returns a hash with raw and parsed response" do
+      response = provider.call(prompt, task)
+      expect(response).to eq(
+        {
+          raw: '{"summary": "Test response"}',
+          parsed: {
+            "summary" => "Test response"
+          }
+        }
+      )
+    end
+
+    context "when the response is not valid JSON" do
+      let(:mock_response) do
+        { "choices" => [{ "message" => { "content" => "not valid json" } }] }
+      end
+
+      it "returns raw response with nil parsed value" do
+        response = provider.call(prompt, task)
+        expect(response).to eq({ raw: "not valid json", parsed: nil })
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lluminary
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Doug Hughes
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-05-16 00:00:00.000000000 Z
+date: 2025-05-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel
@@ -216,11 +216,13 @@ files:
 - lib/lluminary/models/bedrock/amazon_nova_pro_v1.rb
 - lib/lluminary/models/bedrock/anthropic_claude_instant_v1.rb
 - lib/lluminary/models/bedrock/base.rb
+- lib/lluminary/models/google/gemini_20_flash.rb
 - lib/lluminary/models/openai/gpt35_turbo.rb
 - lib/lluminary/provider_error.rb
 - lib/lluminary/providers/anthropic.rb
 - lib/lluminary/providers/base.rb
 - lib/lluminary/providers/bedrock.rb
+- lib/lluminary/providers/google.rb
 - lib/lluminary/providers/openai.rb
 - lib/lluminary/providers/test.rb
 - lib/lluminary/result.rb
@@ -243,9 +245,11 @@ files:
 - spec/lluminary/models/base_spec.rb
 - spec/lluminary/models/bedrock/amazon_nova_pro_v1_spec.rb
 - spec/lluminary/models/bedrock/anthropic_claude_instant_v1_spec.rb
+- spec/lluminary/models/google/gemini_20_flash_spec.rb
 - spec/lluminary/models/openai/gpt35_turbo_spec.rb
 - spec/lluminary/providers/anthropic_spec.rb
 - spec/lluminary/providers/bedrock_spec.rb
+- spec/lluminary/providers/google_spec.rb
 - spec/lluminary/providers/openai_spec.rb
 - spec/lluminary/providers/test_spec.rb
 - spec/lluminary/result_spec.rb