address pr feedback
This commit is contained in:
@@ -148,7 +148,21 @@ def filter_server_tools(server):
|
||||
tools_to_remove.add(tool_name)
|
||||
|
||||
for tool_name in tools_to_remove:
|
||||
server.local_provider.remove_tool(tool_name)
|
||||
try:
|
||||
server.local_provider.remove_tool(tool_name)
|
||||
except AttributeError:
|
||||
logger.warning(
|
||||
"Failed to remove tool '%s': remove_tool not available on server.local_provider",
|
||||
tool_name,
|
||||
)
|
||||
continue
|
||||
except Exception as exc:
|
||||
logger.warning(
|
||||
"Failed to remove tool '%s': %s",
|
||||
tool_name,
|
||||
exc,
|
||||
)
|
||||
continue
|
||||
tools_removed += 1
|
||||
|
||||
if tools_removed > 0:
|
||||
|
||||
@@ -552,7 +552,7 @@ async def create_drive_file(
|
||||
f"[create_drive_file] Invoked. Email: '{user_google_email}', File Name: {file_name}, Folder ID: {folder_id}, fileUrl: {fileUrl}"
|
||||
)
|
||||
|
||||
if not content and not fileUrl and mime_type != FOLDER_MIME_TYPE:
|
||||
if content is None and fileUrl is None and mime_type != FOLDER_MIME_TYPE:
|
||||
raise Exception("You must provide either 'content' or 'fileUrl'.")
|
||||
|
||||
# Create folder (no content or media_body). Prefer create_drive_folder for new code.
|
||||
|
||||
Reference in New Issue
Block a user