Module: Legion::Extensions::Planner::Runners::Planner

Extended by:
Planner
Included in:
Planner
Defined in:
lib/legion/extensions/planner/runners/planner.rb

Instance Method Summary collapse

Instance Method Details

#gather_context(results: nil, work_item: nil, args: nil) ⇒ Object

Raises:

  • (ArgumentError) 


50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/legion/extensions/planner/runners/planner.rb', line 50

def gather_context(results: nil, work_item: nil, args: nil, **)
  results = json_load(results) if results.is_a?(String)
  work_item ||= results&.dig(:work_item) || args&.dig(:work_item)
  raise ArgumentError, "work_item is nil in #{__method__}" if work_item.nil?

  context = Helpers::ContextGatherer.gather(
    repo:         work_item[:repo],
    config:       work_item[:config][:context] || {},
    work_item_id: work_item[:work_item_id],
    keywords:     extract_keywords(work_item)
  )

  { success: true, context: context, context_ref: context[:context_ref] }
end

#plan(results: nil, work_item: nil, args: nil) ⇒ Object

Raises:

  • (ArgumentError) 


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
42
43
44
45
46
47
48
 
# File 'lib/legion/extensions/planner/runners/planner.rb', line 10

def plan(results: nil, work_item: nil, args: nil, **)
  results = json_load(results) if results.is_a?(String)
  work_item ||= results&.dig(:work_item) || args&.dig(:work_item)
  raise ArgumentError, "work_item is nil in #{__method__}" if work_item.nil?

  max_iterations = work_item.dig(:config, :planning, :max_iterations) ||
                   Legion::Settings.dig(:fleet, :planning, :max_iterations) || 5

  context = Helpers::ContextGatherer.gather(
    repo:         work_item[:repo],
    config:       work_item[:config][:context] || {},
    work_item_id: work_item[:work_item_id],
    keywords:     extract_keywords(work_item)
  )

  plan_result = nil
  validation = nil
  max_iterations.times do
    plan_result = generate_plan(work_item: work_item, context: context)
    validation = Helpers::PlanSchema.validate_plan(plan: plan_result)
    break if validation[:valid]
  end

  unless validation[:valid]
    return { success: false, reason: :invalid_plan, errors: validation[:errors],
             work_item: work_item }
  end

  work_item = work_item.merge(
    pipeline: work_item[:pipeline].merge(
      stage:       'planned',
      plan:        plan_result,
      context_ref: context[:context_ref],
      trace:       work_item[:pipeline][:trace] + [build_trace_entry]
    )
  )

  { success: true, work_item: work_item }
end