perf: updated the memory injection warning so it only logs once, rather than after each keystroke
This commit is contained in:
@@ -709,6 +709,10 @@ impl RequestContext {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn extract_role(&self, app: &AppConfig) -> Result<Role> {
|
pub fn extract_role(&self, app: &AppConfig) -> Result<Role> {
|
||||||
|
self.extract_role_impl(app, true)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn extract_role_impl(&self, app: &AppConfig, inject_memory: bool) -> Result<Role> {
|
||||||
let mut role = if let Some(session) = self.session.as_ref() {
|
let mut role = if let Some(session) = self.session.as_ref() {
|
||||||
session.to_role()
|
session.to_role()
|
||||||
} else if let Some(agent) = self.agent.as_ref() {
|
} else if let Some(agent) = self.agent.as_ref() {
|
||||||
@@ -757,6 +761,7 @@ impl RequestContext {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if inject_memory {
|
||||||
let memory_config = self.memory_config();
|
let memory_config = self.memory_config();
|
||||||
if memory_config.enabled {
|
if memory_config.enabled {
|
||||||
let store = MemoryStore {
|
let store = MemoryStore {
|
||||||
@@ -787,6 +792,7 @@ impl RequestContext {
|
|||||||
Err(e) => warn!("memory injection failed: {}", e),
|
Err(e) => warn!("memory injection failed: {}", e),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Ok(self.skill_registry.effective_role(&role, &policy))
|
Ok(self.skill_registry.effective_role(&role, &policy))
|
||||||
}
|
}
|
||||||
@@ -1276,7 +1282,7 @@ impl RequestContext {
|
|||||||
|
|
||||||
pub fn generate_prompt_context(&self, app: &AppConfig) -> HashMap<&str, String> {
|
pub fn generate_prompt_context(&self, app: &AppConfig) -> HashMap<&str, String> {
|
||||||
let mut output = HashMap::new();
|
let mut output = HashMap::new();
|
||||||
let role = self.extract_role(app).unwrap_or_else(|err| {
|
let role = self.extract_role_impl(app, false).unwrap_or_else(|err| {
|
||||||
warn!("failed to compute effective role for prompt rendering: {err}");
|
warn!("failed to compute effective role for prompt rendering: {err}");
|
||||||
Role::default()
|
Role::default()
|
||||||
});
|
});
|
||||||
|
|||||||
Reference in New Issue
Block a user