Class: Aidp::PromptOptimization::CompositionResult

Inherits:
Object
  • Object
show all
Defined in:
lib/aidp/prompt_optimization/context_composer.rb

Overview

Result of context composition

Contains selected fragments and composition statistics

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(selected_fragments:, total_tokens:, budget:, excluded_count:, average_score:) ⇒ CompositionResult

Returns a new instance of CompositionResult.



213
214
215
216
217
218
219
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 213

def initialize(selected_fragments:, total_tokens:, budget:, excluded_count:, average_score:)
  @selected_fragments = selected_fragments
  @total_tokens = total_tokens
  @budget = budget
  @excluded_count = excluded_count
  @average_score = average_score
end

Instance Attribute Details

#average_score ⇒ Object (readonly)

Returns the value of attribute average_score.



211
212
213
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 211

def average_score
  @average_score
end

#budget ⇒ Object (readonly)

Returns the value of attribute budget.



211
212
213
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 211

def budget
  @budget
end

#excluded_count ⇒ Object (readonly)

Returns the value of attribute excluded_count.



211
212
213
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 211

def excluded_count
  @excluded_count
end

#selected_fragments ⇒ Object (readonly)

Returns the value of attribute selected_fragments.



211
212
213
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 211

def selected_fragments
  @selected_fragments
end

#total_tokens ⇒ Object (readonly)

Returns the value of attribute total_tokens.



211
212
213
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 211

def total_tokens
  @total_tokens
end

Instance Method Details

#budget_utilization ⇒ Float

Calculate budget utilization percentage

Returns:

  • (Float) —

    Percentage used (0.0-100.0)



224
225
226
227
228
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 224

def budget_utilization
  return 0.0 if @budget.zero?

  ((@total_tokens.to_f / @budget) * 100).round(2)
end

#fragments_by_type(type) ⇒ Array<Hash>

Get fragments by type

Parameters:

  • type (Symbol) —

    Fragment type (:style_guide, :template, :code)

Returns:

  • (Array<Hash>) —

    Fragments of specified type



248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 248

def fragments_by_type(type)
  @selected_fragments.select do |item|
    case type
    when :style_guide
      item[:fragment].class.name.include?("Fragment") && item[:fragment].respond_to?(:heading)
    when :project_knowledge
      item[:fragment].respond_to?(:linked_paths)
    when :template
      item[:fragment].respond_to?(:category) && !item[:fragment].respond_to?(:type)
    when :code
      item[:fragment].respond_to?(:type) && item[:fragment].respond_to?(:file_path)
    else
      false
    end
  end
end

#over_budget? ⇒ Boolean

Check if budget was exceeded

Returns:

  • (Boolean) —

    True if over budget



240
241
242
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 240

def over_budget?
  @total_tokens > @budget
end

#selected_count ⇒ Integer

Get count of selected fragments

Returns:

  • (Integer) —

    Number of selected fragments



233
234
235
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 233

def selected_count
  @selected_fragments.length
end

#summary ⇒ Hash

Get summary statistics

Returns:

  • (Hash) —

    Composition statistics



268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 268

def summary
  {
    selected_count: selected_count,
    excluded_count: @excluded_count,
    total_tokens: @total_tokens,
    budget: @budget,
    utilization: budget_utilization,
    average_score: @average_score,
    over_budget: over_budget?,
    by_type: {
      style_guide: fragments_by_type(:style_guide).count,
      project_knowledge: fragments_by_type(:project_knowledge).count,
      templates: fragments_by_type(:template).count,
      code: fragments_by_type(:code).count
    }
  }
end

#to_s ⇒ Object 



286
287
288
 
# File 'lib/aidp/prompt_optimization/context_composer.rb', line 286

def to_s
  "CompositionResult<#{selected_count} fragments, #{@total_tokens}/#{@budget} tokens (#{budget_utilization}%)>"
end