From 96e18d74c51aa7bf0d3e68f7a6efbfd4a1e93eb2 Mon Sep 17 00:00:00 2001
From: Shivam Mishra <scm.mymail@gmail.com>
Date: Mon, 4 Aug 2025 16:58:23 +0530
Subject: [PATCH] feat: include schema

---
 enterprise/app/models/concerns/agentable.rb | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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