Module: Clacky::Agent::MessageCompressorHelper

Included in:
Clacky::Agent
Defined in:
lib/clacky/agent/message_compressor_helper.rb

Overview

Message compression functionality for managing conversation history Handles automatic compression when token limits are exceeded

Constant Summary collapse

COMPRESSION_THRESHOLD =

Compression thresholds

150_000
MESSAGE_COUNT_THRESHOLD =

Trigger compression when exceeding this (in tokens)

200
MAX_RECENT_MESSAGES =

Trigger compression when exceeding this (in message count)

20
TARGET_COMPRESSED_TOKENS =

Keep this many recent message pairs intact

10_000
IDLE_COMPRESSION_THRESHOLD =

Target size after compression

20_000

Instance Method Summary collapse

Instance Method Details

#build_chunk_md(messages, chunk_index:, compression_level:, topics: nil) ⇒ String

Build markdown content from a list of messages

Parameters:

  • messages (Array<Hash>) —

    Messages to render

  • chunk_index (Integer) —

    Chunk number for metadata

  • compression_level (Integer) —

    Compression level

  • topics (String, nil) (defaults to: nil) —

    Short topic description extracted from LLM summary

Returns:

  • (String) —

    Markdown content



418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 418

def build_chunk_md(messages, chunk_index:, compression_level:, topics: nil)
  lines = []

  # Front matter
  lines << "---"
  lines << "session_id: #{@session_id}"
  lines << "chunk: #{chunk_index}"
  lines << "compression_level: #{compression_level}"
  lines << "archived_at: #{Time.now.iso8601}"
  lines << "message_count: #{messages.size}"
  lines << "topics: #{topics}" if topics
  lines << "---"
  lines << ""
  lines << "# Session Chunk #{chunk_index}"
  lines << ""
  lines << "> This file contains the original conversation archived during compression."
  lines << "> Use `file_reader` to recall specific details from this conversation."
  lines << ""

  messages.each do |msg|
    role = msg[:role]
    content = msg[:content]

    case role
    when "user"
      lines << "## User"
      lines << ""
      lines << format_message_content(content)
      lines << ""
    when "assistant"
      # If this message is itself a compressed summary, annotate the header
      # so the reader knows the original conversation is in the referenced chunk
      if msg[:compressed_summary] && msg[:chunk_path]
        prev_chunk = File.basename(msg[:chunk_path])
        lines << "## Assistant [Compressed Summary — original conversation at: #{prev_chunk}]"
      else
        lines << "## Assistant"
      end
      lines << ""
      # Include tool calls summary if present
      # Format: "_Tool calls: name | {args_json}_" so replay can restore args for WebUI display.
      if msg[:tool_calls]&.any?
        tc_parts = msg[:tool_calls].map do |tc|
          name = tc.dig(:function, :name) || tc[:name] || ""
          next nil if name.empty?

          args_raw = tc.dig(:function, :arguments) || tc[:arguments] || {}
          args = args_raw.is_a?(String) ? (JSON.parse(args_raw) rescue nil) : args_raw
          if args.is_a?(Hash) && !args.empty?
            # Truncate large string values to keep chunk MD readable
            compact = args.transform_values { |v| v.is_a?(String) && v.length > 200 ? v[0..197] + "..." : v }
            "#{name} | #{compact.to_json}"
          else
            name
          end
        end.compact
        lines << "_Tool calls: #{tc_parts.join("; ")}_"
        lines << ""
      end
      lines << format_message_content(content) if content
      lines << ""
    when "tool"
      tool_name = msg[:name] || "tool"
      lines << "### Tool Result: #{tool_name}"
      lines << ""
      lines << "```"
      lines << truncate_content(content.to_s, max_length: 500)
      lines << "```"
      lines << ""
    end
  end

  lines.join("\n")
end

#calculate_target_recent_count(reduction_needed) ⇒ Object

Calculate how many recent messages to keep based on how much we need to compress



519
520
521
522
523
524
525
526
527
528
529
530
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 519

def calculate_target_recent_count(reduction_needed)
  # We want recent messages to be around 20-30% of the total target
  # This keeps the context window useful without being too large
  tokens_per_message = 500  # Average estimate for a message with content

  # Target recent messages budget (~20% of target compressed size)
  recent_budget = (TARGET_COMPRESSED_TOKENS * 0.2).to_i
  target_messages = (recent_budget / tokens_per_message).to_i

  # Clamp to reasonable bounds
  [[target_messages, 20].max, MAX_RECENT_MESSAGES].min
