8daf6cf6cb
# Pull Request Template ## Description Adds custom tool support to v1 ## Type of change - [x] New feature (non-breaking change which adds functionality) ## 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. <img width="1816" height="958" alt="CleanShot 2026-03-24 at 11 37 33@2x" src="https://github.com/user-attachments/assets/2777a953-8b65-4a2d-88ec-39f395b3fb47" /> <img width="378" height="488" alt="CleanShot 2026-03-24 at 11 38 18@2x" src="https://github.com/user-attachments/assets/f6973c99-efd0-40e4-90fe-4472a2f63cea" /> <img width="1884" height="1452" alt="CleanShot 2026-03-24 at 11 38 32@2x" src="https://github.com/user-attachments/assets/9fba4fc4-0c33-46da-888a-52ec6bad6130" /> ## Checklist: - [x] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented on my code, particularly in hard-to-understand areas - [ ] I have made corresponding changes to the documentation - [ ] My changes generate no new warnings - [ ] I have added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged and published in downstream modules --------- Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com> Co-authored-by: Shivam Mishra <scm.mymail@gmail.com>
48 lines
1.6 KiB
Ruby
48 lines
1.6 KiB
Ruby
# V1-compatible wrapper for custom HTTP tools.
|
|
#
|
|
# V2's HttpTool inherits from Agents::Tool which overrides execute(tool_context, **params),
|
|
# making it incompatible with V1's RubyLLM pipeline that calls execute(**keyword_args).
|
|
#
|
|
# This class bridges the gap: it inherits from BaseTool (RubyLLM::Tool) for V1 compatibility
|
|
# and delegates the actual HTTP execution to HttpTool#perform.
|
|
class Captain::Tools::CustomHttpTool < Captain::Tools::BaseTool
|
|
# BaseTool prepends Instrumentation, but our execute() shadows it in the MRO.
|
|
# Re-prepend so Langfuse captures tool call input/output/timing.
|
|
prepend Captain::Tools::Instrumentation
|
|
|
|
attr_reader :custom_tool
|
|
|
|
def initialize(assistant, custom_tool, conversation: nil)
|
|
@custom_tool = custom_tool
|
|
@conversation = conversation
|
|
super(assistant)
|
|
end
|
|
|
|
def active?
|
|
@custom_tool.enabled?
|
|
end
|
|
|
|
def execute(**params)
|
|
http_tool = Captain::Tools::HttpTool.new(assistant, @custom_tool)
|
|
http_tool.perform(build_tool_context, **params)
|
|
end
|
|
|
|
private
|
|
|
|
def build_tool_context
|
|
state = { account_id: assistant.account_id, assistant_id: assistant.id }
|
|
add_conversation_state(state) if @conversation
|
|
OpenStruct.new(state: state)
|
|
end
|
|
|
|
def add_conversation_state(state)
|
|
state[:conversation] = { id: @conversation.id, display_id: @conversation.display_id }
|
|
state[:contact] = slice_record_attrs(@conversation.contact, :id, :email, :phone_number)
|
|
state[:contact_inbox] = slice_record_attrs(@conversation.contact_inbox, :id, :hmac_verified)
|
|
end
|
|
|
|
def slice_record_attrs(record, *keys)
|
|
record&.attributes&.symbolize_keys&.slice(*keys)
|
|
end
|
|
end
|