From 053c75b2ccbac402f4cd4437020834f2a954a44c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=AC=E6=98=8E?=
 <83812544+Ed1s0nZ@users.noreply.github.com>
Date: Sat, 15 Aug 2026 11:04:35 +0800
Subject: [PATCH] Add files via upload

---
 internal/database/tool_execution_args_lookup.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/internal/database/tool_execution_args_lookup.go b/internal/database/tool_execution_args_lookup.go
index f2583359..cc3a0a63 100644
--- a/internal/database/tool_execution_args_lookup.go
+++ b/internal/database/tool_execution_args_lookup.go
@@ -20,16 +20,20 @@ func (db *DB) FindNearestToolExecutionArguments(conversationID, toolName string,
 	if window <= 0 {
 		window = 5 * time.Second
 	}
+	names := []string{toolName}
+	if !strings.Contains(toolName, "::") {
+		names = append(names, "eino_fs::"+toolName)
+	}
 	start := at.Add(-window)
 	end := at.Add(window)
 	rows, err := db.Query(`
 SELECT id, arguments
 FROM tool_executions
 WHERE conversation_id = ?
-  AND tool_name = ?
+  AND tool_name IN (?, ?)
   AND julianday(start_time) BETWEEN julianday(?) AND julianday(?)
 ORDER BY ABS(julianday(start_time) - julianday(?)) ASC, start_time ASC
-LIMIT 1`, conversationID, toolName, start, end, at)
+LIMIT 1`, conversationID, names[0], names[len(names)-1], start, end, at)
 	if err != nil {
 		return "", nil, err
 	}