Class: RubyLLM::Contract::Pipeline::Base

Inherits:
Object
  • Object
show all
Extended by:
Concerns::EvalHost
Defined in:
lib/ruby_llm/contract/pipeline/base.rb

Class Method Summary collapse

Methods included from Concerns::EvalHost

clear_file_sourced_evals!, compare_models, define_eval, eval_defined?, eval_names, run_eval

Class Method Details

.inherited(subclass) ⇒ Object 



7
8
9
10
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 7

def self.inherited(subclass)
  super
  Contract.register_eval_host(subclass) if respond_to?(:eval_defined?) && eval_defined?
end

.run(input, context: {}, timeout_ms: nil) ⇒ Object 



37
38
39
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 37

def run(input, context: {}, timeout_ms: nil)
  Runner.new(steps: steps, context: context, timeout_ms: timeout_ms, token_budget: token_budget).call(input)
end

.step(step_class, as:, depends_on: nil, model: nil) ⇒ Object

depends_on is accepted for forward compatibility with DAG pipelines (v0.3). Currently, execution is always linear in declaration order.



17
18
19
20
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 17

def step(step_class, as:, depends_on: nil, model: nil)
  validate_dependency!(depends_on) if depends_on
  steps_registry << { step_class: step_class, alias: as, depends_on: depends_on, model: model }
end

.steps ⇒ Object 



22
23
24
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 22

def steps
  steps_registry.map { |s| s.dup.freeze }.freeze
end

.steps_registry ⇒ Object

Internal mutable steps list for registration



27
28
29
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 27

def steps_registry
  @steps_registry ||= []
end

.test(input, responses: {}, timeout_ms: nil) ⇒ Object 



41
42
43
44
45
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 41

def test(input, responses: {}, timeout_ms: nil)
  ordered_responses = steps.map { |step_entry| responses.fetch(step_entry[:alias], "") }
  adapter = Adapters::Test.new(responses: ordered_responses)
  run(input, context: { adapter: adapter }, timeout_ms: timeout_ms)
end

.token_budget(limit = nil) ⇒ Object 



31
32
33
34
35
 
# File 'lib/ruby_llm/contract/pipeline/base.rb', line 31

def token_budget(limit = nil)
  return @token_budget = limit if limit

  @token_budget
end