From 345a21244723b8a87f5d34a413bd2ce44cd771da Mon Sep 17 00:00:00 2001
From: Shivam Mishra <scm.mymail@gmail.com>
Date: Tue, 15 Jul 2025 12:46:44 +0530
Subject: [PATCH] feat: include guidelines and guardrails

---
 enterprise/app/models/captain/assistant.rb      |  6 ++++--
 enterprise/lib/captain/prompts/assistant.liquid | 14 ++++++++++++++
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/enterprise/app/models/captain/assistant.rb b/enterprise/app/models/captain/assistant.rb
index 33027ad59..e37d2e9db 100644
--- a/enterprise/app/models/captain/assistant.rb
+++ b/enterprise/app/models/captain/assistant.rb
@@ -93,11 +93,13 @@ class Captain::Assistant < ApplicationRecord
           key: scenario.title.parameterize.underscore,
           description: scenario.description
         }
-      end
+      end,
+      response_guidelines: response_guidelines || [],
+      guardrails: guardrails || []
     }
   end
 
   def default_avatar_url
     "#{ENV.fetch('FRONTEND_URL', nil)}/assets/images/dashboard/captain/logo.svg"
   end
-end
\ No newline at end of file
+end
diff --git a/enterprise/lib/captain/prompts/assistant.liquid b/enterprise/lib/captain/prompts/assistant.liquid
index a268b325b..63ff494ef 100644
--- a/enterprise/lib/captain/prompts/assistant.liquid
+++ b/enterprise/lib/captain/prompts/assistant.liquid
@@ -10,6 +10,20 @@ You are {{ name }}, a helpful AI assistant for {{ product_name }}.
 
 {{ description }}
 
+{% if response_guidelines.size > 0 -%}
+# Response Guidelines
+{% for guideline in response_guidelines -%}
+- {{ guideline }}
+{% endfor %}
+{% endif -%}
+
+{% if guardrails.size > 0 -%}
+# Guardrails
+{% for guardrail in guardrails -%}
+- {{ guardrail }}
+{% endfor %}
+{% endif -%}
+
 # Available Handoffs
 You can hand off conversations to these specialized scenario agents when appropriate:
 {% for scenario in scenarios -%}