better-notion 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

better_notion/plugins/official/__init__.py

@@ -4,10 +4,12 @@ Official plugins for Better Notion CLI.
 This package contains officially maintained plugins that extend
 the CLI with commonly-needed functionality.
 """
+from better_notion.plugins.official.agents import AgentsPlugin
 from better_notion.plugins.official.productivity import ProductivityPlugin
 
-__all__ = ["ProductivityPlugin"]
+__all__ = ["AgentsPlugin", "ProductivityPlugin"]
 
 OFFICIAL_PLUGINS = [
+    AgentsPlugin,
     ProductivityPlugin,
 ]

better_notion/plugins/official/agents.py

@@ -163,34 +163,36 @@ class AgentsPlugin(PluginInterface):
             try:
                 # Validate role
                 if not RoleManager.is_valid_role(role):
-                    return format_error(
+                    result = format_error(
                         "INVALID_ROLE",
                         f"Invalid role: {role}. Valid roles: {', '.join(RoleManager.get_all_roles())}",
                         retry=False,
                     )
+                else:
+                    # Create .notion file
+                    context = ProjectContext.create(
+                        project_id=project_id,
+                        project_name=project_name,
+                        org_id=org_id,
+                        role=role,
+                        path=Path.cwd(),
+                    )
 
-                # Create .notion file
-                context = ProjectContext.create(
-                    project_id=project_id,
-                    project_name=project_name,
-                    org_id=org_id,
-                    role=role,
-                    path=Path.cwd(),
-                )
-
-                return format_success(
-                    {
-                        "message": "Project initialized successfully",
-                        "project_id": context.project_id,
-                        "project_name": context.project_name,
-                        "org_id": context.org_id,
-                        "role": context.role,
-                        "notion_file": str(Path.cwd() / ".notion"),
-                    }
-                )
+                    result = format_success(
+                        {
+                            "message": "Project initialized successfully",
+                            "project_id": context.project_id,
+                            "project_name": context.project_name,
+                            "org_id": context.org_id,
+                            "role": context.role,
+                            "notion_file": str(Path.cwd() / ".notion"),
+                        }
+                    )
 
             except Exception as e:
-                return format_error("INIT_PROJECT_ERROR", str(e), retry=False)
+                result = format_error("INIT_PROJECT_ERROR", str(e), retry=False)
+
+            typer.echo(result)
 
         # Role management commands
         role_app = typer.Typer(
@@ -219,38 +221,40 @@ class AgentsPlugin(PluginInterface):
             try:
                 # Validate role
                 if not RoleManager.is_valid_role(new_role):
-                    return format_error(
+                    result = format_error(
                         "INVALID_ROLE",
                         f"Invalid role: {new_role}. Valid roles: {', '.join(RoleManager.get_all_roles())}",
                         retry=False,
                     )
-
-                # Load project context
-                if path:
-                    context = ProjectContext.from_path(path)
                 else:
-                    context = ProjectContext.from_current_directory()
-
-                if not context:
-                    return format_error(
-                        "NO_PROJECT_CONTEXT",
-                        "No .notion file found. Are you in a project directory?",
-                        retry=False,
-                    )
-
-                # Update role
-                context.update_role(new_role, path=path or None)
-
-                return format_success(
-                    {
-                        "message": f"Role updated to {new_role}",
-                        "previous_role": context.role,
-                        "new_role": new_role,
-                    }
-                )
+                    # Load project context
+                    if path:
+                        context = ProjectContext.from_path(path)
+                    else:
+                        context = ProjectContext.from_current_directory()
+
+                    if not context:
+                        result = format_error(
+                            "NO_PROJECT_CONTEXT",
+                            "No .notion file found. Are you in a project directory?",
+                            retry=False,
+                        )
+                    else:
+                        # Update role
+                        context.update_role(new_role, path=path or None)
+
+                        result = format_success(
+                            {
+                                "message": f"Role updated to {new_role}",
+                                "previous_role": context.role,
+                                "new_role": new_role,
+                            }
+                        )
 
             except Exception as e:
-                return format_error("ROLE_UPDATE_ERROR", str(e), retry=False)
+                result = format_error("ROLE_UPDATE_ERROR", str(e), retry=False)
+
+            typer.echo(result)
 
         @role_app.command("whoami")
         def role_whoami(
@@ -276,26 +280,28 @@ class AgentsPlugin(PluginInterface):
                     context = ProjectContext.from_current_directory()
 
                 if not context:
-                    return format_error(
+                    result = format_error(
                         "NO_PROJECT_CONTEXT",
                         "No .notion file found. Are you in a project directory?",
                         retry=False,
                     )
+                else:
+                    # Get role description
+                    description = RoleManager.get_role_description(context.role)
 
-                # Get role description
-                description = RoleManager.get_role_description(context.role)
-
-                return format_success(
-                    {
-                        "role": context.role,
-                        "description": description,
-                        "project": context.project_name,
-                        "permissions": RoleManager.get_permissions(context.role),
-                    }
-                )
+                    result = format_success(
+                        {
+                            "role": context.role,
+                            "description": description,
+                            "project": context.project_name,
+                            "permissions": RoleManager.get_permissions(context.role),
+                        }
+                    )
 
             except Exception as e:
-                return format_error("ROLE_ERROR", str(e), retry=False)
+                result = format_error("ROLE_ERROR", str(e), retry=False)
+
+            typer.echo(result)
 
         @role_app.command("list")
         def role_list() -> None:
@@ -322,7 +328,7 @@ class AgentsPlugin(PluginInterface):
                         }
                     )
 
-                return format_success(
+                result = format_success(
                     {
                         "roles": role_info,
                         "total": len(roles),
@@ -330,7 +336,9 @@ class AgentsPlugin(PluginInterface):
                 )
 
             except Exception as e:
-                return format_error("ROLE_LIST_ERROR", str(e), retry=False)
+                result = format_error("ROLE_LIST_ERROR", str(e), retry=False)
+
+            typer.echo(result)
 
         # Register agents app to main CLI
         app.add_typer(agents_app)

{better_notion-1.0.0.dist-info → better_notion-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better-notion
-Version: 1.0.0
+Version: 1.0.1
 Summary: A high-level Python SDK for the Notion API with developer experience in mind. Now with AI agents workflow management system!
 Project-URL: Homepage, https://github.com/nesalia-inc/better-notion
 Project-URL: Documentation, https://github.com/nesalia-inc/better-notion#readme

{better_notion-1.0.0.dist-info → better_notion-1.0.1.dist-info}/RECORD

@@ -105,8 +105,8 @@ better_notion/plugins/__init__.py,sha256=ZbaLy0BS4_zWfEyCW0Xv9EhK427cEhrKIap_8DR
 better_notion/plugins/base.py,sha256=G2asUEvQXWIiyWvNtB1hgg9JXZhDayi7Jvfq15FeQWY,5255
 better_notion/plugins/loader.py,sha256=ewCG_Thv7Khh8IsHLB17U1Hi0cWpq0YG3w37hrpbeRk,7710
 better_notion/plugins/state.py,sha256=jH_tZWvC35hqLO4qwl2Kwq9ziWVavwCEUcCqy3s5wMY,3780
-better_notion/plugins/official/__init__.py,sha256=4c0kaPLOqUQ7vIISgCm4Y9TG306Vew5A8dfDrLsAGwc,327
-better_notion/plugins/official/agents.py,sha256=WyDR1335dt_aGXk2v-WgZJUYCijyAW-gYMDg1ZzA72Y,12220
+better_notion/plugins/official/__init__.py,sha256=rPg5vdk1cEANVstMPzxcWmImtsOpdSR40JSml7h1uUk,426
+better_notion/plugins/official/agents.py,sha256=8Cwx5fkNMaYc1bavS48jyvaXCTMYWgCaIMZH0y9YSbM,12696
 better_notion/plugins/official/productivity.py,sha256=_-whP4pYA4HufE1aUFbIdhrjU-O9njI7xUO_Id2M1J8,8726
 better_notion/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 better_notion/utils/helpers.py,sha256=HgFuUQlG_HzBOB0z2GA9RxPLoXgwRc0DIxa9Fg6C-Jk,2337
@@ -119,8 +119,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.0.0.dist-info/METADATA,sha256=okvm0q4eprZu0_TPkDb8UO7bd8HeJBiFlLRzwLDnV-0,11143
-better_notion-1.0.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_notion-1.0.0.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
-better_notion-1.0.0.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
-better_notion-1.0.0.dist-info/RECORD,,
+better_notion-1.0.1.dist-info/METADATA,sha256=twwihoncbT6FKJ_VmFtC_5g2zXSY12q-YAWFm4hLaqQ,11143
+better_notion-1.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+better_notion-1.0.1.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
+better_notion-1.0.1.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
+better_notion-1.0.1.dist-info/RECORD,,