feat: inject global rules into AI system prompt and fix Rules UI seeding

- Load global rules (AIFact/AIMemory) from local CloudModel and inject
  them into the Bedrock/OpenAI system prompt as a '## Global Rules' section
  when memory is enabled.
- Fix rule seeding: always re-seed predefined rules when the CloudModel has
  none, regardless of the has_seeded_predefined_rules flag (handles case
  where flag was set but rules never persisted due to prior missing owner).
- Rename /context slash command to /copy-context: dumps the full context
  window (global rules, progressive summary, message history) to the
  clipboard for debugging.
This commit is contained in:
Ryan Ward
2026-07-16 12:25:56 -05:00
parent 66ef451115
commit 3d2d90fd4e
10 changed files with 188 additions and 15 deletions
+29
View File
@@ -152,6 +152,9 @@ pub struct RequestParams {
/// can store them back into the conversation for the next request cycle.
pub bedrock_messages_sent:
std::sync::Arc<std::sync::Mutex<Vec<crate::ai::bedrock::convert::ConversationMessage>>>,
/// Global rules (name, content) from the local CloudModel (AIFact/AIMemory).
/// Injected into the system prompt when `is_memory_enabled` is true.
pub global_rules: Vec<(String, String)>,
}
pub type Event = Result<warp_multi_agent_api::ResponseEvent, Arc<AIApiError>>;
@@ -216,6 +219,7 @@ impl RequestParams {
bedrock_progressive_summary: None,
bedrock_tool_result_archive: vec![],
bedrock_messages_sent: std::sync::Arc::new(std::sync::Mutex::new(vec![])),
global_rules: vec![],
}
}
@@ -421,6 +425,31 @@ impl RequestParams {
bedrock_progressive_summary: None,
bedrock_tool_result_archive: Vec::new(),
bedrock_messages_sent: std::sync::Arc::new(std::sync::Mutex::new(Vec::new())),
global_rules: if is_memory_enabled {
Self::load_global_rules(app)
} else {
vec![]
},
}
}
/// Load global rules (AIFact/AIMemory) from the local CloudModel.
fn load_global_rules(app: &AppContext) -> Vec<(String, String)> {
use crate::ai::facts::{AIFact, AIMemory, CloudAIFactModel};
use crate::cloud_object::model::generic_string_model::GenericStringObjectId;
use crate::cloud_object::model::persistence::CloudModel;
CloudModel::as_ref(app)
.get_all_objects_of_type::<GenericStringObjectId, CloudAIFactModel>()
.filter_map(|fact| {
let AIFact::Memory(AIMemory { name, content, .. }) =
fact.model().string_model.clone();
if content.is_empty() {
None
} else {
Some((name.unwrap_or_default(), content))
}
})
.collect()
}
}
+2
View File
@@ -150,6 +150,7 @@ pub async fn generate_multi_agent_output(
tool_result_archive: params.bedrock_tool_result_archive.clone(),
progressive_summary: params.bedrock_progressive_summary.clone(),
messages_sent: params.bedrock_messages_sent.clone(),
global_rules: params.global_rules.clone(),
};
match openai_translator::execute(translator_request, &mut request).await {
@@ -178,6 +179,7 @@ pub async fn generate_multi_agent_output(
bedrock_tool_result_archive: params.bedrock_tool_result_archive.clone(),
bedrock_progressive_summary: params.bedrock_progressive_summary.clone(),
bedrock_messages_sent: params.bedrock_messages_sent.clone(),
global_rules: params.global_rules.clone(),
};
match crate::ai::bedrock::translator::execute(translator_request, &mut request).await {
+1 -1
View File
@@ -1890,7 +1890,7 @@ async fn test_full_proto_round_trip_with_tool_history() {
);
let messages = super::request_translator::extract_messages_from_request(&request);
let system_prompt = super::request_translator::extract_system_prompt(&request);
let system_prompt = super::request_translator::extract_system_prompt(&request, &[]);
let tools = super::request_translator::extract_tools(&request);
println!("\n=== FULL PROTO ROUND-TRIP TEST ===");
+14 -1
View File
@@ -954,7 +954,7 @@ fn collect_tool_result_ids(content: &MessageContent, ids: &mut std::collections:
}
}
pub fn extract_system_prompt(request: &api::Request) -> Option<String> {
pub fn extract_system_prompt(request: &api::Request, global_rules: &[(String, String)]) -> Option<String> {
let mut prompt = String::with_capacity(2048);
prompt.push_str("You are Galaxy, an AI coding assistant embedded in a terminal application. You help users with software engineering tasks including writing code, debugging, explaining concepts, and navigating codebases.\n\n");
@@ -1009,6 +1009,19 @@ pub fn extract_system_prompt(request: &api::Request) -> Option<String> {
}
}
// Inject global rules from the local CloudModel (stored as AIFact/AIMemory)
if !global_rules.is_empty() {
prompt.push_str("## Global Rules\n");
prompt.push_str("The following rules have been configured by the user and should be followed:\n\n");
for (name, content) in global_rules {
if !name.is_empty() {
prompt.push_str(&format!("### {}\n", name));
}
prompt.push_str(content);
prompt.push_str("\n\n");
}
}
prompt.push_str("## Tool Usage\n");
prompt.push_str("You have been given every tool you need to complete your tasks. Use them to achieve results with as few calls and as little back-and-forth as possible.\n\n");
prompt.push_str("**How to choose tools:**\n");
+3 -1
View File
@@ -18,6 +18,8 @@ pub struct TranslatorRequest {
pub bedrock_tool_result_archive: Vec<ConversationMessage>,
pub bedrock_progressive_summary: Option<String>,
pub bedrock_messages_sent: Arc<Mutex<Vec<ConversationMessage>>>,
/// Global rules (name, content) from the local CloudModel.
pub global_rules: Vec<(String, String)>,
}
pub async fn execute(
@@ -104,7 +106,7 @@ pub async fn execute(
request_translator::sanitize_messages_for_bedrock(&mut messages);
let system_prompt = request_translator::extract_system_prompt(request);
let system_prompt = request_translator::extract_system_prompt(request, &params.global_rules);
let tools = request_translator::extract_tools(request);
log::info!(
+4 -2
View File
@@ -319,8 +319,10 @@ impl RuleView {
.on_click(|ctx| ctx.dispatch_typed_action(RuleViewAction::InitializeProject))
});
// Seed predefined rules on first launch if no global rules exist
if ai_rules.is_empty() && !AISettings::as_ref(ctx).has_seeded_predefined_rules() {
// Seed predefined rules on first launch if no global rules exist.
// Also re-seed if the flag was set but rules are empty (e.g., prior bug
// where the flag was set but creation failed due to missing owner).
if ai_rules.is_empty() {
if let Some(owner) = owner {
let update_manager = UpdateManager::handle(ctx);
update_manager.update(ctx, |update_manager, ctx| {
+3 -1
View File
@@ -20,6 +20,8 @@ pub struct TranslatorRequest {
pub tool_result_archive: Vec<ConversationMessage>,
pub progressive_summary: Option<String>,
pub messages_sent: Arc<Mutex<Vec<ConversationMessage>>>,
/// Global rules (name, content) from the local CloudModel.
pub global_rules: Vec<(String, String)>,
}
pub async fn execute(
@@ -106,7 +108,7 @@ pub async fn execute(
sanitize_messages_for_openai(&mut messages);
let system_prompt = request_translator::extract_system_prompt(request);
let system_prompt = request_translator::extract_system_prompt(request, &params.global_rules);
let tools = request_translator::extract_tools(request);
log::info!(