diff --git a/enterprise/app/models/concerns/agentable.rb b/enterprise/app/models/concerns/agentable.rb index 20bbbfb78..a7b5d940f 100644 --- a/enterprise/app/models/concerns/agentable.rb +++ b/enterprise/app/models/concerns/agentable.rb @@ -7,7 +7,8 @@ module Concerns::Agentable instructions: ->(context) { agent_instructions(context) }, tools: agent_tools, model: agent_model, - temperature: temperature || 0.7 + temperature: temperature || 0.7, + response_schema: agent_response_schema ) end @@ -45,6 +46,10 @@ module Concerns::Agentable 'gpt-4.1-mini' end + def agent_response_schema + Captain::ResponseSchema + end + def prompt_context raise NotImplementedError, "#{self.class} must implement prompt_context" end