ooodc/PicoBot
1
0
Fork 0
Code Pull Requests Activity

feat:记忆修改隔离

Browse Source
This commit is contained in:
oudecheng 2026-05-28 14:55:52 +08:00
parent 598d425c28
commit 7755164df5
2 changed files with 3 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/tools/memory_manage.rs
View File

@ -182,11 +182,7 @@ fn scope_key_from_context(context: &ToolContext) -> Result<String, ToolResult> {
.channel_name .channel_name
.as_deref() .as_deref()
.ok_or_else(|| error_result("memory_manage requires channel_name in tool context"))?; .ok_or_else(|| error_result("memory_manage requires channel_name in tool context"))?;
let sender_id = context Ok(channel_name.to_string())
.sender_id
.as_deref()
.ok_or_else(|| error_result("memory_manage requires sender_id in tool context"))?;
Ok(format!("{}:{}", channel_name, sender_id))
} }
fn memory_to_json(memory: MemoryRecord) -> serde_json::Value { fn memory_to_json(memory: MemoryRecord) -> serde_json::Value {
@ -229,7 +225,6 @@ mod tests {
let tool = MemoryManageTool::new(store); let tool = MemoryManageTool::new(store);
let context = ToolContext { let context = ToolContext {
channel_name: Some(TEST_CHANNEL.to_string()), channel_name: Some(TEST_CHANNEL.to_string()),
sender_id: Some("user-1".to_string()),
chat_id: Some("chat-1".to_string()), chat_id: Some("chat-1".to_string()),
session_id: Some(format!("{}:chat-1", TEST_CHANNEL)), session_id: Some(format!("{}:chat-1", TEST_CHANNEL)),
message_id: Some("msg-1".to_string()), message_id: Some("msg-1".to_string()),
@ -279,7 +274,6 @@ mod tests {
let tool = MemoryManageTool::new(store); let tool = MemoryManageTool::new(store);
let context = ToolContext { let context = ToolContext {
channel_name: Some(TEST_CHANNEL.to_string()), channel_name: Some(TEST_CHANNEL.to_string()),
sender_id: Some("user-1".to_string()),
..ToolContext::default() ..ToolContext::default()
}; };

10
src/tools/memory_search.rs
View File

@ -189,11 +189,7 @@ fn scope_key_from_context(context: &ToolContext) -> Result<String, ToolResult> {
.channel_name .channel_name
.as_deref() .as_deref()
.ok_or_else(|| error_result("memory_search requires channel_name in tool context"))?; .ok_or_else(|| error_result("memory_search requires channel_name in tool context"))?;
let sender_id = context Ok(channel_name.to_string())
.sender_id
.as_deref()
.ok_or_else(|| error_result("memory_search requires sender_id in tool context"))?;
Ok(format!("{}:{}", channel_name, sender_id))
} }
fn memory_to_json(memory: MemoryRecord) -> serde_json::Value { fn memory_to_json(memory: MemoryRecord) -> serde_json::Value {
@ -236,7 +232,7 @@ mod tests {
store store
.put_memory(&crate::storage::MemoryUpsert { .put_memory(&crate::storage::MemoryUpsert {
scope_kind: "user".to_string(), scope_kind: "user".to_string(),
scope_key: format!("{}:user-1", TEST_CHANNEL), scope_key: TEST_CHANNEL.to_string(),
namespace: "preferences".to_string(), namespace: "preferences".to_string(),
memory_key: "language".to_string(), memory_key: "language".to_string(),
content: "User prefers Chinese responses".to_string(), content: "User prefers Chinese responses".to_string(),
@ -252,7 +248,6 @@ mod tests {
let tool = MemorySearchTool::new(store); let tool = MemorySearchTool::new(store);
let context = ToolContext { let context = ToolContext {
channel_name: Some(TEST_CHANNEL.to_string()), channel_name: Some(TEST_CHANNEL.to_string()),
sender_id: Some("user-1".to_string()),
chat_id: Some("chat-1".to_string()), chat_id: Some("chat-1".to_string()),
session_id: Some(format!("{}:chat-1", TEST_CHANNEL)), session_id: Some(format!("{}:chat-1", TEST_CHANNEL)),
message_id: Some("msg-2".to_string()), message_id: Some("msg-2".to_string()),
@ -310,7 +305,6 @@ mod tests {
let tool = MemorySearchTool::new(store); let tool = MemorySearchTool::new(store);
let context = ToolContext { let context = ToolContext {
channel_name: Some(TEST_CHANNEL.to_string()), channel_name: Some(TEST_CHANNEL.to_string()),
sender_id: Some("user-1".to_string()),
..ToolContext::default() ..ToolContext::default()
}; };