better-notion 1.3.0__py3-none-any.whl → 1.4.0__py3-none-any.whl

better_notion/plugins/official/agents.py

@@ -343,72 +343,258 @@ class AgentsPlugin(CombinedPluginInterface):
             except Exception as e:
                 return format_error("ROLE_LIST_ERROR", str(e), retry=False)
 
-        # Register agents app to main CLI
+        # Register the agents app to main CLI FIRST
         app.add_typer(agents_app)
 
-        # Import and register CRUD commands as sub-commands of agents
+        # Import CLI functions for CRUD commands
         from better_notion.plugins.official import agents_cli
 
         # Organizations commands (under agents)
         orgs_app = typer.Typer(name="orgs", help="Organizations management commands")
-        orgs_app.command("list")(agents_cli.orgs_list)
-        orgs_app.command("get")(agents_cli.orgs_get)
-        orgs_app.command("create")(agents_cli.orgs_create)
+
+        @orgs_app.command("list")
+        def orgs_list_cmd():
+            typer.echo(agents_cli.orgs_list())
+
+        @orgs_app.command("get")
+        def orgs_get_cmd(org_id: str):
+            typer.echo(agents_cli.orgs_get(org_id))
+
+        @orgs_app.command("create")
+        def orgs_create_cmd(
+            name: str = typer.Option(..., "--name", "-n"),
+            slug: str = typer.Option(..., "--slug"),
+            description: str = typer.Option("", "--description", "-d"),
+            repository_url: str = typer.Option("", "--repository-url"),
+            status: str = typer.Option("Active", "--status"),
+        ):
+            typer.echo(agents_cli.orgs_create(name, slug, description, repository_url, status))
+
         agents_app.add_typer(orgs_app)
 
         # Projects commands (under agents)
         projects_app = typer.Typer(name="projects", help="Projects management commands")
-        projects_app.command("list")(projects_list_with_cli := lambda **kwargs: agents_cli.projects_list(**kwargs))
-        projects_app.command("get")(projects_get_with_cli := lambda project_id: agents_cli.projects_get(project_id))
-        projects_app.command("create")(lambda **kwargs: agents_cli.projects_create(**kwargs))
+
+        @projects_app.command("list")
+        def projects_list_cmd(org_id: str = typer.Option(None, "--org-id", "-o")):
+            typer.echo(agents_cli.projects_list(org_id))
+
+        @projects_app.command("get")
+        def projects_get_cmd(project_id: str):
+            typer.echo(agents_cli.projects_get(project_id))
+
+        @projects_app.command("create")
+        def projects_create_cmd(
+            name: str = typer.Option(..., "--name", "-n"),
+            org_id: str = typer.Option(..., "--org-id", "-o"),
+            slug: str = typer.Option("", "--slug"),
+            description: str = typer.Option("", "--description", "-d"),
+            repository: str = typer.Option("", "--repository"),
+            status: str = typer.Option("Active", "--status"),
+            tech_stack: str = typer.Option("", "--tech-stack"),
+        ):
+            typer.echo(agents_cli.projects_create(name, org_id, slug, description, repository, status, tech_stack))
+
         agents_app.add_typer(projects_app)
 
         # Versions commands (under agents)
         versions_app = typer.Typer(name="versions", help="Versions management commands")
-        versions_app.command("list")(lambda **kwargs: agents_cli.versions_list(**kwargs))
-        versions_app.command("get")(lambda version_id: agents_cli.versions_get(version_id))
-        versions_app.command("create")(lambda **kwargs: agents_cli.versions_create(**kwargs))
+
+        @versions_app.command("list")
+        def versions_list_cmd(project_id: str = typer.Option(None, "--project-id", "-p")):
+            typer.echo(agents_cli.versions_list(project_id))
+
+        @versions_app.command("get")
+        def versions_get_cmd(version_id: str):
+            typer.echo(agents_cli.versions_get(version_id))
+
+        @versions_app.command("create")
+        def versions_create_cmd(
+            name: str = typer.Option(..., "--name", "-n"),
+            project_id: str = typer.Option(..., "--project-id", "-p"),
+            status: str = typer.Option("Planning", "--status"),
+            type_: str = typer.Option("Minor", "--type", "-t"),
+            branch_name: str = typer.Option("", "--branch-name"),
+        ):
+            typer.echo(agents_cli.versions_create(name, project_id, status, type_, branch_name))
+
         agents_app.add_typer(versions_app)
 
         # Tasks commands (under agents)
         tasks_app = typer.Typer(name="tasks", help="Tasks management commands")
