lluminary 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cacc0edbbc80b69f4c95d6b9888707651dffc5189432920b1c814abed095ea01
-  data.tar.gz: 647f143f0be6053b7ca80868e2d6bd5058cfee8fae0d0e12a51ad59f8e8d5233
+  metadata.gz: 187dc38fab8e09fd65ffb2db359e914f16dc79ace8ff9d058de01f547db0921c
+  data.tar.gz: 6a9a08f01d36966529c3c97e1877834df7a59e41347f68ffcba4c37a56597a4e
 SHA512:
-  metadata.gz: 63dce0789e7b175ad324d7de8f7bbd4c90da575e314f14cbd3c4ab9c6b3cab5b942cd279a15b545ac322f267bda221efd3988c6b7e9c7f2f930f1de0c1af87ac
-  data.tar.gz: 30a523b37b850f6603e466e05c49ceb8eeb03fb8ee81c3b237b97b175ace6082e0e97ace99b210e2807fb1d9feab2520a4b95753d5690e7fb46d02df1faa0389
+  metadata.gz: 202c4b2f46a3a0f35e978406055781fe3c85d390ca5e520955b8fad80734dbe6434a9034767f149141876a28a69e50e2636c0064d1119ede7e2314df7e49bac1
+  data.tar.gz: f0c17df74edfc0db4b7eb97917355f6c32da8761fa294ddd030f86bdc0540f208ae5220700b2d1a752ad25a1665367f7bfc6f02e51df3fe76b4ed79a51bef46a

data/lib/lluminary/models/anthropic/claude_3_5_sonnet.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# require_relative "../base"
+
+module Lluminary
+  module Models
+    module Anthropic
+      class Claude35Sonnet < Lluminary::Models::Base
+        NAME = "claude-3-5-sonnet-latest"
+
+        def compatible_with?(provider_name)
+          provider_name == :anthropic
+        end
+      end
+    end
+  end
+end

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/anthropic.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require "anthropic"
+require "json"
+require_relative "../provider_error"
+
+module Lluminary
+  module Providers
+    # Provider for Anthropic's models.
+    # Implements the Base provider interface for Anthropic's API.
+    class Anthropic < Base
+      NAME = :anthropic
+      DEFAULT_MODEL = Models::Anthropic::Claude35Sonnet
+
+      attr_reader :client, :config
+
+      def initialize(**config_overrides)
+        super
+        @config = { model: DEFAULT_MODEL }.merge(config)
+        @client = ::Anthropic::Client.new(api_key: config[:api_key])
+      end
+
+      def call(prompt, _task)
+        message =
+          client.messages.create(
+            max_tokens: 1024, # TODO: make this configurable
+            messages: [{ role: "user", content: prompt }],
+            model: model.class::NAME
+          )
+
+        content = message.content.first.text
+
+        {
+          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 }
+      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,15 +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
           else
             raise ArgumentError, "Unknown provider: #{provider_name}"
           end

data/lib/lluminary.rb

@@ -3,10 +3,16 @@
 require_relative "lluminary/version"
 require_relative "lluminary/result"
 require_relative "lluminary/task"
-# automatically require all providers
-Dir[File.join(__dir__, "lluminary/providers/*.rb")].each { |file| require file }
-# automatically require all models
+# require base model first
+require_relative "lluminary/models/base"
+# automatically require all models first
 Dir[File.join(__dir__, "lluminary/models/**/*.rb")].each { |file| require file }
+# require base provider first
+require_relative "lluminary/providers/base"
+# then require all other providers
+Dir[File.join(__dir__, "lluminary/providers/*.rb")].each do |file|
+  require file unless file.end_with?("base.rb")
+end
 require_relative "lluminary/config"
 
 # Lluminary is a framework for building and running LLM-powered tasks.

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/anthropic_spec.rb

