apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-06 20:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix bugs and pattern inconsistencies in google drive tools

Browse source
This commit is contained in:
CREDO23 2026-02-20 19:41:29 +02:00
parent 9abe136646
commit 856d258318
3 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
surfsense_backend/app/agents/new_chat/tools/google_drive/create_file.py
View file

@ -76,6 +76,9 @@ def create_create_google_drive_file_tool(
logger.error(f"Failed to fetch creation context: {context['error']}")
return {"status": "error", "message": context["error"]}
logger.info(
f"Requesting approval for creating Google Drive file: name='{name}', type='{file_type}'"
)
approval = interrupt(
{
"type": "google_drive_file_creation",
@ -97,6 +100,7 @@ def create_create_google_drive_file_tool(
decisions = decisions_raw if isinstance(decisions_raw, list) else [decisions_raw]
decisions = [d for d in decisions if isinstance(d, dict)]
if not decisions:
logger.warning("No approval decision received")
return {"status": "error", "message": "No approval decision received"}
decision = decisions[0]

10
surfsense_backend/app/agents/new_chat/tools/google_drive/trash_file.py
View file

@ -70,6 +70,15 @@ def create_trash_google_drive_file_tool(
file_id = file["file_id"]
connector_id_from_context = context["account"]["id"]
if not file_id:
return {
"status": "error",
"message": "File ID is missing from the indexed document. Please re-index the file and try again.",
}
logger.info(
f"Requesting approval for trashing Google Drive file: '{file_name}' (file_id={file_id})"
)
approval = interrupt(
{
"type": "google_drive_file_trash",
@ -88,6 +97,7 @@ def create_trash_google_drive_file_tool(
decisions = decisions_raw if isinstance(decisions_raw, list) else [decisions_raw]
decisions = [d for d in decisions if isinstance(d, dict)]
if not decisions:
logger.warning("No approval decision received")
return {"status": "error", "message": "No approval decision received"}
decision = decisions[0]