end

#compress_messages_if_needed(force: false) ⇒ Hash?

Check if compression is needed and return compression context

Parameters:

  • force (Boolean) (defaults to: false) —

    Force compression even if thresholds not met

Returns:

  • (Hash, nil) —

    Compression context or nil if not needed



77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 77

def compress_messages_if_needed(force: false)
  # Check if compression is enabled
  return nil unless @config.enable_compression

  # Use actual API-reported tokens from last request
  total_tokens = @previous_total_tokens
  message_count = @history.size

  # Force compression (for idle compression) - use lower threshold
  if force
    # Only compress if we have more than MAX_RECENT_MESSAGES + system message
    return nil unless message_count > MAX_RECENT_MESSAGES + 1
    # Also require minimum message count to make compression worthwhile
    return nil unless total_tokens >= IDLE_COMPRESSION_THRESHOLD
  else
    # Normal compression - check thresholds
    # Either: token count exceeds threshold OR message count exceeds threshold
    token_threshold_exceeded = total_tokens >= COMPRESSION_THRESHOLD
    message_count_exceeded = message_count >= MESSAGE_COUNT_THRESHOLD

    # Only compress if we exceed at least one threshold
    return nil unless token_threshold_exceeded || message_count_exceeded
  end

  # Calculate how much we need to reduce
  reduction_needed = total_tokens - TARGET_COMPRESSED_TOKENS

  # Don't compress if reduction is minimal (< 10% of current size)
  # Only apply this check when triggered by token threshold (not for force mode)
  if !force && token_threshold_exceeded && reduction_needed < (total_tokens * 0.1)
    return nil
  end

  # If only message count threshold is exceeded, force compression
  # to keep conversation history manageable

  # Calculate target size for recent messages based on compression level
  target_recent_count = calculate_target_recent_count(reduction_needed)

  # Increment compression level for progressive summarization
  @compression_level += 1

  # Get the most recent N messages, ensuring tool_calls/tool results pairs are kept together
  all_messages = @history.to_a
  recent_messages = get_recent_messages_with_tool_pairs(all_messages, target_recent_count)
  recent_messages = [] if recent_messages.nil?

  # Build compression instruction message (to be inserted into conversation)
  compression_message = @message_compressor.build_compression_message(all_messages, recent_messages: recent_messages)

  return nil if compression_message.nil?

  # Return compression context for agent to handle
  {
    compression_message: compression_message,
    recent_messages: recent_messages,
    original_token_count: total_tokens,
    original_message_count: @history.size,
    compression_level: @compression_level
  }
end

#empty_extraction_data ⇒ Object

Helper: empty extraction data



650
651
652
653
654
655
656
657
658
659
660
661
662
663
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 650

def empty_extraction_data
  {
    user_msgs: 0,
    assistant_msgs: 0,
    tool_msgs: 0,
    tools_used: [],
    files_created: [],
    files_modified: [],
    decisions: [],
    completed_tasks: [],
    in_progress: nil,
    shell_results: []
  }
end

#extract_decision_text(content) ⇒ Object

Helper: extract decision text from content (returns array of decisions or empty array)



623
624
625
626
627
628
629
630
631
632
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 623

def extract_decision_text(content)
  return [] unless content.is_a?(String)
  return [] unless content.include?("decision") || content.include?("chose to") || content.include?("using")

  sentences = content.split(/[.!?]/).select do |s|
    s.include?("decision") || s.include?("chose") || s.include?("using") ||
    s.include?("decided") || s.include?("will use") || s.include?("selected")
  end
  sentences.map(&:strip).map { |s| s[0..100] }
end

#extract_from_messages(messages, role_filter = nil, &block) ⇒ Object

Helper: safely extract from messages with proper nil handling



594
595
596
597
598
599
600
601
602
603
604
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 594

def extract_from_messages(messages, role_filter = nil, &block)
  return [] if messages.nil?

  results = messages
    .select { |m| role_filter.nil? || m[:role] == role_filter.to_s }
    .map(&block)
    .compact

  # Flatten if we have nested arrays (from methods returning arrays of items)
  results.any? { |r| r.is_a?(Array) } ? results.flatten.uniq : results.uniq
end

#extract_key_information(messages) ⇒ Object

Extract key information from messages for summarization



563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 563

