Module: RubyLLM::Template::ChatExtension

Defined in:
lib/ruby_llm/template/chat_extension.rb

Instance Method Summary collapse

Instance Method Details

#with_template(template_name, context = {}) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/ruby_llm/template/chat_extension.rb', line 6

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