-        tasks_app.command("list")(lambda **kwargs: agents_cli.tasks_list(**kwargs))
-        tasks_app.command("get")(lambda task_id: agents_cli.tasks_get(task_id))
-        tasks_app.command("create")(lambda **kwargs: agents_cli.tasks_create(**kwargs))
-        tasks_app.command("next")(lambda **kwargs: agents_cli.tasks_next(**kwargs))
-        tasks_app.command("claim")(lambda task_id: agents_cli.tasks_claim(task_id))
-        tasks_app.command("start")(lambda task_id: agents_cli.tasks_start(task_id))
-        tasks_app.command("complete")(lambda **kwargs: agents_cli.tasks_complete(**kwargs))
-        tasks_app.command("can-start")(lambda task_id: agents_cli.tasks_can_start(task_id))
+
+        @tasks_app.command("list")
+        def tasks_list_cmd(
+            version_id: str = typer.Option(None, "--version-id", "-v"),
+            status: str = typer.Option(None, "--status", "-s"),
+        ):
+            typer.echo(agents_cli.tasks_list(version_id, status))
+
+        @tasks_app.command("get")
+        def tasks_get_cmd(task_id: str):
+            typer.echo(agents_cli.tasks_get(task_id))
+
+        @tasks_app.command("create")
+        def tasks_create_cmd(
+            title: str = typer.Option(..., "--title", "-t"),
+            version_id: str = typer.Option(..., "--version-id", "-v"),
+            type_: str = typer.Option("New Feature", "--type"),
+            priority: str = typer.Option("Medium", "--priority"),
+            estimated_hours: float = typer.Option(0, "--estimated-hours"),
+        ):
+            typer.echo(agents_cli.tasks_create(title, version_id, type_, priority, estimated_hours))
+
+        @tasks_app.command("next")
+        def tasks_next_cmd(project_id: str = typer.Option(None, "--project-id", "-p")):
+            typer.echo(agents_cli.tasks_next(project_id))
+
+        @tasks_app.command("claim")
+        def tasks_claim_cmd(task_id: str):
+            typer.echo(agents_cli.tasks_claim(task_id))
+
+        @tasks_app.command("start")
+        def tasks_start_cmd(task_id: str):
+            typer.echo(agents_cli.tasks_start(task_id))
+
+        @tasks_app.command("complete")
+        def tasks_complete_cmd(
+            task_id: str,
+            actual_hours: float = typer.Option(0, "--actual-hours"),
+        ):
+            typer.echo(agents_cli.tasks_complete(task_id, actual_hours))
+
+        @tasks_app.command("can-start")
+        def tasks_can_start_cmd(task_id: str):
+            typer.echo(agents_cli.tasks_can_start(task_id))
+
         agents_app.add_typer(tasks_app)
 
         # Ideas commands (under agents)
         ideas_app = typer.Typer(name="ideas", help="Ideas management commands")