def extract_key_information(messages)
  return empty_extraction_data if messages.nil?

  {
    # Message counts
    user_msgs: messages.count { |m| m[:role] == "user" },
    assistant_msgs: messages.count { |m| m[:role] == "assistant" },
    tool_msgs: messages.count { |m| m[:role] == "tool" },

    # Tools used
    tools_used: extract_from_messages(messages, :assistant) { |m| extract_tool_names(m[:tool_calls]) },

    # Files created/modified
    files_created: extract_from_messages(messages, :tool) { |m| filter_write_results(parse_write_result(m[:content]), :created) },
    files_modified: extract_from_messages(messages, :tool) { |m| filter_write_results(parse_write_result(m[:content]), :modified) },

    # Key decisions (limit to first 5)
    decisions: extract_from_messages(messages, :assistant) { |m| extract_decision_text(m[:content]) }.first(5),

    # Completed tasks (from TODO results)
    completed_tasks: extract_from_messages(messages, :tool) { |m| filter_todo_results(parse_todo_result(m[:content]), :completed) },

    # Current in-progress work
    in_progress: find_in_progress(messages),

    # Key results from shell commands
    shell_results: extract_from_messages(messages, :tool) { |m| parse_shell_result(m[:content]) }
  }
end

#extract_tool_names(tool_calls) ⇒ Object

Helper: extract tool names from tool_calls



607
608
609
610
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 607

def extract_tool_names(tool_calls)
  return [] unless tool_calls.is_a?(Array)
  tool_calls.map { |tc| tc.dig(:function, :name) }
end

#filter_todo_results(result, status) ⇒ Object

Helper: filter todo results by status



618
619
620
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 618

def filter_todo_results(result, status)
  result && result[:status] == status ? result[:task] : nil
end

#filter_write_results(result, action) ⇒ Object

Helper: filter write results by action



613
614
615
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 613

def filter_write_results(result, action)
  result && result[:action] == action ? result[:file] : nil
end

#find_in_progress(messages) ⇒ Object

Helper: find in-progress task



635
636
637
638
639
640
641
642
643
644
645
646
647
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 635

def find_in_progress(messages)
  return nil if messages.nil?

  messages.reverse_each do |m|
    if m[:role] == "tool"
      content = m[:content].to_s
      if content.include?("in progress") || content.include?("working on")
        return content[/[Tt]ODO[:\s]+(.+)/, 1]&.strip || content[/[Ww]orking[Oo]n[:\s]+(.+)/, 1]&.strip
      end
    end
  end
  nil
end

#format_message_content(content) ⇒ Object

Format message content (handles string or array of content blocks)



494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 494

def format_message_content(content)
  return "" if content.nil?
  return content.to_s if content.is_a?(String)

  # Handle array of content blocks (e.g., text + images)
  if content.is_a?(Array)
    content.map do |block|
      if block.is_a?(Hash) && block[:type] == "text"
        block[:text].to_s
      else
        "[#{block[:type] || 'content'}]"
      end
    end.join("\n")
  else
    content.to_s
  end
end

#generate_hierarchical_summary(messages) ⇒ Object

Generate hierarchical summary based on compression level Level 1: Detailed summary with files, decisions, features Level 2: Concise summary with key items Level 3: Minimal summary (just project type) Level 4+: Ultra-minimal (single line)



537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 537

def generate_hierarchical_summary(messages)
  level = @compression_level

  # Extract key information from messages
  extracted = extract_key_information(messages)

  summary_text = case level
  when 1
    generate_level1_summary(extracted)
  when 2
    generate_level2_summary(extracted)
  when 3
    generate_level3_summary(extracted)
  else
    generate_level4_summary(extracted)
  end

  {
    role: "user",
    content: "[SYSTEM][COMPRESSION LEVEL #{level}] #{summary_text}",
    system_injected: true,
    compression_level: level
  }
end

#generate_level1_summary(data) ⇒ Object

Level 1: Detailed summary (for first compression)



707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 707

