mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-07 16:19:09 +00:00
25d4886b9a
Port five upstream PRs onto dev: - #1621 / #1658: durable dispatch with loopback webhook defense, create_durable_run helper, _config_with_prepare_run_id, degradation to None for relative/loopback completion webhook URLs - #1696: run-level completion webhook deduplication (replace claim-then-post with post-then-flag per run_id), DeferredErrorModel for graph-factory resilience, ToolRetryMiddleware for task subagents, TimeoutWrapupMiddleware for all three graphs - #1697: lazy-load __init__.py for agent.middleware, agent.tools, agent.dashboard (PEP 562); defer heavy imports (exa_py in web_search, agent.webapp in request_pr_review, deepagents in sandbox.py); add ttl_cache.py with stale-while-revalidate for tool loaders Refs: #137
120 lines
4.3 KiB
Python
120 lines
4.3 KiB
Python
import sys
|
|
from types import ModuleType
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
_TOOL_MODULES = {
|
|
"add_finding": ".add_finding",
|
|
"enter_plan_mode": ".enter_plan_mode",
|
|
"fetch_url": ".fetch_url",
|
|
"http_request": ".http_request",
|
|
"linear_comment": ".linear_comment",
|
|
"linear_create_issue": ".linear_create_issue",
|
|
"linear_delete_issue": ".linear_delete_issue",
|
|
"linear_get_issue": ".linear_get_issue",
|
|
"linear_get_issue_comments": ".linear_get_issue_comments",
|
|
"linear_list_teams": ".linear_list_teams",
|
|
"linear_update_issue": ".linear_update_issue",
|
|
"list_findings": ".list_findings",
|
|
"list_review_findings": ".list_review_findings",
|
|
"open_pull_request": ".open_pull_request",
|
|
"publish_review": ".publish_review",
|
|
"read_repo_file": ".read_repo_file",
|
|
"report_platform_issue": ".report_platform_issue",
|
|
"request_pr_review": ".request_pr_review",
|
|
"reply_to_finding_thread": ".reply_to_finding_thread",
|
|
"resolve_finding_thread": ".resolve_finding_thread",
|
|
"save_plan": ".save_plan",
|
|
"schedule_thread_wakeup": ".schedule_thread_wakeup",
|
|
"search_repo_code": ".search_repo_code",
|
|
"slack_add_reaction": ".slack_add_reaction",
|
|
"slack_read_thread_messages": ".slack_read_thread_messages",
|
|
"slack_start_new_thread": ".slack_start_new_thread",
|
|
"slack_thread_reply": ".slack_thread_reply",
|
|
"update_finding": ".update_finding",
|
|
"web_search": ".web_search",
|
|
}
|
|
|
|
__all__ = [
|
|
"add_finding",
|
|
"enter_plan_mode",
|
|
"fetch_url",
|
|
"http_request",
|
|
"linear_comment",
|
|
"linear_create_issue",
|
|
"linear_delete_issue",
|
|
"linear_get_issue",
|
|
"linear_get_issue_comments",
|
|
"linear_list_teams",
|
|
"linear_update_issue",
|
|
"list_findings",
|
|
"list_review_findings",
|
|
"open_pull_request",
|
|
"publish_review",
|
|
"read_repo_file",
|
|
"report_platform_issue",
|
|
"request_pr_review",
|
|
"reply_to_finding_thread",
|
|
"resolve_finding_thread",
|
|
"save_plan",
|
|
"schedule_thread_wakeup",
|
|
"search_repo_code",
|
|
"slack_add_reaction",
|
|
"slack_read_thread_messages",
|
|
"slack_start_new_thread",
|
|
"slack_thread_reply",
|
|
"update_finding",
|
|
"web_search",
|
|
]
|
|
|
|
if TYPE_CHECKING:
|
|
from .add_finding import add_finding
|
|
from .enter_plan_mode import enter_plan_mode
|
|
from .fetch_url import fetch_url
|
|
from .http_request import http_request
|
|
from .linear_comment import linear_comment
|
|
from .linear_create_issue import linear_create_issue
|
|
from .linear_delete_issue import linear_delete_issue
|
|
from .linear_get_issue import linear_get_issue
|
|
from .linear_get_issue_comments import linear_get_issue_comments
|
|
from .linear_list_teams import linear_list_teams
|
|
from .linear_update_issue import linear_update_issue
|
|
from .list_findings import list_findings
|
|
from .list_review_findings import list_review_findings
|
|
from .open_pull_request import open_pull_request
|
|
from .publish_review import publish_review
|
|
from .read_repo_file import read_repo_file
|
|
from .reply_to_finding_thread import reply_to_finding_thread
|
|
from .report_platform_issue import report_platform_issue
|
|
from .request_pr_review import request_pr_review
|
|
from .resolve_finding_thread import resolve_finding_thread
|
|
from .save_plan import save_plan
|
|
from .schedule_thread_wakeup import schedule_thread_wakeup
|
|
from .search_repo_code import search_repo_code
|
|
from .slack_add_reaction import slack_add_reaction
|
|
from .slack_read_thread_messages import slack_read_thread_messages
|
|
from .slack_start_new_thread import slack_start_new_thread
|
|
from .slack_thread_reply import slack_thread_reply
|
|
from .update_finding import update_finding
|
|
from .web_search import web_search
|
|
|
|
|
|
def _load_tool(name: str) -> Any:
|
|
module_name = _TOOL_MODULES.get(name)
|
|
if module_name is None:
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
|
from importlib import import_module
|
|
|
|
value = getattr(import_module(module_name, __name__), name)
|
|
globals()[name] = value
|
|
return value
|
|
|
|
|
|
class _LazyToolsModule(ModuleType):
|
|
def __getattribute__(self, name: str) -> Any:
|
|
tool_map = ModuleType.__getattribute__(self, "__dict__").get("_TOOL_MODULES", {})
|
|
if name in tool_map:
|
|
return _load_tool(name)
|
|
return ModuleType.__getattribute__(self, name)
|
|
|
|
|
|
sys.modules[__name__].__class__ = _LazyToolsModule
|