-        ideas_app.command("list")(lambda **kwargs: agents_cli.ideas_list(**kwargs))
-        ideas_app.command("get")(lambda idea_id: agents_cli.ideas_get(idea_id))
-        ideas_app.command("create")(lambda **kwargs: agents_cli.ideas_create(**kwargs))
-        ideas_app.command("review")(lambda count: agents_cli.ideas_review(count))
-        ideas_app.command("accept")(lambda idea_id: agents_cli.ideas_accept(idea_id))
-        ideas_app.command("reject")(lambda idea_id, reason: agents_cli.ideas_reject(idea_id, reason))
+
+        @ideas_app.command("list")
+        def ideas_list_cmd(
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            status: str = typer.Option(None, "--status", "-s"),
+        ):
+            typer.echo(agents_cli.ideas_list(project_id, status))
+
+        @ideas_app.command("get")
+        def ideas_get_cmd(idea_id: str):
+            typer.echo(agents_cli.ideas_get(idea_id))
+
+        @ideas_app.command("create")
+        def ideas_create_cmd(
+            title: str = typer.Option(..., "--title", "-t"),
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            category: str = typer.Option("Feature", "--category"),
+            description: str = typer.Option("", "--description", "-d"),
+            proposed_solution: str = typer.Option("", "--proposed-solution"),
+            benefits: str = typer.Option("", "--benefits"),
+            effort_estimate: str = typer.Option("Medium", "--effort-estimate"),
+            context: str = typer.Option("", "--context"),
+        ):
+            typer.echo(agents_cli.ideas_create(title, project_id, category, description, proposed_solution, benefits, effort_estimate, context))
+
+        @ideas_app.command("review")
+        def ideas_review_cmd(count: int = typer.Option(10, "--count", "-c")):
+            typer.echo(agents_cli.ideas_review(count))
+
+        @ideas_app.command("accept")
+        def ideas_accept_cmd(idea_id: str):
+            typer.echo(agents_cli.ideas_accept(idea_id))
+
+        @ideas_app.command("reject")
+        def ideas_reject_cmd(
+            idea_id: str,
+            reason: str = typer.Option("", "--reason", "-r"),
+        ):
+            typer.echo(agents_cli.ideas_reject(idea_id, reason))
+
         agents_app.add_typer(ideas_app)
 
         # Work Issues commands (under agents)
         work_issues_app = typer.Typer(name="work-issues", help="Work Issues management commands")
-        work_issues_app.command("list")(lambda **kwargs: agents_cli.work_issues_list(**kwargs))
-        work_issues_app.command("get")(lambda issue_id: agents_cli.work_issues_get(issue_id))
-        work_issues_app.command("create")(lambda **kwargs: agents_cli.work_issues_create(**kwargs))
-        work_issues_app.command("resolve")(lambda issue_id, resolution: agents_cli.work_issues_resolve(issue_id, resolution))
-        work_issues_app.command("blockers")(lambda project_id: agents_cli.work_issues_blockers(project_id))
+
+        @work_issues_app.command("list")
+        def work_issues_list_cmd(
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            status: str = typer.Option(None, "--status", "-s"),
+        ):
+            typer.echo(agents_cli.work_issues_list(project_id, status))
+
+        @work_issues_app.command("get")
+        def work_issues_get_cmd(issue_id: str):
+            typer.echo(agents_cli.work_issues_get(issue_id))
+
+        @work_issues_app.command("create")
+        def work_issues_create_cmd(
+            title: str = typer.Option(..., "--title", "-t"),
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            task_id: str = typer.Option(None, "--task-id"),
+            type_: str = typer.Option("Blocker", "--type"),
+            severity: str = typer.Option("Medium", "--severity"),
+            description: str = typer.Option("", "--description", "-d"),
+            context: str = typer.Option("", "--context"),
+            proposed_solution: str = typer.Option("", "--proposed-solution"),
+        ):
+            typer.echo(agents_cli.work_issues_create(title, project_id, task_id, type_, severity, description, context, proposed_solution))
+
+        @work_issues_app.command("resolve")
+        def work_issues_resolve_cmd(
+            issue_id: str,
+            resolution: str = typer.Option("", "--resolution", "-r"),
+        ):
+            typer.echo(agents_cli.work_issues_resolve(issue_id, resolution))
+
+        @work_issues_app.command("blockers")
+        def work_issues_blockers_cmd(project_id: str):
+            typer.echo(agents_cli.work_issues_blockers(project_id))
+
         agents_app.add_typer(work_issues_app)
 
         # Incidents commands (under agents)
         incidents_app = typer.Typer(name="incidents", help="Incidents management commands")
