open-swe/agent/middleware/__init__.py

from .check_message_queue import check_message_queue_before_model
from .ensure_no_empty_msg import ensure_no_empty_msg
from .notify_step_limit import notify_step_limit_reached
from .open_pr import open_pr_if_needed
from .sanitize_tool_inputs import SanitizeToolInputsMiddleware
from .tool_error_handler import ToolErrorMiddleware

__all__ = [
    "SanitizeToolInputsMiddleware",
    "ToolErrorMiddleware",
    "check_message_queue_before_model",
    "ensure_no_empty_msg",
    "notify_step_limit_reached",
    "open_pr_if_needed",
]
refactor: extract shared utils from server.py into agent/utils/ package 2026-02-10 11:22:52 -08:00			`from .check_message_queue import check_message_queue_before_model`
fix: Middleware to ensure no empty msg (#1010) 2026-03-04 17:59:52 -08:00			`from .ensure_no_empty_msg import ensure_no_empty_msg`
fix: notify users via Slack when agent hits model call step limit (#1204) * fix: notify users via Slack when agent hits model call step limit - Root cause: GraphRecursionError at 1000 steps bypassed all @after_agent middleware including open_pr_if_needed, leaving users with no notification - Change: Added ModelCallLimitMiddleware(run_limit=60) to intercept gracefully before the hard recursion limit, and added notify_step_limit_reached @after_agent middleware to post a Slack thread reply when the limit fires - Verified: 107 existing tests pass, no regressions * fix: harden step-limit Slack notification Ensure the step-limit notification runs after the PR safety net and cover the new middleware behavior with focused unit tests. --------- Co-authored-by: LangSmith Forge <forge-agent@langsmith.ai> Co-authored-by: Johannes du Plessis <johannes@langchain.dev> 2026-05-01 14:24:25 -07:00			`from .notify_step_limit import notify_step_limit_reached`
refactor: extract inline middleware from server.py into separate files 2026-02-09 15:34:18 -08:00			`from .open_pr import open_pr_if_needed`
fix: coerce malformed integer strings in read_file offset/limit params (#1216) - Root cause: LLM occasionally generates strings like '1, 80' or '170, "limit": 60' for integer fields, causing a Pydantic ValidationError and wasting an LLM turn - Change: add SanitizeToolInputsMiddleware in agent/middleware/sanitize_tool_inputs.py that extracts the leading integer from any string value in offset/limit before the call reaches Pydantic validation; registered before ToolErrorMiddleware in server.py - Verified: 14 unit tests covering all three production trace patterns pass Co-authored-by: LangSmith Forge <forge-agent@langsmith.ai> Co-authored-by: Johannes du Plessis <johannes@langchain.dev> 2026-05-01 14:29:48 -07:00			`from .sanitize_tool_inputs import SanitizeToolInputsMiddleware`
Refactor middleware into package and update repo config 2026-02-09 12:53:34 -08:00			`from .tool_error_handler import ToolErrorMiddleware`

refactor: extract inline middleware from server.py into separate files 2026-02-09 15:34:18 -08:00			`__all__ = [`
fix: coerce malformed integer strings in read_file offset/limit params (#1216) - Root cause: LLM occasionally generates strings like '1, 80' or '170, "limit": 60' for integer fields, causing a Pydantic ValidationError and wasting an LLM turn - Change: add SanitizeToolInputsMiddleware in agent/middleware/sanitize_tool_inputs.py that extracts the leading integer from any string value in offset/limit before the call reaches Pydantic validation; registered before ToolErrorMiddleware in server.py - Verified: 14 unit tests covering all three production trace patterns pass Co-authored-by: LangSmith Forge <forge-agent@langsmith.ai> Co-authored-by: Johannes du Plessis <johannes@langchain.dev> 2026-05-01 14:29:48 -07:00			`"SanitizeToolInputsMiddleware",`
refactor: extract inline middleware from server.py into separate files 2026-02-09 15:34:18 -08:00			`"ToolErrorMiddleware",`
			`"check_message_queue_before_model",`
fix: Middleware to ensure no empty msg (#1010) 2026-03-04 17:59:52 -08:00			`"ensure_no_empty_msg",`
fix: notify users via Slack when agent hits model call step limit (#1204) * fix: notify users via Slack when agent hits model call step limit - Root cause: GraphRecursionError at 1000 steps bypassed all @after_agent middleware including open_pr_if_needed, leaving users with no notification - Change: Added ModelCallLimitMiddleware(run_limit=60) to intercept gracefully before the hard recursion limit, and added notify_step_limit_reached @after_agent middleware to post a Slack thread reply when the limit fires - Verified: 107 existing tests pass, no regressions * fix: harden step-limit Slack notification Ensure the step-limit notification runs after the PR safety net and cover the new middleware behavior with focused unit tests. --------- Co-authored-by: LangSmith Forge <forge-agent@langsmith.ai> Co-authored-by: Johannes du Plessis <johannes@langchain.dev> 2026-05-01 14:24:25 -07:00			`"notify_step_limit_reached",`
refactor: extract inline middleware from server.py into separate files 2026-02-09 15:34:18 -08:00			`"open_pr_if_needed",`
			`]`