lluminary 0.1.2 → 0.1.3

data/lib/lluminary/task.rb

@@ -1,20 +1,15 @@
 # frozen_string_literal: true
 require "ostruct"
+require "json"
 require_relative "schema"
 require_relative "validation_error"
-require_relative "field_description"
-require "json"
+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.
   # Provides the core functionality for defining and running LLM-powered tasks.
-  #
-  # @example Creating a custom task
-  #   class MyTask < Lluminary::Task
-  #     def run
-  #       # Task implementation
-  #     end
-  #   end
   class Task
     class << self
       def input_schema(&block)
@@ -43,11 +38,7 @@ module Lluminary
             raise ArgumentError, "Unknown provider: #{provider_name}"
           end
 
-        # Merge global config with task-specific config
-        global_config = Lluminary.config.provider_config(provider_name)
-        merged_config = global_config.merge(config)
-
-        @provider = provider_class.new(**merged_config)
+        @provider = provider_class.new(**config)
       end
 
       def call(input = {})
@@ -122,11 +113,11 @@ module Lluminary
     end
 
     def prompt
-      <<~PROMPT
-        #{task_prompt}
+      @prompt ||= self.class.provider.model.format_prompt(self)
+    end
 
-        #{json_schema_example}
-      PROMPT
+    def task_prompt
+      raise NotImplementedError, "Subclasses must implement task_prompt"
     end
 
     private
@@ -172,7 +163,7 @@ module Lluminary
       # Validate after merging
       @output.valid?
 
-      @prompt = prompt
+      prompt
     end
 
     def define_input_methods
@@ -181,68 +172,11 @@ module Lluminary
       end
     end
 
-    def task_prompt
-      raise NotImplementedError, "Subclasses must implement task_prompt"
-    end
-
-    def json_schema_example
-      return "{}" if fields.empty?
-
-      <<~SCHEMA.chomp
-        You must respond with ONLY a valid JSON object. Do not include any other text, explanations, or formatting.
-        The JSON object must contain the following fields:
-
-        #{generate_field_descriptions}
-
-        Your response must be ONLY this JSON object:
-        #{example_json}
-      SCHEMA
-    end
-
-    def fields
-      @fields ||= self.class.output_fields
-    end
-
-    def generate_field_descriptions
-      fields
-        .map do |name, field|
-          # Get validations for this field
-          validations =
-            self
-              .class
-              .instance_variable_get(:@output_schema)
-              &.validations_for(name) || []
-          field_with_validations = field.merge(validations: validations)
-          FieldDescription.new(name, field_with_validations).to_schema_s
-        end
-        .join("\n\n")
-    end
-
-    def example_json
-      json =
-        fields.each_with_object({}) do |(name, field), hash|
-          hash[name] = case field[:type]
-          when :string
-            "your #{name} here"
-          when :integer
-            0
-          when :datetime
-            "2024-01-01T12:00:00+00:00"
-          when :boolean
-            true
-          when :float
-            0.0
-          end
-        end
-
-      JSON.pretty_generate(json)
-    end
-
     def to_result
       Result.new(
         raw_response: @output&.raw_response,
         output: @parsed_response,
-        prompt: @prompt
+        prompt: prompt
       )
     end
   end

data/lib/lluminary.rb

@@ -3,8 +3,10 @@
 require_relative "lluminary/version"
 require_relative "lluminary/result"
 require_relative "lluminary/task"
-require_relative "lluminary/providers/base"
-require_relative "lluminary/providers/openai"
+# automatically require all providers
+Dir[File.join(__dir__, "lluminary/providers/*.rb")].each { |file| require file }
+# automatically require all models
+Dir[File.join(__dir__, "lluminary/models/**/*.rb")].each { |file| require file }
 require_relative "lluminary/config"
 
 # Lluminary is a framework for building and running LLM-powered tasks.

data/spec/examples/meal_suggester_spec.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+require "spec_helper"
+require_relative "../../examples/meal_suggester"
+
+RSpec.describe MealSuggester do
+  let(:valid_ingredients) { %w[eggs bread butter] }
+  let(:valid_count) { 3 }
+  let(:valid_params) do
+    { ingredients: valid_ingredients, suggestions_count: valid_count }
+  end
+
+  describe "input validation" do
+    it "accepts valid parameters" do
+      expect { described_class.call!(**valid_params) }.not_to raise_error
+    end
+
+    it "requires ingredients to be present" do
+      expect do
+        described_class.call!(ingredients: [], suggestions_count: valid_count)
+      end.to raise_error(Lluminary::ValidationError)
+    end
+
+    it "requires suggestions_count to be present" do
+      expect do
+        described_class.call!(
+          ingredients: valid_ingredients,
+          suggestions_count: nil
+        )
+      end.to raise_error(Lluminary::ValidationError)
+    end
+  end
+
+  describe "output validation" do
+    let(:result) { described_class.call(**valid_params) }
+
+    it "returns an array of meal suggestions" do
+      expect(result.output.meal_suggestions).to be_an(Array)
+    end
+
+    it "returns the requested number of suggestions" do
+      expect(result.output.meal_suggestions.length).to eq(valid_count)
+    end
+
+    it "returns string suggestions" do
+      expect(result.output.meal_suggestions).to all(be_a(String))
+    end
+
+    it "returns non-empty suggestions" do
+      expect(result.output.meal_suggestions).to all(be_present)
+    end
+  end
+
+  describe "prompt generation" do
+    let(:result) { described_class.call(**valid_params) }
+
+    it "includes the ingredients in the prompt" do
+      expect(result.prompt).to include(valid_ingredients.inspect)
+    end
+
+    it "includes the suggestions count in the prompt" do
+      expect(result.prompt).to include(valid_count.to_s)
+    end
+  end
+end