7cec4ebaae
Extract and pass image attachments from the latest user message to the runner, excluding the last user message from the context for processing. Fixes #13588 # Pull Request Template ## Description Adds image support to captain v2 ## Type of change Please delete options that are not relevant. - [x] Bug fix (non-breaking change which fixes an issue) ## How Has This Been Tested? Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration. specs and local testing <img width="754" height="1008" alt="image" src="https://github.com/user-attachments/assets/914cbc2c-9d30-42d0-87d4-9e5430845c87" /> langfuse also shows media correctly with the instrumentation code: <img width="1800" height="1260" alt="image" src="https://github.com/user-attachments/assets/ce0f5fa6-b1a5-42ec-a213-9a82b1751037" /> ## Checklist: - [x] My code follows the style guidelines of this project - [x] I have performed a self-review of my code - [x] I have commented on my code, particularly in hard-to-understand areas - [ ] I have made corresponding changes to the documentation - [x] My changes generate no new warnings - [x] I have added tests that prove my fix is effective or that my feature works - [x] New and existing unit tests pass locally with my changes - [x] Any dependent changes have been merged and published in downstream modules --------- Co-authored-by: Shivam Mishra <scm.mymail@gmail.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
54 lines
1.8 KiB
Ruby
54 lines
1.8 KiB
Ruby
module Captain::Assistant::RunnerCallbacksHelper
|
|
private
|
|
|
|
def add_callbacks_to_runner(runner)
|
|
runner = add_agent_thinking_callback(runner) if @callbacks[:on_agent_thinking]
|
|
runner = add_tool_start_callback(runner) if @callbacks[:on_tool_start]
|
|
runner = add_tool_complete_callback(runner) if @callbacks[:on_tool_complete]
|
|
runner = add_agent_handoff_callback(runner) if @callbacks[:on_agent_handoff]
|
|
runner
|
|
end
|
|
|
|
def register_trace_input_callback(runner)
|
|
runner.on_agent_thinking do |_agent_name, _input, context_wrapper|
|
|
tracing = context_wrapper&.context&.dig(:__otel_tracing)
|
|
next unless tracing
|
|
|
|
trace_input = context_wrapper.context[:captain_v2_trace_current_input]
|
|
tracing[:pending_llm_input] = trace_input if trace_input.present?
|
|
end
|
|
end
|
|
|
|
def add_agent_thinking_callback(runner)
|
|
runner.on_agent_thinking do |*args|
|
|
@callbacks[:on_agent_thinking].call(*args)
|
|
rescue StandardError => e
|
|
Rails.logger.warn "[Captain] Callback error for agent_thinking: #{e.message}"
|
|
end
|
|
end
|
|
|
|
def add_tool_start_callback(runner)
|
|
runner.on_tool_start do |*args|
|
|
@callbacks[:on_tool_start].call(*args)
|
|
rescue StandardError => e
|
|
Rails.logger.warn "[Captain] Callback error for tool_start: #{e.message}"
|
|
end
|
|
end
|
|
|
|
def add_tool_complete_callback(runner)
|
|
runner.on_tool_complete do |*args|
|
|
@callbacks[:on_tool_complete].call(*args)
|
|
rescue StandardError => e
|
|
Rails.logger.warn "[Captain] Callback error for tool_complete: #{e.message}"
|
|
end
|
|
end
|
|
|
|
def add_agent_handoff_callback(runner)
|
|
runner.on_agent_handoff do |*args|
|
|
@callbacks[:on_agent_handoff].call(*args)
|
|
rescue StandardError => e
|
|
Rails.logger.warn "[Captain] Callback error for agent_handoff: #{e.message}"
|
|
end
|
|
end
|
|
end
|