public/galaxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve agent prompting and classifier latency; clean up build warnings

Browse Source
This commit is contained in:
rkw6086
2026-09-15 14:25:00 -05:00
parent 66b73b1aac
commit 76d492aa75
69 changed files with 368 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/persistence/src/model.rs
+3
View File
@@ -1088,6 +1088,9 @@ pub struct LocalAgentGroup {
#[derive(Debug, Serialize, Deserialize, Clone, Default)]
pub struct AgentConversationData {
/// Display-only tool intent, scoped to this conversation and keyed by call ID.
#[serde(default, skip_serializing_if = "HashMap::is_empty")]
pub tool_titles: HashMap<String, String>,
#[serde(default, skip_serializing_if = "AgentBackend::is_provider")]
pub agent_backend: AgentBackend,
/// Versioned application-owned snapshot for an active direct-provider run.
crates/persistence/src/model_tests.rs
+19
View File
@@ -139,6 +139,7 @@ fn agent_conversation_data_roundtrips_last_event_sequence() {
agent_backend: Default::default(),
active_provider_run_json: None,
local_agent_groups: Vec::new(),
tool_titles: Default::default(),
server_conversation_token: None,
conversation_usage_metadata: None,
reverted_action_ids: None,
@@ -243,6 +244,7 @@ fn agent_conversation_data_roundtrips_remote_child_marker() {
agent_backend: Default::default(),
active_provider_run_json: None,
local_agent_groups: Vec::new(),
tool_titles: Default::default(),
server_conversation_token: None,
conversation_usage_metadata: None,
reverted_action_ids: None,
@@ -272,6 +274,7 @@ fn agent_conversation_data_roundtrips_optimistic_root_marker() {
agent_backend: Default::default(),
active_provider_run_json: None,
local_agent_groups: Vec::new(),
tool_titles: Default::default(),
server_conversation_token: None,
conversation_usage_metadata: None,
reverted_action_ids: None,
@@ -313,6 +316,7 @@ fn agent_conversation_data_skips_serializing_none_last_event_sequence() {
agent_backend: Default::default(),
active_provider_run_json: None,
local_agent_groups: Vec::new(),
tool_titles: Default::default(),
server_conversation_token: None,
conversation_usage_metadata: None,
reverted_action_ids: None,
@@ -344,6 +348,7 @@ fn agent_conversation_data_roundtrips_pinned() {
agent_backend: Default::default(),
active_provider_run_json: None,
local_agent_groups: Vec::new(),
tool_titles: Default::default(),
server_conversation_token: None,
conversation_usage_metadata: None,
reverted_action_ids: None,
@@ -373,6 +378,7 @@ fn agent_conversation_data_skips_serializing_unpinned() {
agent_backend: Default::default(),
active_provider_run_json: None,
local_agent_groups: Vec::new(),
tool_titles: Default::default(),
server_conversation_token: None,
conversation_usage_metadata: None,
reverted_action_ids: None,
@@ -436,3 +442,16 @@ fn model_token_usage_replay_skips_non_custom_endpoint_entries() {
};
assert!(warp_only.to_proto_custom_endpoint_usage().is_none());
}
#[test]
fn tool_titles_survive_roundtrip_and_are_optional_for_old_conversations() {
let mut data = AgentConversationData::default();
data.tool_titles
.insert("call-1".into(), "Find cache invalidation paths".into());
let mut json = serde_json::to_value(&data).unwrap();
let restored: AgentConversationData = serde_json::from_value(json.clone()).unwrap();
assert_eq!(restored.tool_titles, data.tool_titles);
json.as_object_mut().unwrap().remove("tool_titles");
let legacy: AgentConversationData = serde_json::from_value(json).unwrap();
assert!(legacy.tool_titles.is_empty());
}