lluminary 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cacc0edbbc80b69f4c95d6b9888707651dffc5189432920b1c814abed095ea01
-  data.tar.gz: 647f143f0be6053b7ca80868e2d6bd5058cfee8fae0d0e12a51ad59f8e8d5233
+  metadata.gz: '015093c0cffd9d6e752bfd3bd6b03b16fb3740f33f1e7b11285908a4be4f3317'
+  data.tar.gz: 4120b07419eee36bfe7d9b00cf2224ae35739ac21be9e31f3b45cb531e88b2f8
 SHA512:
-  metadata.gz: 63dce0789e7b175ad324d7de8f7bbd4c90da575e314f14cbd3c4ab9c6b3cab5b942cd279a15b545ac322f267bda221efd3988c6b7e9c7f2f930f1de0c1af87ac
-  data.tar.gz: 30a523b37b850f6603e466e05c49ceb8eeb03fb8ee81c3b237b97b175ace6082e0e97ace99b210e2807fb1d9feab2520a4b95753d5690e7fb46d02df1faa0389
+  metadata.gz: d7424d0cdfbace6684373aa11dbc3d73d5ae2e4b7dddcb2f28eff13d5ac07e5a69672fca7da7fed0ad02be92faceb45d80bc5d339906e584e5d2732584e6a0c9
+  data.tar.gz: e529bd7a10437c74240b8849880108268c43b6ff6312c047b0b4809b75c80d0e800cbe2d7dc51b7fea09bdf6acf1e4af716defdd9937bf81cce8eb1b4b3bb63b

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

@@ -34,6 +34,9 @@ module Lluminary
           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/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/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/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.2
 platform: ruby
 authors:
 - Doug Hughes
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-05-04 00:00:00.000000000 Z
+date: 2025-05-16 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,12 +211,14 @@ 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/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/openai.rb
@@ -211,8 +227,6 @@ files:
 - 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
@@ -230,6 +244,7 @@ files:
 - spec/lluminary/models/bedrock/amazon_nova_pro_v1_spec.rb
 - spec/lluminary/models/bedrock/anthropic_claude_instant_v1_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/openai_spec.rb
 - spec/lluminary/providers/test_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