-        incidents_app.command("list")(lambda **kwargs: agents_cli.incidents_list(**kwargs))
-        incidents_app.command("get")(lambda incident_id: agents_cli.incidents_get(incident_id))
-        incidents_app.command("create")(lambda **kwargs: agents_cli.incidents_create(**kwargs))
-        incidents_app.command("resolve")(lambda incident_id, resolution: agents_cli.incidents_resolve(incident_id, resolution))
-        incidents_app.command("mttr")(lambda **kwargs: agents_cli.incidents_mttr(**kwargs))
-        incidents_app.command("sla-violations")(lambda: agents_cli.incidents_sla_violations())
+
+        @incidents_app.command("list")
+        def incidents_list_cmd(
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            severity: str = typer.Option(None, "--severity", "-s"),
+        ):
+            typer.echo(agents_cli.incidents_list(project_id, severity))
+
+        @incidents_app.command("get")
+        def incidents_get_cmd(incident_id: str):
+            typer.echo(agents_cli.incidents_get(incident_id))
+
+        @incidents_app.command("create")
+        def incidents_create_cmd(
+            title: str = typer.Option(..., "--title", "-t"),
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            affected_version_id: str = typer.Option(None, "--affected-version-id"),
+            severity: str = typer.Option("Medium", "--severity"),
+            type_: str = typer.Option("Bug", "--type"),
+        ):
+            typer.echo(agents_cli.incidents_create(title, project_id, affected_version_id, severity, type_))
+
+        @incidents_app.command("resolve")
+        def incidents_resolve_cmd(
+            incident_id: str,
+            resolution: str = typer.Option("", "--resolution", "-r"),
+        ):
+            typer.echo(agents_cli.incidents_resolve(incident_id, resolution))
+
+        @incidents_app.command("mttr")
+        def incidents_mttr_cmd(
+            project_id: str = typer.Option(None, "--project-id", "-p"),
+            within_days: int = typer.Option(30, "--within-days", "-d"),
+        ):
+            typer.echo(agents_cli.incidents_mttr(project_id, within_days))
+
+        @incidents_app.command("sla-violations")
+        def incidents_sla_violations_cmd():
+            typer.echo(agents_cli.incidents_sla_violations())
+
         agents_app.add_typer(incidents_app)
 
     def register_sdk_models(self) -> dict[str, type]:

better_notion/plugins/official/agents_cli.py

@@ -37,6 +37,27 @@ def get_workspace_config() -> dict:
         return json.load(f)
 
 
+def register_agents_sdk_plugin(client: NotionClient) -> None:
+    """Register the agents SDK plugin with a NotionClient instance.
+
+    This helper function registers all models, caches, and managers
+    for the agents workflow system.
+
+    Args:
+        client: NotionClient instance
+    """
+    from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
+
+    plugin = AgentsSDKPlugin()
+    plugin.initialize(client)
+
+    # Register models, caches, and managers
+    models = plugin.register_models()
+    caches = plugin.register_caches(client)
+    managers = plugin.register_managers(client)
+    client.register_sdk_plugin(models=models, caches=caches, managers=managers)
+
+
 # ===== ORGANIZATIONS =====
 
 def orgs_list() -> str:
@@ -54,7 +75,12 @@ def orgs_list() -> str:
             from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
             plugin = AgentsSDKPlugin()
             plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+
+            # Register models, caches, and managers
+            models = plugin.register_models()
+            caches = plugin.register_caches(client)
+            managers = plugin.register_managers(client)
+            client.register_sdk_plugin(models=models, caches=caches, managers=managers)
 
             # Get manager
             manager = client.plugin_manager("organizations")