def generate_level1_summary(data)
  parts = []

  parts << "Previous conversation summary (#{data[:user_msgs]} user requests, #{data[:assistant_msgs]} responses, #{data[:tool_msgs]} tool calls):"

  # Files created
  if data[:files_created].any?
    files_list = data[:files_created].map { |f| File.basename(f) }.join(", ")
    parts << "Created: #{files_list}"
  end

  # Files modified
  if data[:files_modified].any?
    files_list = data[:files_modified].map { |f| File.basename(f) }.join(", ")
    parts << "Modified: #{files_list}"
  end

  # Completed tasks
  if data[:completed_tasks].any?
    tasks_list = data[:completed_tasks].first(3).join(", ")
    parts << "Completed: #{tasks_list}"
  end

  # In progress
  if data[:in_progress]
    parts << "In Progress: #{data[:in_progress]}"
  end

  # Key decisions
  if data[:decisions].any?
    decisions_text = data[:decisions].map { |d| d.gsub(/\n/, " ").strip }.join("; ")
    parts << "Decisions: #{decisions_text}"
  end

  # Tools used
  if data[:tools_used].any?
    parts << "Tools: #{data[:tools_used].join(', ')}"
  end

  parts << "Continuing with recent conversation..."
  parts.join("\n")
end

#generate_level2_summary(data) ⇒ Object

Level 2: Concise summary (for second compression)



751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 751

def generate_level2_summary(data)
  parts = []

  parts << "Conversation summary:"

  # Key files (limit to most important)
  all_files = (data[:files_created] + data[:files_modified]).uniq
  if all_files.any?
    key_files = all_files.first(5).map { |f| File.basename(f) }.join(", ")
    parts << "Files: #{key_files}"
  end

  # Key accomplishments
  accomplishments = []
  accomplishments << "#{data[:completed_tasks].size} tasks completed" if data[:completed_tasks].any?
  accomplishments << "#{data[:tool_msgs]} tools executed" if data[:tool_msgs] > 0
  accomplishments << "Level #{data[:completed_tasks].size + 1} progress" if data[:in_progress]

  parts << accomplishments.join(", ") if accomplishments.any?

  parts << "Recent context follows..."
  parts.join("\n")
end

#generate_level3_summary(data) ⇒ Object

Level 3: Minimal summary (for third compression)



776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 776

def generate_level3_summary(data)
  parts = []

  parts << "Project progress:"

  # Just counts and key items
  all_files = (data[:files_created] + data[:files_modified]).uniq
  parts << "#{all_files.size} files modified, #{data[:completed_tasks].size} tasks done"

  if data[:in_progress]
    parts << "Currently: #{data[:in_progress]}"
  end

  parts << "See recent messages for details."
  parts.join("\n")
end

#generate_level4_summary(data) ⇒ Object

Level 4: Ultra-minimal summary (for fourth+ compression)



794
795
796
797
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 794

def generate_level4_summary(data)
  all_files = (data[:files_created] + data[:files_modified]).uniq
  "Progress: #{data[:completed_tasks].size} tasks, #{all_files.size} files. Recent: #{data[:tools_used].last(3).join(', ')}"
end

#get_recent_messages_with_tool_pairs(messages, count) ⇒ Array

Get recent messages while preserving tool_calls/tool_results pairs. Handles both canonical format (role: “tool”) and legacy Anthropic-native format (role: “user” with tool_result content blocks).

Parameters:

  • messages (Array) —

    All messages

  • count (Integer) —

    Target number of recent messages to keep

Returns:

  • (Array) —

    Recent messages with complete tool pairs



239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 239

def get_recent_messages_with_tool_pairs(messages, count)
  return [] if messages.nil? || messages.empty?

  messages_to_include = Set.new
  i = messages.size - 1
  messages_collected = 0

  while i >= 0 && messages_collected < count
    msg = messages[i]

    # Never include the system message — it is always prepended separately
    # by rebuild_with_compression. Including it here would cause it to appear
    # twice in the rebuilt history, inflating token counts on every compression.
    if msg[:role] == "system"
      i -= 1
      next
    end

    if messages_to_include.include?(i)
      i -= 1
      next
    end

    messages_to_include.add(i)
    messages_collected += 1

    # assistant with tool_calls → also pull in all following tool results
    if msg[:role] == "assistant" && msg[:tool_calls]&.any?
      pull_tool_results_after(messages, i, messages_to_include)
    end

    # tool result (canonical or legacy Anthropic) → also pull in its assistant
    if tool_result_message?(msg)
      pull_assistant_before(messages, i, messages_to_include) do |added|
        messages_collected += 1 if added
      end
    end

    i -= 1
  end

  recent_messages = messages_to_include.to_a.sort.map { |idx| messages[idx] }

  # Truncate large tool results to prevent token bloat
  recent_messages.map do |msg|
    truncate_tool_result(msg)
  end
end

