Add selective memory_pick branch apply

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: gouhongshen

memoria/crates/memoria-api/src/lib.rs

@@ -84,14 +84,12 @@ async fn call_log_mw(
             // which is necessary because JSON-RPC errors return HTTP 200.
             if !is_dashboard && !path.starts_with("/v1/mcp") {
                 if let Some(reporter) = &state.stats_reporter {
-                    reporter.report(
-                        memoria_service::stats_reporter::StatsEvent::ApiCallLogged {
-                            user_id: uid.clone(),
-                            path: path.clone(),
-                            is_mcp: false,
-                            is_success: status_code < 400,
-                        },
-                    );
+                    reporter.report(memoria_service::stats_reporter::StatsEvent::ApiCallLogged {
+                        user_id: uid.clone(),
+                        path: path.clone(),
+                        is_mcp: false,
+                        is_success: status_code < 400,
+                    });
                 }
             }
             if let Some(mask) = should_mark_metrics_dirty(&method, &path, status_code) {
@@ -147,6 +145,7 @@ fn should_mark_metrics_dirty(
                 } else if path.starts_with("/v1/snapshots/") && path.ends_with("/rollback")
                     || path.starts_with("/v1/branches/") && path.ends_with("/checkout")
                     || path.starts_with("/v1/branches/") && path.ends_with("/merge")
+                    || path.starts_with("/v1/branches/") && path.ends_with("/pick")
                 {
                     Some(DirtyMask::FULL)
                 } else if path.starts_with("/v1/sessions/") && path.ends_with("/summary") {
@@ -283,6 +282,10 @@ pub fn build_router(state: AppState) -> Router {
             "/v1/branches/:name/diff",
             get(routes::snapshots::diff_branch),
         )
+        .route(
+            "/v1/branches/:name/pick",
+            post(routes::snapshots::pick_branch),
+        )
         .route(
             "/v1/branches/:name",
             delete(routes::snapshots::delete_branch),

memoria/crates/memoria-api/src/models.rs

@@ -284,6 +284,44 @@ fn default_strategy() -> String {
     "accept".to_string()
 }
 
+#[derive(Debug, Deserialize, Serialize)]
+pub struct PickRequest {
+    #[serde(default = "default_pick_target")]
+    pub target: String,
+    #[serde(default = "default_pick_strategy")]
+    pub strategy: String,
+    pub selector: PickSelector,
+}
+
+fn default_pick_target() -> String {
+    "main".to_string()
+}
+
+fn default_pick_strategy() -> String {
+    "fail".to_string()
+}
+
+fn default_pick_top_k() -> i64 {
+    5
+}
+
+#[derive(Debug, Deserialize, Serialize)]
+#[serde(tag = "type", rename_all = "snake_case")]
+pub enum PickSelector {
+    KeyList {
+        keys: Vec<String>,
+    },
+    SnapshotRange {
+        from_snapshot: String,
+        to_snapshot: String,
+    },
+    Retrieve {
+        query: String,
+        #[serde(default = "default_pick_top_k")]
+        top_k: i64,
+    },
+}
+
 // ── Helpers ───────────────────────────────────────────────────────────────────
 
 pub fn parse_memory_type(s: &str) -> Result<MemoryType, String> {

memoria/crates/memoria-api/src/routes/mcp.rs

@@ -91,7 +91,7 @@ fn mcp_tool_dirty_mask(tool: &str) -> Option<crate::metrics_summary::DirtyMask>
         "memory_snapshot" | "memory_snapshot_delete" => Some(DirtyMask::SNAPSHOT),
         "memory_rollback" => Some(DirtyMask::FULL),
         "memory_branch" | "memory_branch_delete" => Some(DirtyMask::BRANCH),
-        "memory_checkout" | "memory_merge" => Some(DirtyMask::FULL),
+        "memory_checkout" | "memory_merge" | "memory_pick" => Some(DirtyMask::FULL),
         _ => None,
     }
 }
@@ -133,14 +133,12 @@ pub async fn mcp_handler(
                 RpcMeta::err($code),
             );
             if let Some(reporter) = &state.stats_reporter {
-                reporter.report(
-                    memoria_service::stats_reporter::StatsEvent::ApiCallLogged {
-                        user_id: auth.user_id.clone(),
-                        path: $path.to_string(),
-                        is_mcp: true,
-                        is_success: false,
-                    },
-                );
+                reporter.report(memoria_service::stats_reporter::StatsEvent::ApiCallLogged {
+                    user_id: auth.user_id.clone(),
+                    path: $path.to_string(),
+                    is_mcp: true,
+                    is_success: false,
+                });
             }
             return Json($body).into_response();
         }};

memoria/crates/memoria-api/src/routes/snapshots.rs

@@ -13,7 +13,7 @@ use crate::{
     routes::memory::{api_err, api_err_typed},
     state::AppState,
 };
-use memoria_core::TrustTier;
+use memoria_core::{MemoriaError, TrustTier};
 use memoria_git::GitForDataService;
 use std::sync::Arc;
 
@@ -66,6 +66,27 @@ async fn git_call(
     Ok(json!({ "result": text }))
 }
 
+async fn git_call_pick(
+    state: &AppState,
+    user_id: &str,
+    args: serde_json::Value,
+) -> Result<serde_json::Value, (StatusCode, String)> {
+    let result =
+        memoria_mcp::git_tools::call("memory_pick", args, &state.git, &state.service, user_id)
+            .await
+            .map_err(|e| match e {
+                MemoriaError::Validation(msg) if msg.starts_with("Conflict:") => {
+                    (StatusCode::CONFLICT, msg)
+                }
+                other => api_err_typed(other),
+            })?;
+    let text = result["content"][0]["text"]
+        .as_str()
+        .unwrap_or("")
+        .to_string();
+    Ok(json!({ "result": text }))
+}
+
 async fn user_snapshot_store(
     state: &AppState,
     user_id: &str,
@@ -687,6 +708,26 @@ pub async fn diff_branch(
     Ok(Json(r))
 }
 
+pub async fn pick_branch(
+    State(state): State<AppState>,
+    AuthUser { user_id, .. }: AuthUser,
+    Path(name): Path<String>,
+    Json(req): Json<PickRequest>,
+) -> Result<Json<serde_json::Value>, (StatusCode, String)> {
+    let r = git_call_pick(
+        &state,
+        &user_id,
+        json!({
+            "source": name,
+            "target": req.target,
+            "strategy": req.strategy,
+            "selector": req.selector,
+        }),
+    )
+    .await?;
+    Ok(Json(r))
+}
+
 pub async fn delete_branch(
     State(state): State<AppState>,
     AuthUser { user_id, .. }: AuthUser,