@@ -94,10 +120,7 @@ def orgs_get(org_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get organization
             manager = client.plugin_manager("organizations")
@@ -143,10 +166,7 @@ def orgs_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Create organization
             manager = client.plugin_manager("organizations")
@@ -191,10 +211,7 @@ def projects_list(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager
             manager = client.plugin_manager("projects")
@@ -236,10 +253,7 @@ def projects_get(project_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get project
             manager = client.plugin_manager("projects")
@@ -294,10 +308,7 @@ def projects_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Parse tech stack
             tech_stack_list = tech_stack.split(",") if tech_stack else None
@@ -348,10 +359,7 @@ def versions_list(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager
             manager = client.plugin_manager("versions")
@@ -394,10 +402,7 @@ def versions_get(version_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get version
             manager = client.plugin_manager("versions")
@@ -446,10 +451,7 @@ def versions_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Create version
             manager = client.plugin_manager("versions")
@@ -496,10 +498,7 @@ def tasks_list(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager
             manager = client.plugin_manager("tasks")
@@ -542,10 +541,7 @@ def tasks_get(task_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get task
             manager = client.plugin_manager("tasks")
@@ -598,10 +594,7 @@ def tasks_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Parse dependencies
             dependency_ids = dependencies.split(",") if dependencies else None
@@ -655,10 +648,7 @@ def tasks_next(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager
             manager = client.plugin_manager("tasks")
@@ -703,10 +693,7 @@ def tasks_claim(task_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get and claim task
             manager = client.plugin_manager("tasks")
@@ -745,10 +732,7 @@ def tasks_start(task_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get and start task
             manager = client.plugin_manager("tasks")
@@ -800,10 +784,7 @@ def tasks_complete(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get and complete task
             manager = client.plugin_manager("tasks")
@@ -843,10 +824,7 @@ def tasks_can_start(task_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get task and check
             manager = client.plugin_manager("tasks")
@@ -907,10 +885,7 @@ def ideas_list(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and list
             manager = client.plugin_manager("ideas")
@@ -956,10 +931,7 @@ def ideas_get(idea_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get idea
             from better_notion.plugins.official.agents_sdk.models import Idea
@@ -1013,10 +985,7 @@ def ideas_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get workspace config
             workspace_config = get_workspace_config()
@@ -1088,10 +1057,7 @@ def ideas_review(count: int = 10) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and review batch
             manager = client.plugin_manager("ideas")
@@ -1132,10 +1098,7 @@ def ideas_accept(idea_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get idea and accept
             from better_notion.plugins.official.agents_sdk.models import Idea
@@ -1170,10 +1133,7 @@ def ideas_reject(idea_id: str, reason: str = "") -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get idea and reject
             from better_notion.plugins.official.agents_sdk.models import Idea
@@ -1220,10 +1180,7 @@ def work_issues_list(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and list
             manager = client.plugin_manager("work_issues")
@@ -1270,10 +1227,7 @@ def work_issues_get(issue_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get issue
             from better_notion.plugins.official.agents_sdk.models import WorkIssue
@@ -1326,10 +1280,7 @@ def work_issues_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get workspace config
             workspace_config = get_workspace_config()
@@ -1396,10 +1347,7 @@ def work_issues_resolve(issue_id: str, resolution: str = "") -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get issue and resolve
             from better_notion.plugins.official.agents_sdk.models import WorkIssue
@@ -1434,10 +1382,7 @@ def work_issues_blockers(project_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and find blockers
             manager = client.plugin_manager("work_issues")
@@ -1488,10 +1433,7 @@ def incidents_list(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and list
             manager = client.plugin_manager("incidents")
@@ -1537,10 +1479,7 @@ def incidents_get(incident_id: str) -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get incident
             from better_notion.plugins.official.agents_sdk.models import Incident
@@ -1592,10 +1531,7 @@ def incidents_create(
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get workspace config
             workspace_config = get_workspace_config()
@@ -1661,10 +1597,7 @@ def incidents_resolve(incident_id: str, resolution: str = "") -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get incident and resolve
             from better_notion.plugins.official.agents_sdk.models import Incident
@@ -1700,10 +1633,7 @@ def incidents_mttr(project_id: Optional[str] = None, within_days: int = 30) -> s
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and calculate MTTR
             manager = client.plugin_manager("incidents")
@@ -1738,10 +1668,7 @@ def incidents_sla_violations() -> str:
             client = get_client()
 
             # Register SDK plugin
-            from better_notion.plugins.official.agents_sdk.plugin import AgentsSDKPlugin
-            plugin = AgentsSDKPlugin()
-            plugin.initialize(client)
-            client.register_sdk_plugin(plugin)
+            register_agents_sdk_plugin(client)
 
             # Get manager and find violations
             manager = client.plugin_manager("incidents")

{better_notion-1.3.0.dist-info → better_notion-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better-notion
-Version: 1.3.0
+Version: 1.4.0
 Summary: A high-level Python SDK for the Notion API with developer experience in mind.
 Project-URL: Homepage, https://github.com/nesalia-inc/better-notion
 Project-URL: Documentation, https://github.com/nesalia-inc/better-notion#readme

{better_notion-1.3.0.dist-info → better_notion-1.4.0.dist-info}/RECORD

@@ -107,8 +107,8 @@ better_notion/plugins/base.py,sha256=3h9jOZzS--UqmVW3RREtcQ2h1GTWWPUryTencsJKhTM
 better_notion/plugins/loader.py,sha256=zCWsMdJyvZs1IHFm0zjEiqm_l_5jB1Uw4x30Kq8rLS4,9527
 better_notion/plugins/state.py,sha256=jH_tZWvC35hqLO4qwl2Kwq9ziWVavwCEUcCqy3s5wMY,3780
 better_notion/plugins/official/__init__.py,sha256=rPg5vdk1cEANVstMPzxcWmImtsOpdSR40JSml7h1uUk,426
-better_notion/plugins/official/agents.py,sha256=Co-P4unrxNkbm_wYcUHor6WC_neoyHKSkeFyPbEAyVw,19488
-better_notion/plugins/official/agents_cli.py,sha256=rFF3cIXXzxOnvd5dnaR1CESvs1mwmR1hBv1lVADcIu8,56317
+better_notion/plugins/official/agents.py,sha256=UF9zn7DNBX-3_bmpjrHhbNd4dZXQIGa5ljhCC71SRcA,25939
+better_notion/plugins/official/agents_cli.py,sha256=v2yem60ER6EVRt9u1f5xF3MOExQeQ4lGqGCQ6z7pV18,51706
 better_notion/plugins/official/productivity.py,sha256=_-whP4pYA4HufE1aUFbIdhrjU-O9njI7xUO_Id2M1J8,8726
 better_notion/plugins/official/agents_sdk/__init__.py,sha256=luQBzZLsJ7fC5U0jFu8dfzMviiXj2SBZXcTohM53wkQ,725
 better_notion/plugins/official/agents_sdk/managers.py,sha256=dkpGozJhKXvz8rFqc2tAWyL6MeVzFVcjGMC0Ed2Z1dQ,29813
@@ -125,8 +125,8 @@ better_notion/utils/agents/rbac.py,sha256=8ZA8Y7wbOiVZDbpjpH7iC35SZrZ0jl4fcJ3xWC
 better_notion/utils/agents/schemas.py,sha256=e_lpGGO12FXtfqFyI91edj9xc5RUtuA6pU7Sk6ip7xg,21784
 better_notion/utils/agents/state_machine.py,sha256=xUBEeDTbU1Xq-rsRo2sbr6AUYpYrV9DTHOtZT2cWES8,6699
 better_notion/utils/agents/workspace.py,sha256=T8mP_DNIWhI1-k6UydJINsEJ6kQxQ6_Pa44ul58k088,12370
-better_notion-1.3.0.dist-info/METADATA,sha256=MP9kADRDZFuIjs6PQ4EHuTK8HfNNMYehK8MC9JckJKg,11096
-better_notion-1.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_notion-1.3.0.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
-better_notion-1.3.0.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
-better_notion-1.3.0.dist-info/RECORD,,
+better_notion-1.4.0.dist-info/METADATA,sha256=WXldrn_zOf9rVjRSQJmWLLr_ixIdAOtZxGW2BhWNmZc,11096
+better_notion-1.4.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+better_notion-1.4.0.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
+better_notion-1.4.0.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
+better_notion-1.4.0.dist-info/RECORD,,