@@ -0,0 +1,104 @@
+# frozen_string_literal: true
+require "spec_helper"
+require "lluminary/providers/anthropic"
+
+RSpec.describe Lluminary::Providers::Anthropic do
+  let(:config) { { api_key: "test-key" } }
+  let(:provider) { described_class.new(**config) }
+
+  describe "#client" do
+    it "returns the Anthropic client instance" do
+      expect(provider.client).to be_a(Anthropic::Client)
+    end
+  end
+
+  describe "#models" do
+    let(:mock_models_response) do
+      mock_model_info_1 = double("ModelInfo", id: "claude-3-5-sonnet-latest")
+      mock_model_info_2 = double("ModelInfo", id: "claude-3-haiku-20240307")
+
+      double(
+        "Page",
+        data: [mock_model_info_1, mock_model_info_2],
+        has_more: false,
+        first_id: "claude-3-5-sonnet-latest",
+        last_id: "claude-3-haiku-20240307"
+      )
+    end
+
+    before do
+      models_client = double("ModelsClient")
+      allow_any_instance_of(Anthropic::Client).to receive(:models).and_return(
+        models_client
+      )
+      allow(models_client).to receive(:list).and_return(mock_models_response)
+    end
+
+    it "returns an array of model IDs as strings" do
+      expect(provider.models).to eq(
+        %w[claude-3-5-sonnet-latest claude-3-haiku-20240307]
+      )
+    end
+  end
+
+  describe "#call" do
+    let(:prompt) { "Test prompt" }
+    let(:task) { "Test task" }
+    let(:mock_response) do
+      OpenStruct.new(
+        content: [OpenStruct.new(text: '{"summary": "Test response"}')]
+      )
+    end
+
+    before do
+      messages_client = double("MessagesClient")
+      allow_any_instance_of(Anthropic::Client).to receive(:messages).and_return(
+        messages_client
+      )
+      allow(messages_client).to receive(:create).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
+        OpenStruct.new(content: [OpenStruct.new(text: "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
+
+  describe "#model" do
+    it "returns the default model when not specified" do
+      expect(provider.model).to be_a(
+        Lluminary::Models::Anthropic::Claude35Sonnet
+      )
+    end
+
+    it "returns the specified model when provided in config" do
+      model_class = double("ModelClass")
+      model_instance = double("ModelInstance")
+
+      allow(model_class).to receive(:new).and_return(model_instance)
+
+      custom_provider =
+        described_class.new(model: model_class, api_key: "test-key")
+
+      expect(custom_provider.model).to eq(model_instance)
+    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

data/spec/lluminary/task_spec.rb

@@ -19,6 +19,9 @@ RSpec.describe Lluminary::Task do
   end
 
   let(:task_with_test) { Class.new(described_class) { use_provider :test } }
+  let(:task_with_anthropic) do
+    Class.new(described_class) { use_provider :anthropic, api_key: "test" }
+  end
 
   describe ".call" do
     it "returns a result with a raw response from the provider" do
@@ -44,6 +47,12 @@ RSpec.describe Lluminary::Task do
       task_class.provider = custom_provider
       expect(task_class.provider).to eq(custom_provider)
     end
+
+    it "with :anthropic provider sets the Anthropic provider" do
+      expect(task_with_anthropic.provider).to be_a(
+        Lluminary::Providers::Anthropic
+      )
+    end
   end
 
   describe ".use_provider" do
@@ -76,6 +85,15 @@ RSpec.describe Lluminary::Task do
       )
     end
 
+    it "with :anthropic instantiates Anthropic provider with config" do
+      task_class.use_provider(:anthropic, api_key: "test")
+      expect(task_class.provider).to be_a(Lluminary::Providers::Anthropic)
+      expect(task_class.provider.config).to include(
+        api_key: "test",
+        model: Lluminary::Models::Anthropic::Claude35Sonnet
+      )
+    end
+
     it "raises ArgumentError for unknown provider" do
       expect { task_class.use_provider(:unknown) }.to raise_error(
         ArgumentError,

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lluminary
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Doug Hughes
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-05-04 00:00:00.000000000 Z
+date: 2025-05-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel
@@ -30,6 +30,20 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '9'
+- !ruby/object:Gem::Dependency
+  name: anthropic-sdk-beta
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0.pre.beta.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0.pre.beta.6
 - !ruby/object:Gem::Dependency
   name: aws-sdk-bedrock
   requirement: !ruby/object:Gem::Requirement
@@ -157,33 +171,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '1.50'
 - !ruby/object:Gem::Dependency
-  name: syntax_tree
+  name: simplecov
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.2'
+        version: 0.22.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.2'
+        version: 0.22.0
 - !ruby/object:Gem::Dependency
-  name: simplecov
+  name: syntax_tree
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.22.0
+        version: '6.2'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.22.0
+        version: '6.2'
 description: 'Lluminary is a framework for building applications that leverage Large
   Language Models. It provides a structured way to define tasks, manage prompts, and
   handle LLM interactions.
@@ -197,22 +211,24 @@ extra_rdoc_files: []
 files:
 - lib/lluminary.rb
 - lib/lluminary/config.rb
+- lib/lluminary/models/anthropic/claude_3_5_sonnet.rb
 - lib/lluminary/models/base.rb
 - 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
 - lib/lluminary/schema.rb
 - lib/lluminary/schema_model.rb
 - lib/lluminary/task.rb
-- lib/lluminary/tasks/describe_openai_model.rb
-- lib/lluminary/tasks/identify_and_describe_open_ai_models.rb
 - lib/lluminary/validation_error.rb
 - lib/lluminary/version.rb
 - spec/examples/analyze_text_spec.rb
@@ -229,8 +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

data/lib/lluminary/tasks/describe_openai_model.rb

@@ -1,61 +0,0 @@
-# frozen_string_literal: true
-
-module Lluminary
-  module Tasks
-    class DescribeOpenAiModel < Lluminary::Task
-      use_provider :openai
-
-      input_schema do
-        string :model, description: "The OpenAI model to describe"
-      end
-
-      # {
-      #   "id": "gpt-4o-2024-11-20",
-      #   "family": "gpt-4o",
-      #   "variant": "standard",
-      #   "release_date": "2024-11-20",
-      #   "status": "GA",
-      #   "inputs": {"text": true, "image": true, "audio": false},
-      #   "outputs": {"text": true, "audio": false}
-      # }
-
-      output_schema do
-        hash :model_description, description: "The description of the model" do
-          string :id,
-                 description:
-                   "The full OpenAI API model ID being described. EG: 'gpt-4o-2024-11-20'"
-          string :family,
-                 description:
-                   "The OpenAI model family. EG: 'gpt-4o' or 'gpt-4.1-mini'"
-          string :variant, description: "The OpenAI model variant"
-          string :release_date,
-                 description: "The model's release date, if known."
-          string :status,
-                 description: "The OpenAI model status. EG: GA or preview"
-          hash :inputs, description: "The model's inputs" do
-            boolean :text, description: "Whether the model can process text"
-            boolean :image, description: "Whether the model can process images"
-            boolean :audio, description: "Whether the model can process audio"
-            string :other_inputs,
-                   description: "Other inputs the model can process"
-          end
-          hash :outputs, description: "The model's outputs" do
-            boolean :text, description: "Whether the model can output text"
-            boolean :image, description: "Whether the model can output images"
-            boolean :audio, description: "Whether the model can output audio"
-            string :other_outputs,
-                   description: "Other outputs the model can return"
-          end
-        end
-      end
-
-      def task_prompt
-        <<~PROMPT
-          You are an expert in OpenAI models. You will be given a model ID and asked to describe the model using structured data.
-
-          Model ID: #{input.model}
-        PROMPT
-      end
-    end
-  end
-end

data/lib/lluminary/tasks/identify_and_describe_open_ai_models.rb

@@ -1,51 +0,0 @@
-# frozen_string_literal: true
-
-module Lluminary
-  module Tasks
-    class IdentifyAndDescribeOpenAiModels < Lluminary::Task
-      use_provider :bedrock, model: Lluminary::Models::Bedrock::AmazonNovaProV1
-
-      input_schema do
-        array :models, description: "List of OpenAI models" do
-          string
-        end
-      end
-
-      output_schema do
-        array :root_models,
-              description: "List of root models and their versions" do
-          hash do
-            string :name,
-                   description:
-                     "The root name of the model. For example, 'gpt-4' or 'gpt-4o'"
-            array :versions,
-                  description:
-                    "List of versions of the root model. For example, '0125-preview' or '0613' or '2024-04-09'" do
-              string
-            end
-          end
-        end
-      end
-
-      def task_prompt
-        <<~PROMPT
-          You are an expert in OpenAI models. You will be given a list of OpenAI models and asked to group them together by the "root" model type and capability and list the various versions of the root model.
-
-          Keep in mind that some "root" models have names with the same root name but different capabilities. For example, "gpt-4o" and "gpt-4o-audio" are distinct models, since they have different capabilities and each has their own versions.
-          
-          "gpt-4.5-preview" and "gpt-4.5-preview-2025-02-27" are examples of the "gpt-4.5" root model. There are two versions of the "gpt-4.5" root model: "preview" and "preview-2025-02-27".
-
-          Given the following list of models, please group them together by the "root" model type and list their versions. 
-
-          Your response will be used to generate code that will make use of the models and their verisons. 
-
-          It's critical that you represent every model and version from the following list in your response. Any model or version that is missed will be excluded from subsequent code generation and that will make them very, very sad. We don't want any sad models.
-
-          DO NOT include any other models or versions in your response other than those from ones listed below. Use your expertise in OpenAI models to distinguish between different "root" models and their versions.
-
-          Models: #{models.join(", ")}
-        PROMPT
-      end
-    end
-  end
-end