mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-27 19:25:15 +02:00
feat: remove WIP status HITL notion, linear and google drive tools
This commit is contained in:
parent
00b3aac1e3
commit
4474b73f29
2 changed files with 18 additions and 19 deletions
|
|
@ -292,6 +292,18 @@ async def create_surfsense_deep_agent(
|
||||||
]
|
]
|
||||||
modified_disabled_tools.extend(linear_tools)
|
modified_disabled_tools.extend(linear_tools)
|
||||||
|
|
||||||
|
# Disable Google Drive action tools if no Google Drive connector is configured
|
||||||
|
has_google_drive_connector = (
|
||||||
|
available_connectors is not None
|
||||||
|
and "GOOGLE_DRIVE_FILE" in available_connectors
|
||||||
|
)
|
||||||
|
if not has_google_drive_connector:
|
||||||
|
google_drive_tools = [
|
||||||
|
"create_google_drive_file",
|
||||||
|
"delete_google_drive_file",
|
||||||
|
]
|
||||||
|
modified_disabled_tools.extend(google_drive_tools)
|
||||||
|
|
||||||
# Build tools using the async registry (includes MCP tools)
|
# Build tools using the async registry (includes MCP tools)
|
||||||
_t0 = time.perf_counter()
|
_t0 = time.perf_counter()
|
||||||
tools = await build_tools_async(
|
tools = await build_tools_async(
|
||||||
|
|
|
||||||
|
|
@ -245,7 +245,8 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
requires=["user_id", "search_space_id", "db_session", "thread_visibility"],
|
requires=["user_id", "search_space_id", "db_session", "thread_visibility"],
|
||||||
),
|
),
|
||||||
# =========================================================================
|
# =========================================================================
|
||||||
# LINEAR TOOLS - create, update, delete issues (WIP - hidden from UI)
|
# LINEAR TOOLS - create, update, delete issues
|
||||||
|
# Auto-disabled when no Linear connector is configured (see chat_deepagent.py)
|
||||||
# =========================================================================
|
# =========================================================================
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="create_linear_issue",
|
name="create_linear_issue",
|
||||||
|
|
@ -256,8 +257,6 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="update_linear_issue",
|
name="update_linear_issue",
|
||||||
|
|
@ -268,8 +267,6 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="delete_linear_issue",
|
name="delete_linear_issue",
|
||||||
|
|
@ -280,11 +277,10 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
# =========================================================================
|
# =========================================================================
|
||||||
# NOTION TOOLS - create, update, delete pages (WIP - hidden from UI)
|
# NOTION TOOLS - create, update, delete pages
|
||||||
|
# Auto-disabled when no Notion connector is configured (see chat_deepagent.py)
|
||||||
# =========================================================================
|
# =========================================================================
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="create_notion_page",
|
name="create_notion_page",
|
||||||
|
|
@ -295,8 +291,6 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="update_notion_page",
|
name="update_notion_page",
|
||||||
|
|
@ -307,8 +301,6 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="delete_notion_page",
|
name="delete_notion_page",
|
||||||
|
|
@ -319,11 +311,10 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
# =========================================================================
|
# =========================================================================
|
||||||
# GOOGLE DRIVE TOOLS - create files, delete files (WIP - hidden from UI)
|
# GOOGLE DRIVE TOOLS - create files, delete files
|
||||||
|
# Auto-disabled when no Google Drive connector is configured (see chat_deepagent.py)
|
||||||
# =========================================================================
|
# =========================================================================
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="create_google_drive_file",
|
name="create_google_drive_file",
|
||||||
|
|
@ -334,8 +325,6 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
ToolDefinition(
|
ToolDefinition(
|
||||||
name="delete_google_drive_file",
|
name="delete_google_drive_file",
|
||||||
|
|
@ -346,8 +335,6 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
|
||||||
user_id=deps["user_id"],
|
user_id=deps["user_id"],
|
||||||
),
|
),
|
||||||
requires=["db_session", "search_space_id", "user_id"],
|
requires=["db_session", "search_space_id", "user_id"],
|
||||||
enabled_by_default=False,
|
|
||||||
hidden=True,
|
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue