apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-03 21:02:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: implement MCP Tool Trust methods for managing trusted tools

Browse source 
Added methods to trust and untrust tools in the MCP connector's "Always Allow" list, allowing for streamlined tool usage without HITL approval. This enhancement supports better management of trusted tools within the application.
This commit is contained in:
Anish Sarkar 2026-04-13 20:21:46 +05:30
parent 71cd04b05e
commit 1c9c496e01
5 changed files with 126 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
surfsense_web/lib/chat/streaming-state.ts
View file

@ -148,9 +148,10 @@ export function addToolCall(
toolsWithUI: Set<string>,
toolCallId: string,
toolName: string,
args: Record<string, unknown>
args: Record<string, unknown>,
force = false,
): void {
if (toolsWithUI.has(toolName)) {
if (force || toolsWithUI.has(toolName)) {
state.contentParts.push({
type: "tool-call",
toolCallId,
@ -175,13 +176,20 @@ export function updateToolCall(
}
}
function _hasInterruptResult(part: ContentPart): boolean {
if (part.type !== "tool-call") return false;
const r = (part as { result?: unknown }).result;
return typeof r === "object" && r !== null && "__interrupt__" in r;
}
export function buildContentForUI(
state: ContentPartsState,
toolsWithUI: Set<string>
): ThreadMessageLike["content"] {
const filtered = state.contentParts.filter((part) => {
if (part.type === "text") return part.text.length > 0;
if (part.type === "tool-call") return toolsWithUI.has(part.toolName);
if (part.type === "tool-call")
return toolsWithUI.has(part.toolName) || _hasInterruptResult(part);
if (part.type === "data-thinking-steps") return true;
return false;
});
@ -199,7 +207,10 @@ export function buildContentForPersistence(
for (const part of state.contentParts) {
if (part.type === "text" && part.text.length > 0) {
parts.push(part);
} else if (part.type === "tool-call" && toolsWithUI.has(part.toolName)) {
} else if (
part.type === "tool-call" &&
(toolsWithUI.has(part.toolName) || _hasInterruptResult(part))
) {
parts.push(part);
} else if (part.type === "data-thinking-steps") {
parts.push(part);