mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-08-15 23:50:32 +02:00
Add files via upload
This commit is contained in:
@@ -20,16 +20,20 @@ func (db *DB) FindNearestToolExecutionArguments(conversationID, toolName string,
|
|||||||
if window <= 0 {
|
if window <= 0 {
|
||||||
window = 5 * time.Second
|
window = 5 * time.Second
|
||||||
}
|
}
|
||||||
|
names := []string{toolName}
|
||||||
|
if !strings.Contains(toolName, "::") {
|
||||||
|
names = append(names, "eino_fs::"+toolName)
|
||||||
|
}
|
||||||
start := at.Add(-window)
|
start := at.Add(-window)
|
||||||
end := at.Add(window)
|
end := at.Add(window)
|
||||||
rows, err := db.Query(`
|
rows, err := db.Query(`
|
||||||
SELECT id, arguments
|
SELECT id, arguments
|
||||||
FROM tool_executions
|
FROM tool_executions
|
||||||
WHERE conversation_id = ?
|
WHERE conversation_id = ?
|
||||||
AND tool_name = ?
|
AND tool_name IN (?, ?)
|
||||||
AND julianday(start_time) BETWEEN julianday(?) AND julianday(?)
|
AND julianday(start_time) BETWEEN julianday(?) AND julianday(?)
|
||||||
ORDER BY ABS(julianday(start_time) - julianday(?)) ASC, start_time ASC
|
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 {
|
if err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user