#handle_compression_response(response, compression_context, progress: nil) ⇒ Object

Handle compression response and rebuild message list

Parameters:

  • response (Hash) —

    LLM response

  • compression_context (Hash) —

    context returned by compress_messages_if_needed

  • progress (#finish, nil) (defaults to: nil) —

    Owned progress handle from the caller’s with_progress block. When provided, the final summary message is delivered via progress.finish(final_message: …) instead of the legacy show_progress(phase: “done”) — this lets ensure in the caller guarantee cleanup even if this method raises mid-way.



147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 147

def handle_compression_response(response, compression_context, progress: nil)
  # Extract compressed content from response
  compressed_content = response[:content]

  # Note: Cost tracking is already handled by call_llm, no need to track again here

  # Rebuild message list with compression
  # Note: we need to remove the compression instruction message we just added
  original_messages = @history.to_a[0..-2]  # All except the last (compression instruction)

  # Archive compressed messages to a chunk MD file before discarding them.
  #
  # IMPORTANT: chunk_index and previous_chunks MUST come from disk, not from
  # message history. Each compression's rebuild_with_compression keeps only
  # ONE compressed_summary message (the new one), dropping older summaries
  # and embedding their references into the new summary's content. So
  # counting compressed_summary messages in history caps at 1 from the
  # second compression onward — causing chunk-2.md to be overwritten on
  # every subsequent compression, and losing references to chunk-1.md.
  #
  # Disk is the only durable source of truth: chunk files survive process
  # restarts, session reloads, and message rebuilds. SessionManager owns
  # all chunk file I/O (naming, writing, discovery) — we just ask it.
  sm = session_manager
  existing_chunks = sm.chunks_for_current(@session_id, @created_at)
  chunk_index = sm.next_chunk_index(@session_id, @created_at)

  # Extract topics from the LLM response to store in both the chunk MD front
  # matter and the compressed_summary message hash (for future chunk indexing).
  topics = @message_compressor.parse_topics(compressed_content)

  chunk_path = save_compressed_chunk(
    original_messages,
    compression_context[:recent_messages],
    chunk_index: chunk_index,
    compression_level: compression_context[:compression_level],
    topics: topics
  )

  # Build previous_chunks index from the disk-discovered chunks (already
  # sorted by index ascending). This gives the new summary a complete
  # chronological index of all older archives so the AI can recall any
  # past chunk via file_reader, not just the most recent one.
  previous_chunks = existing_chunks.map do |c|
    { basename: c[:basename], path: c[:path], topics: c[:topics] }
  end

  @history.replace_all(@message_compressor.rebuild_with_compression(
    compressed_content,
    original_messages: original_messages,
    recent_messages: compression_context[:recent_messages],
    chunk_path: chunk_path,
    topics: topics,
    previous_chunks: previous_chunks
  ))

  # Reset to the estimated size of the rebuilt (small) history.
  # The compression call_llm reported the OLD large token count, so
  # @previous_total_tokens would still be above COMPRESSION_THRESHOLD —
  # without this reset the very next think() would re-trigger compression
  # immediately, causing an infinite loop (especially after image uploads
  # where base64 data inflates token counts dramatically).
  @previous_total_tokens = @history.estimate_tokens

  # Track this compression
  @compressed_summaries << {
    level: compression_context[:compression_level],
    message_count: compression_context[:original_message_count],
    timestamp: Time.now.iso8601,
    strategy: :insert_then_compress,
    chunk_path: chunk_path
  }

  # Show compression info (use estimated tokens from rebuilt history)
  compression_summary = "History compressed (~#{compression_context[:original_token_count]} -> ~#{@history.estimate_tokens} tokens, " \
    "level #{compression_context[:compression_level]})"
  if progress
    # Owned-handle path: the caller's ensure block will still call
    # handle.finish; finishing here with a final_message means that
    # later finish (with no final_message) is a no-op (idempotent).
    progress.finish(final_message: compression_summary)
  else
    @ui&.show_progress(compression_summary, progress_type: "idle_compress", phase: "done")
  end
end

#parse_shell_result(content) ⇒ Object 



690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 690

def parse_shell_result(content)
  return nil unless content.is_a?(String)

  if content.include?("passed") || content.include?("success")
    "tests passed"
  elsif content.include?("failed") || content.include?("error")
    "command failed"
  elsif content =~ /bundle install|npm install|go mod download/
    "dependencies installed"
  elsif content.include?("Installed")
    content[/Installed:\s*(.+)/, 1]&.strip
  else
    nil
  end
end

#parse_todo_result(content) ⇒ Object 



678
679
680
681
682
683
684
685
686
687
688
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 678

def parse_todo_result(content)
  return nil unless content.is_a?(String)

  if content.include?("completed")
    { status: :completed, task: content[/completed[:\s]*(.+)/i, 1]&.strip || "task" }
  elsif content.include?("added")
    { status: :added, task: content[/added[:\s]*(.+)/i, 1]&.strip || "task" }
  else
    nil
  end
end

#parse_write_result(content) ⇒ Object 



665
666
667
668
669
670
671
672
673
674
675
676
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 665

def parse_write_result(content)
  return nil unless content.is_a?(String)

  # Check for "Created: path" or "Updated: path" patterns
  if content.include?("Created:")
    { action: :created, file: content[/Created:\s*(.+)/, 1]&.strip }
  elsif content.include?("Updated:") || content.include?("modified")
    { action: :modified, file: content[/Updated:\s*(.+)/, 1]&.strip || content[/File written to:\s*(.+)/, 1]&.strip }
  else
    nil
  end
end

#pull_assistant_before(messages, tool_result_idx, include_set) ⇒ Object

Walk backwards from tool_result_idx to find and mark its assistant message. Also marks all sibling tool results for that assistant. Yields true if the assistant was newly added (for caller to increment count).



329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 329

def pull_assistant_before(messages, tool_result_idx, include_set)
  result_ids = tool_result_ids(messages[tool_result_idx])

  j = tool_result_idx - 1
  while j >= 0
    prev = messages[j]
    if prev[:role] == "assistant" && prev[:tool_calls]&.any?
      call_ids = prev[:tool_calls].map { |tc| tc[:id] }
      if (call_ids & result_ids).any?
        newly_added = include_set.add?(j)
        yield newly_added

        # Also pull all sibling tool results for this assistant
        pull_tool_results_after(messages, j, include_set)
        break
      end
    end
    j -= 1
  end
end

#pull_tool_results_after(messages, assistant_idx, include_set) ⇒ Object

Mark all tool results immediately following messages. Stops at the first non-tool-result message.



312
313
314
315
316
317
318
319
320
321
322
323
324
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 312

def pull_tool_results_after(messages, assistant_idx, include_set)
  call_ids = messages[assistant_idx][:tool_calls].map { |tc| tc[:id] }
  j = assistant_idx + 1
  while j < messages.size
    nxt = messages[j]
    if tool_result_for?(nxt, call_ids)
      include_set.add(j)
    elsif !tool_result_message?(nxt)
      break
    end
    j += 1
  end
end

#save_compressed_chunk(original_messages, recent_messages, chunk_index:, compression_level:, topics: nil) ⇒ String?

Save the messages being compressed to a chunk MD file for future recall. The filesystem concerns (path, write, chmod) are delegated to SessionManager; this method is responsible only for the business rules of WHAT gets archived.

Parameters:

  • original_messages (Array<Hash>) —

    All messages before compression (excluding compression instruction)

  • recent_messages (Array<Hash>) —

    Recent messages being kept (to exclude from chunk)

  • chunk_index (Integer) —

    Sequential chunk number

  • compression_level (Integer) —

    Compression level

  • topics (String, nil) (defaults to: nil) —

    Short topic description for chunk index card

Returns:

  • (String, nil) —

    Path to saved chunk file, or nil if save failed



382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 382

def save_compressed_chunk(original_messages, recent_messages, chunk_index:, compression_level:, topics: nil)
  return nil unless @session_id && @created_at

  # Messages being compressed = original minus system message minus recent messages
  # Also exclude system-injected scaffolding (session context, memory prompts, etc.)
  # — these are internal CLI metadata and must not appear in chunk MD or WebUI history.
  # Also exclude previous compressed_summary messages: they are index cards pointing
  # to older chunk files and must NOT be embedded inside a new chunk, otherwise
  # parse_chunk_md_to_rounds would follow the nested reference and create circular
  # chunk chains (chunk-2 → chunk-1 → ... → chunk-2).
  recent_set = recent_messages.to_a
  messages_to_archive = original_messages.reject do |m|
    m[:role] == "system" || m[:system_injected] || m[:compressed_summary] || recent_set.include?(m)
  end

  return nil if messages_to_archive.empty?

  md_content = build_chunk_md(messages_to_archive,
                              chunk_index: chunk_index,
                              compression_level: compression_level,
                              topics: topics)

  # Delegate filesystem concerns (path assembly, write, chmod) to SessionManager —
  # it owns the on-disk layout for sessions and their chunk archives.
  session_manager.write_chunk(@session_id, @created_at, chunk_index, md_content)
rescue => e
  @ui&.log("Failed to save chunk MD: #{e.message}", level: :warn)
  nil
end

#tool_result_for?(msg, call_ids) ⇒ Boolean

Returns true if msg is a tool result that matches any of the given call IDs.

Returns:

  • (Boolean) 


306
307
308
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 306

def tool_result_for?(msg, call_ids)
  tool_result_message?(msg) && (tool_result_ids(msg) & call_ids).any?
end

#tool_result_ids(msg) ⇒ Object

Returns the tool_call IDs referenced in a tool result message.



297
298
299
300
301
302
303
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 297

def tool_result_ids(msg)
  if MessageFormat::OpenAI.tool_result_message?(msg)
    MessageFormat::OpenAI.tool_call_ids(msg)
  else
    MessageFormat::Anthropic.tool_use_ids(msg)
  end
end

#tool_result_message?(msg) ⇒ Boolean

Returns true if msg is a tool result, regardless of storage format. Canonical: role:“tool” | Legacy Anthropic-native: role:“user” + tool_result blocks

Returns:

  • (Boolean) 


291
292
293
294
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 291

def tool_result_message?(msg)
  MessageFormat::OpenAI.tool_result_message?(msg) ||
    MessageFormat::Anthropic.tool_result_message?(msg)
end

#trigger_idle_compression ⇒ Object

Trigger compression during idle time (user-friendly, interruptible) Returns true if compression was performed, false otherwise



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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 17

def trigger_idle_compression
  # Check if we should compress (force mode) BEFORE opening any UI, so
  # "skipped" doesn't flash a spinner on screen.
  compression_context = compress_messages_if_needed(force: true)
  if compression_context.nil?
    Clacky::Logger.info(
      "Idle compression skipped",
      enable_compression: @config.enable_compression,
      previous_total_tokens: @previous_total_tokens,
      history_size: @history.size,
      idle_threshold: IDLE_COMPRESSION_THRESHOLD,
      max_recent_messages: MAX_RECENT_MESSAGES
    )
    return false
  end

  # Own the progress indicator through +with_progress+: the ensure
  # block guarantees the spinner/ticker is released even when the
  # user interrupts mid-way (AgentInterrupted from current thread)
  # or the LLM call fails. No more orphan gray tickers.
  #
  # When @ui is nil (tests / headless) we still need to run the
  # compression work — safe-navigation with a block would silently
  # skip it, so branch explicitly.
  compression_message = compression_context[:compression_message]
  @history.append(compression_message)

  run_compression = lambda do |handle|
    begin
      response = call_llm
      handle_compression_response(response, compression_context, progress: handle)
      true
    rescue Clacky::AgentInterrupted => e
      @ui&.log("Idle compression canceled: #{e.message}", level: :info)
      @history.rollback_before(compression_message)
      false
    rescue => e
      @ui&.log("Idle compression failed: #{e.message}", level: :error)
      @history.rollback_before(compression_message)
      false
    end
  end

  if @ui
    result = nil
    @ui.with_progress(
      message: "Idle detected. Compressing conversation to optimize costs...",
      style: :quiet
    ) do |handle|
      result = run_compression.call(handle)
    end
    result
  else
    run_compression.call(nil)
  end
end

#truncate_content(text, max_length: 500) ⇒ Object

Truncate long content with a note



513
514
515
516
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 513

def truncate_content(text, max_length: 500)
  return text if text.length <= max_length
  "#{text[0...max_length]}\n... [truncated, #{text.length} chars total]"
end

#truncate_tool_result(msg) ⇒ Object

Truncate oversized tool result content to avoid token bloat.



351
352
353
354
355
356
357
358
 
# File 'lib/clacky/agent/message_compressor_helper.rb', line 351

def truncate_tool_result(msg)
  if MessageFormat::OpenAI.tool_result_message?(msg) &&
      msg[:content].is_a?(String) && msg[:content].length > 2000
    msg.merge(content: msg[:content][0..2000] + "...\n[Content truncated - exceeded 2000 characters]")
  else
    msg
  end
end