ruby_llm-template 0.1.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 75ab8a9d634742e2ce6cfd5b8af64bf28891e7ee8d7d6e49283ecc40cbe3e922
+  data.tar.gz: bf7ca2149ada527fa50bc51828c623a38d3f517648b8b548896d92b665920217
+SHA512:
+  metadata.gz: 9ca1c964c2629d14461bc13965227141ffcbcde5104fd01a988452787d6fbf52a69cea8a65d7133a38e4d8858e6f57cef206fd157f20886cb6b0c277e024e691
+  data.tar.gz: 2badb609535562c1e274ff141708f3d14cb9f8cb565f8dc00e9346e8d601f03bd0071d131eaa2b26fd1c2f67756b4b5385967b73463085e6dd3cb2b695222175

data/.rspec

@@ -0,0 +1,3 @@
+--require spec_helper
+--format documentation
+--color

data/.standard.yml

@@ -0,0 +1,3 @@
+# For available configuration options, see:
+#   https://github.com/standardrb/standard
+ruby_version: 3.1

data/CHANGELOG.md

@@ -0,0 +1,63 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [0.1.0] - 2025-01-28
+
+### Added
+- Initial release of RubyLLM::Template
+- Template management system for RubyLLM with ERB support
+- Configuration system for template directories
+- Support for system, user, assistant, and schema message templates
+- **RubyLLM::Schema Integration**: Support for `schema.rb` files using the RubyLLM::Schema DSL
+- Rails integration with automatic configuration and generator
+- Comprehensive test suite with 37 test cases
+- Error handling with descriptive messages
+- Documentation and examples
+
+### Features
+- **Template Organization**: Structure prompts in folders with separate ERB files for each message role
+- **ERB Templating**: Full ERB support with context variables and Ruby logic
+- **Schema Definition**: Use `schema.rb` files with RubyLLM::Schema DSL for type-safe, dynamic schemas
+- **Rails Integration**: Seamless Rails integration with generators and automatic configuration
+- **Configurable**: Set custom template directories per environment
+- **Schema Support**: Automatic schema loading and application with fallback to JSON
+- **Error Handling**: Clear error messages for common issues
+- **Smart Dependencies**: Optional RubyLLM::Schema dependency with graceful fallbacks
+
+### Schema Features
+- **Ruby DSL**: Use RubyLLM::Schema for clean, type-safe schema definitions
+- **Context Variables**: Access template context variables within schema.rb files
+- **Dynamic Schemas**: Generate schemas based on runtime conditions
+- **Schema-Only Approach**: Exclusively supports schema.rb files with clear error messages
+- **No JSON Fallback**: Eliminates error-prone JSON string manipulation
+
+### Usage
+```ruby
+# Basic usage with schema.rb
+RubyLLM.chat.with_template(:extract_metadata, document: @document).complete
+
+# Context variables available in both ERB and schema.rb
+RubyLLM.chat.with_template(:extract_metadata, 
+  document: @document,
+  categories: ["finance", "technology"],
+  max_items: 10
+).complete
+```
+
+### Template Structure
+```
+prompts/extract_metadata/
+├── system.txt.erb    # System message
+├── user.txt.erb      # User prompt with ERB
+├── assistant.txt.erb # Optional assistant message
+└── schema.rb         # RubyLLM::Schema definition
+```
+
+### Rails Setup
+```bash
+rails generate ruby_llm_template:install
+```

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2025 Daniel Friis
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,272 @@
+# RubyLLM::Template
+
+A flexible template management system for [RubyLLM](https://github.com/crmne/ruby_llm) that allows you to organize and reuse ERB templates.
+
+```ruby
+RubyLLM.chat.with_template(:extract_metadata, document: @document).complete
+
+# ----------------------------------
+# Retrieves the following files:
+# ----------------------------------
+# prompts/
+#   extract_metadata/
+#     ├── system.txt.erb    # System message
+#     ├── user.txt.erb      # User prompt  
+#     ├── assistant.txt.erb # Assistant message (optional)
+#     └── schema.rb         # RubyLLM::Schema definition (optional)
+
+```
+
+## Features
+
+- 🎯 **Organized Templates**: Structure your prompts in folders with separate files for system, user, assistant, and schema messages
+- 🔄 **ERB Templating**: Use full ERB power with context variables and Ruby logic
+- ⚙️ **Configurable**: Set custom template directories per environment
+- 🚀 **Rails Integration**: Seamless Rails integration with generators and automatic configuration
+- 🧪 **Well Tested**: Comprehensive test suite ensuring reliability
+- 📦 **Minimal Dependencies**: Only depends on RubyLLM and standard Ruby libraries
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'ruby_llm'
+gem 'ruby_llm-template'
+gem 'ruby_llm-schema'  # Optional for schema.rb support
+```
+
+And then execute:
+
+```bash
+bundle install
+```
+
+### Rails Setup
+
+If you're using Rails, run the generator to set up the template system:
+
+```bash
+rails generate ruby_llm_template:install
+```
+
+This will:
+- Create `config/initializers/ruby_llm_template.rb`
+- Create `app/prompts/` directory
+- Generate an example template at `app/prompts/extract_metadata/`
+
+## Quick Start
+
+### 1. Create a Template
+
+Create a directory structure like this:
+
+```
+prompts/
+  extract_metadata/
+    ├── system.txt.erb    # System message
+    ├── user.txt.erb      # User prompt  
+    ├── assistant.txt.erb # Assistant message (optional)
+    └── schema.rb         # RubyLLM::Schema definition (optional)
+```
+
+### 2. Write Your Templates
+
+**`prompts/extract_metadata/system.txt.erb`**:
+```erb
+You are an expert document analyzer. Extract metadata from documents in a structured format.
+```
+
+**`prompts/extract_metadata/user.txt.erb`**:
+```erb
+Please analyze this document: <%= document %>
+
+<% if additional_context %>
+Additional context: <%= additional_context %>
+<% end %>
+```
+
+**`prompts/extract_metadata/schema.rb`**:
+```ruby
+# Using RubyLLM::Schema DSL for clean, type-safe schemas
+RubyLLM::Schema.create do
+  string :title, description: "Document title"
+  array :topics, description: "Main topics" do
+    string
+  end
+  string :summary, description: "Brief summary"
+  
+  # Optional fields with validation
+  number :confidence, required: false, minimum: 0, maximum: 1
+  
+  # Nested objects
+  object :metadata, required: false do
+    string :author
+    string :date, format: "date"
+  end
+end
+```
+
+### 3. Use the Template
+
+```ruby
+# Basic usage
+RubyLLM.chat.with_template(:extract_metadata, document: @document).complete
+
+# With context variables
+RubyLLM.chat.with_template(:extract_metadata, 
+  document: @document,
+  additional_context: "Focus on technical details"
+).complete
+
+# Chaining with other RubyLLM methods
+RubyLLM.chat
+  .with_template(:extract_metadata, document: @document)
+  .with_model("gpt-4")
+  .complete
+```
+
+## Configuration
+
+### Non-Rails Applications
+
+```ruby
+RubyLlm::Template.configure do |config|
+  config.template_directory = "/path/to/your/prompts"
+end
+```
+
+### Rails Applications
+
+The gem automatically configures itself to use `Rails.root.join("app", "prompts")`, but you can override this in `config/initializers/ruby_llm_template.rb`:
+
+```ruby
+RubyLlm::Template.configure do |config|
+  config.template_directory = Rails.root.join("app", "ai_prompts")
+end
+```
+
+## Template Structure
+
+Each template is a directory containing ERB files for different message roles:
+
+- **`system.txt.erb`** - System message that sets the AI's behavior
+- **`user.txt.erb`** - User message/prompt  
+- **`assistant.txt.erb`** - Pre-filled assistant message (optional)
+- **`schema.rb`** - RubyLLM::Schema definition for structured output (optional)
+
+Templates are processed in order: system → user → assistant → schema
+
+## ERB Context
+
+All context variables passed to `with_template` are available in your ERB templates:
+
+```erb
+Hello <%= name %>!
+
+<% if urgent %>
+🚨 URGENT: <%= message %>
+<% else %>
+📋 Regular: <%= message %>
+<% end %>
+
+Processing <%= documents.length %> documents:
+<% documents.each_with_index do |doc, i| %>
+  <%= i + 1 %>. <%= doc.title %>
+<% end %>
+```
+
+## Advanced Usage
+
+### Complex Templates
+
+```ruby
+# Template with conditional logic and loops
+RubyLLM.chat.with_template(:analyze_reports,
+  reports: @reports,
+  priority: "high",
+  include_charts: true,
+  deadline: 1.week.from_now
+).complete
+```
+
+### Multiple Template Calls
+
+```ruby
+chat = RubyLLM.chat
+  .with_template(:initialize_session, user: current_user)
+  .with_template(:load_context, project: @project)
+  
+# Add more messages dynamically
+chat.ask("What should we focus on first?")
+```
+
+### Schema Definition with RubyLLM::Schema
+
+The gem integrates with [RubyLLM::Schema](https://github.com/danielfriis/ruby_llm-schema) to provide a clean Ruby DSL for defining JSON schemas. Use `schema.rb` files instead of JSON:
+
+```ruby
+# prompts/analyze_results/schema.rb
+RubyLLM::Schema.create do
+  number :confidence, minimum: 0, maximum: 1, description: "Analysis confidence"
+  
+  array :results, description: "Analysis results" do
+    object do
+      string :item, description: "Result item"
+      number :score, minimum: 0, maximum: 100, description: "Item score"
+      
+      # Context variables are available
+      string :category, enum: categories if defined?(categories)
+    end
+  end
+  
+  # Optional nested structures
+  object :metadata, required: false do
+    string :model_version
+    string :timestamp, format: "date-time"
+  end
+end
+```
+
+**Benefits of `schema.rb`:**
+- 🎯 **Type-safe**: Ruby DSL with built-in validation
+- 🔄 **Dynamic**: Access template context variables for conditional schemas
+- 📝 **Readable**: Clean, self-documenting Ruby syntax vs JSON
+- 🔧 **Flexible**: Generate schemas based on runtime conditions
+- 🚀 **No JSON**: Eliminate error-prone JSON string manipulation
+
+**Schema-Only Approach**: The gem exclusively supports `schema.rb` files with RubyLLM::Schema. If you have a `schema.rb` file but the gem isn't installed, you'll get a clear error message.
+
+## Error Handling
+
+The gem provides clear error messages for common issues:
+
+```ruby
+begin
+  RubyLLM.chat.with_template(:extract_metadata).complete
+rescue RubyLlm::Template::Error => e
+  puts e.message 
+  # "Template 'extract_metadata' not found in /path/to/prompts"
+  # "Schema file 'extract_metadata/schema.rb' found but RubyLLM::Schema gem is not installed"
+end
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To run the test suite:
+
+```bash
+bundle exec rspec
+```
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/danielfriis/ruby_llm-template.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+require "standard/rake"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/examples/basic_usage.rb

@@ -0,0 +1,147 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Example of using RubyLLM::Template
+# This file demonstrates basic usage without actually calling RubyLLM APIs
+
+require_relative "../lib/ruby_llm/template"
+
+# Configure template directory
+RubyLlm::Template.configure do |config|
+  config.template_directory = File.join(__dir__, "prompts")
+end
+
+# Create example prompts directory
+prompts_dir = File.join(__dir__, "prompts", "extract_metadata")
+FileUtils.mkdir_p(prompts_dir)
+
+# Create example template files
+File.write(File.join(prompts_dir, "system.txt.erb"), <<~ERB)
+  You are an expert document analyzer. Your task is to extract metadata from the provided document.
+  
+  Please analyze the document carefully and extract relevant information such as:
+  - Document type
+  - Key topics  
+  - Important dates
+  - Main entities mentioned
+  
+  Provide your analysis in a structured format.
+ERB
+
+File.write(File.join(prompts_dir, "user.txt.erb"), <<~ERB)
+  Please analyze the following document and extract its metadata:
+  
+  Document: <%= document %>
+  
+  <% if additional_context %>
+  Additional context: <%= additional_context %>
+  <% end %>
+  
+  Focus areas: <%= focus_areas.join(", ") if defined?(focus_areas) && focus_areas.any? %>
+ERB
+
+# Create schema.rb file using RubyLLM::Schema DSL
+File.write(File.join(prompts_dir, "schema.rb"), <<~RUBY)
+  # Mock RubyLLM::Schema for this example
+  module RubyLLM
+    class Schema
+      def self.create(&block)
+        instance = new
+        instance.instance_eval(&block)
+        instance
+      end
+      
+      def initialize
+        @schema = {type: "object", properties: {}, required: []}
+      end
+      
+      def string(name, **options)
+        @schema[:properties][name] = {type: "string"}.merge(options.except(:required))
+        @schema[:required] << name unless options[:required] == false
+      end
+      
+      def array(name, **options, &block)
+        prop = {type: "array"}.merge(options.except(:required))
+        prop[:items] = {type: "string"} if !block_given?
+        @schema[:properties][name] = prop
+        @schema[:required] << name unless options[:required] == false
+      end
+      
+      def to_json_schema
+        {name: "ExtractMetadataSchema", schema: @schema}
+      end
+    end
+  end
+
+  # The actual schema definition
+  RubyLLM::Schema.create do
+    string :document_type, description: "The type of document (e.g., report, article, email)"
+    
+    array :key_topics, description: "Main topics discussed in the document"
+    
+    array :important_dates, required: false, description: "Significant dates mentioned"
+    
+    # Context variables are available in schema.rb files
+    focus_count = defined?(focus_areas) ? focus_areas&.length || 3 : 3
+  end
+RUBY
+
+puts "🎯 RubyLLM::Template Example"
+puts "=" * 40
+
+# Mock chat object that demonstrates the extension
+class MockChat
+  include RubyLlm::Template::ChatExtension
+
+  def initialize
+    @messages = []
+    @schema = nil
+  end
+
+  def add_message(role:, content:)
+    @messages << {role: role, content: content}
+    puts "📝 Added #{role} message: #{content[0..100]}#{"..." if content.length > 100}"
+  end
+
+  def with_schema(schema)
+    @schema = schema
+    if schema.is_a?(Hash) && schema[:schema]
+      puts "📋 Schema applied: #{schema[:name]} with #{schema[:schema][:properties]&.keys&.length || 0} properties"
+    else
+      puts "📋 Schema applied with #{schema.keys.length} properties"
+    end
+    self
+  end
+
+  def complete
+    puts "\n🤖 Chat would now be sent to AI with:"
+    puts "   - #{@messages.length} messages"
+    puts "   - Schema: #{@schema ? "Yes" : "No"}"
+    puts "\n💬 Messages:"
+    @messages.each_with_index do |msg, i|
+      puts "   #{i + 1}. [#{msg[:role].upcase}] #{msg[:content][0..80]}#{"..." if msg[:content].length > 80}"
+    end
+    self
+  end
+end
+
+# Simulate the usage
+begin
+  chat = MockChat.new
+
+  # This demonstrates the desired API:
+  # RubyLLM.chat.with_template(:extract_metadata, context).complete
+  chat.with_template(:extract_metadata,
+    document: "Q3 Financial Report: Revenue increased 15% to $2.3M. Key challenges include supply chain delays affecting Q4 projections.",
+    additional_context: "Focus on financial metrics and future outlook",
+    focus_areas: ["revenue", "challenges", "projections"]).complete
+rescue RubyLlm::Template::Error => e
+  puts "❌ Error: #{e.message}"
+end
+
+puts "\n✅ Example completed successfully!"
+puts "\nTo use with real RubyLLM:"
+puts "  RubyLLM.chat.with_template(:extract_metadata, document: @document).complete"
+
+# Clean up example files
+FileUtils.rm_rf(File.join(__dir__, "prompts")) if Dir.exist?(File.join(__dir__, "prompts"))

data/lib/generators/ruby_llm_template/install_generator.rb

@@ -0,0 +1,117 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+
+module RubyLlmTemplate
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      desc "Install RubyLLM Template system"
+
+      def self.source_root
+        @source_root ||= File.expand_path("templates", __dir__)
+      end
+
+      def create_initializer
+        create_file "config/initializers/ruby_llm_template.rb", <<~RUBY
+          # frozen_string_literal: true
+
+          RubyLlm::Template.configure do |config|
+                      # Set the directory where your prompts are stored
+          # Default: Rails.root.join("app", "prompts")
+          # config.template_directory = Rails.root.join("app", "prompts")
+          end
+        RUBY
+      end
+
+      def create_template_directory
+        empty_directory "app/prompts"
+
+        create_file "app/prompts/.keep", ""
+
+        # Create an example template
+        create_example_template
+      end
+
+      def show_readme
+        say <<~MESSAGE
+          
+          RubyLLM Template has been installed!
+          
+          Prompts directory: app/prompts/
+          Configuration: config/initializers/ruby_llm_template.rb
+          
+          Example usage:
+            RubyLLM.chat.with_template(:extract_metadata, document: @document).complete
+          
+          Template structure:
+            app/prompts/extract_metadata/
+              ├── system.txt.erb    # System message
+              ├── user.txt.erb      # User prompt
+              ├── assistant.txt.erb # Assistant message (optional)
+              └── schema.rb         # RubyLLM::Schema definition (optional)
+          
+          Get started by creating your first template!
+        MESSAGE
+      end
+
+      private
+
+      def create_example_template
+        example_dir = "app/prompts/extract_metadata"
+        empty_directory example_dir
+
+        create_file "#{example_dir}/system.txt.erb", <<~ERB
+          You are an expert document analyzer. Your task is to extract metadata from the provided document.
+          
+          Please analyze the document carefully and extract relevant information such as:
+          - Document type
+          - Key topics
+          - Important dates
+          - Main entities mentioned
+          
+          Provide your analysis in a structured format.
+        ERB
+
+        create_file "#{example_dir}/user.txt.erb", <<~ERB
+          Please analyze the following document and extract its metadata:
+          
+          <% if defined?(document) && document %>
+          Document: <%= document %>
+          <% else %>
+          [Document content will be provided here]
+          <% end %>
+          
+          <% if defined?(additional_context) && additional_context %>
+          Additional context: <%= additional_context %>
+          <% end %>
+        ERB
+
+        create_file "#{example_dir}/schema.rb", <<~RUBY
+          # frozen_string_literal: true
+
+          # Schema definition using RubyLLM::Schema DSL
+          # See: https://github.com/danielfriis/ruby_llm-schema
+
+          RubyLLM::Schema.create do
+            string :document_type, description: "The type of document (e.g., report, article, email)"
+            
+            array :key_topics, description: "Main topics discussed in the document" do
+              string
+            end
+            
+            array :important_dates, required: false, description: "Significant dates mentioned in the document" do
+              string format: "date"
+            end
+            
+            array :entities, required: false, description: "Named entities found in the document" do
+              object do
+                string :name
+                string :type, enum: ["person", "organization", "location", "other"]
+              end
+            end
+          end
+        RUBY
+      end
+    end
+  end
+end

data/lib/ruby_llm/template/chat_extension.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module RubyLlm
+  module Template
+    module ChatExtension
+      def with_template(template_name, context = {})
+        loader = RubyLlm::Template::Loader.new(template_name)
+
+        unless loader.template_exists?
+          raise RubyLlm::Template::Error, "Template '#{template_name}' not found in #{RubyLlm::Template.configuration.template_directory}"
+        end
+
+        # Apply templates in a specific order to maintain conversation flow
+        template_order = ["system", "user", "assistant"]
+
+        template_order.each do |role|
+          next unless loader.available_roles.include?(role)
+
+          content = loader.render_template(role, context)
+          next unless content && !content.strip.empty?
+
+          add_message(role: role, content: content.strip)
+        end
+
+        # Handle schema separately if it exists
+        if loader.available_roles.include?("schema")
+          schema_result = loader.render_template("schema", context)
+
+          if schema_result
+            if schema_result.respond_to?(:to_json_schema)
+              # It's a RubyLLM::Schema instance
+              with_schema(schema_result.to_json_schema)
+            else
+              # It's a schema class
+              with_schema(schema_result)
+            end
+          end
+        end
+
+        self
+      end
+    end
+  end
+end

data/lib/ruby_llm/template/configuration.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module RubyLlm
+  module Template
+    class Configuration
+      attr_writer :template_directory
+
+      def initialize
+        @template_directory = nil
+      end
+
+      def template_directory
+        @template_directory || default_template_directory
+      end
+
+      private
+
+      def default_template_directory
+        if defined?(Rails) && Rails.respond_to?(:root) && Rails.root
+          Rails.root.join("app", "prompts")
+        else
+          File.join(Dir.pwd, "prompts")
+        end
+      end
+    end
+  end
+end

data/lib/ruby_llm/template/loader.rb

@@ -0,0 +1,146 @@
+# frozen_string_literal: true
+
+require "erb"
+require "pathname"
+
+begin
+  require "ruby_llm/schema"
+rescue LoadError
+  # RubyLLM::Schema not available, schema.rb files won't work
+end
+
+module RubyLlm
+  module Template
+    class Loader
+      SUPPORTED_ROLES = %w[system user assistant schema].freeze
+
+      def initialize(template_name, template_directory: nil)
+        @template_name = template_name.to_s
+        @template_directory = Pathname.new(template_directory || RubyLlm::Template.configuration.template_directory)
+        @template_path = @template_directory.join(@template_name)
+      end
+
+      def render_template(role, context = {})
+        return nil unless SUPPORTED_ROLES.include?(role.to_s)
+
+        # Handle schema role specially - only support .rb files
+        if role.to_s == "schema"
+          return render_schema_template(context)
+        end
+
+        # Handle regular ERB template
+        file_name = "#{role}.txt.erb"
+        template_file = @template_path.join(file_name)
+
+        return nil unless File.exist?(template_file)
+
+        template_content = File.read(template_file)
+        erb = ERB.new(template_content)
+
+        # Create a binding with the context variables
+        binding_context = create_binding_context(context)
+        erb.result(binding_context)
+      rescue => e
+        raise Error, "Failed to render template '#{@template_name}/#{file_name}': #{e.message}"
+      end
+
+      def available_roles
+        return [] unless Dir.exist?(@template_path)
+
+        roles = []
+
+        # Check for ERB templates (excluding schema.txt.erb)
+        Dir.glob("*.txt.erb", base: @template_path).each do |file|
+          role = File.basename(file, ".txt.erb")
+          next if role == "schema" # Skip schema.txt.erb files
+          roles << role if SUPPORTED_ROLES.include?(role)
+        end
+
+        # Check for schema.rb file
+        if File.exist?(@template_path.join("schema.rb"))
+          roles << "schema" unless roles.include?("schema")
+        end
+
+        roles.uniq
+      end
+
+      def template_exists?
+        Dir.exist?(@template_path) && !available_roles.empty?
+      end
+
+      def load_schema_class(context = {})
+        schema_file = @template_path.join("schema.rb")
+        return nil unless File.exist?(schema_file)
+        return nil unless defined?(RubyLLM::Schema)
+
+        schema_content = File.read(schema_file)
+        schema_context = create_schema_context(context)
+
+        # Evaluate the schema file
+        result = schema_context.instance_eval(schema_content, schema_file.to_s)
+
+        # Handle different patterns:
+        # 1. RubyLLM::Schema.create { } pattern - returns instance
+        if result.is_a?(RubyLLM::Schema) || result.respond_to?(:to_json_schema)
+          return result
+        end
+
+        # 2. Class definition pattern - look for TemplateClass::Schema
+        template_class_name = @template_name.to_s.split("_").map(&:capitalize).join
+        schema_class_name = "#{template_class_name}::Schema"
+
+        schema_class = constantize_safe(schema_class_name)
+        return schema_class if schema_class
+
+        raise Error, "Schema file must return a RubyLLM::Schema instance or define class '#{schema_class_name}'"
+      rescue => e
+        raise Error, "Failed to load schema from '#{@template_name}/schema.rb': #{e.message}"
+      end
+
+      private
+
+      def render_schema_template(context = {})
+        # Only support schema.rb files with RubyLLM::Schema
+        schema_instance = load_schema_class(context)
+        return schema_instance if schema_instance
+
+        # If there's a schema.rb file but RubyLLM::Schema isn't available, error
+        schema_file = @template_path.join("schema.rb")
+        if File.exist?(schema_file) && !defined?(RubyLLM::Schema)
+          raise Error, "Schema file '#{@template_name}/schema.rb' found but RubyLLM::Schema gem is not installed. Add 'gem \"ruby_llm-schema\"' to your Gemfile."
+        end
+
+        nil
+      end
+
+      def create_binding_context(context)
+        # Create a new binding with the context variables available
+        context.each do |key, value|
+          define_singleton_method(key) { value }
+        end
+
+        binding
+      end
+
+      def create_schema_context(context)
+        schema_context = Object.new
+        context.each do |key, value|
+          schema_context.instance_variable_set("@#{key}", value)
+          schema_context.define_singleton_method(key) { value }
+        end
+        schema_context
+      end
+
+      def constantize_safe(class_name)
+        if defined?(Rails)
+          class_name.constantize
+        else
+          # Simple constantize for non-Rails environments
+          Object.const_get(class_name)
+        end
+      rescue NameError
+        nil
+      end
+    end
+  end
+end

data/lib/ruby_llm/template/railtie.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module RubyLlm
+  module Template
+    class Railtie < Rails::Railtie
+      initializer "ruby_llm_template.configure" do |app|
+        # Set default template directory for Rails applications
+        RubyLlm::Template.configure do |config|
+          config.template_directory ||= app.root.join("app", "prompts")
+        end
+      end
+    end
+  end
+end

data/lib/ruby_llm/template/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module RubyLlm
+  module Template
+    VERSION = "0.1.2"
+  end
+end

data/lib/ruby_llm/template.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require_relative "template/version"
+require_relative "template/configuration"
+require_relative "template/loader"
+require_relative "template/chat_extension"
+
+# Load Rails integration if Rails is available
+begin
+  require "rails"
+  require_relative "template/railtie"
+rescue LoadError
+  # Rails not available
+end
+
+module RubyLlm
+  module Template
+    class Error < StandardError; end
+
+    def self.configuration
+      @configuration ||= Configuration.new
+    end
+
+    def self.configure
+      yield(configuration)
+    end
+
+    def self.reset_configuration!
+      @configuration = nil
+    end
+  end
+end
+
+# Extend RubyLLM's Chat class if it's available
+begin
+  require "ruby_llm"
+
+  if defined?(RubyLLM) && RubyLLM.respond_to?(:chat)
+    # We need to extend the actual chat class returned by RubyLLM.chat
+    # This is a monkey patch approach, but necessary for the API we want
+
+    module RubyLLMChatTemplateExtension
+      def self.extended(base)
+        base.extend(RubyLlm::Template::ChatExtension)
+      end
+    end
+
+    # Hook into RubyLLM.chat to extend the returned object
+    module RubyLLMTemplateHook
+      def chat(*args, **kwargs)
+        chat_instance = super
+        chat_instance.extend(RubyLlm::Template::ChatExtension)
+        chat_instance
+      end
+    end
+
+    if defined?(RubyLLM)
+      RubyLLM.singleton_class.prepend(RubyLLMTemplateHook)
+    end
+  end
+rescue LoadError
+  # RubyLLM not available, extension will be loaded when it becomes available
+end

data/sig/ruby_llm/template.rbs

@@ -0,0 +1,6 @@
+module RubyLlm
+  module Template
+    VERSION: String
+    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+  end
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification
+name: ruby_llm-template
+version: !ruby/object:Gem::Version
+  version: 0.1.2
+platform: ruby
+authors:
+- Daniel Friis
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2025-09-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruby_llm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: ruby_llm-schema
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.12'
+description: RubyLLM::Template provides a flexible template system for RubyLLM, allowing
+  you to organize chat prompts, system messages, and schemas in ERB template files
+  for easy reuse and maintenance.
+email:
+- d@friis.me
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".standard.yml"
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- Rakefile
+- examples/basic_usage.rb
+- lib/generators/ruby_llm_template/install_generator.rb
+- lib/ruby_llm/template.rb
+- lib/ruby_llm/template/chat_extension.rb
+- lib/ruby_llm/template/configuration.rb
+- lib/ruby_llm/template/loader.rb
+- lib/ruby_llm/template/railtie.rb
+- lib/ruby_llm/template/version.rb
+- sig/ruby_llm/template.rbs
+homepage: https://github.com/danielfriis/ruby_llm-template
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/danielfriis/ruby_llm-template
+  source_code_uri: https://github.com/danielfriis/ruby_llm-template
+  changelog_uri: https://github.com/danielfriis/ruby_llm-template/blob/main/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.1.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.19
+signing_key:
+specification_version: 4
+summary: Template management system for RubyLLM - organize and reuse ERB templates
+  for AI chat interactions
+test_files: []