```diff
--- test_files/107-original.txt	2025-03-07 19:06:21
+++ test_files/107-modified.txt	2025-03-07 19:06:21
@@ -4,19 +4,20 @@
 from pathlib import Path
 
 import click
-from swekit.exceptions import SWEKitError
-from swekit.scaffold import AgenticFramework, scaffold
 
 from composio.cli.utils.params import EnumParam, PathParam
 
+from swekit.exceptions import SWEKitError
+from swekit.scaffold import AgentType, AgenticFramework, scaffold
 
+
 @click.group(name="swekit")
 def swekit() -> None:
     """Composio Coder CLI for managing the coding workspace and tasks."""
 
 
 @swekit.command(name="scaffold")
-@click.argument("framework", type=EnumParam(cls=AgenticFramework))
+@click.argument("type", type=EnumParam(cls=AgentType))
 @click.option(
     "-n",
     "--name",
@@ -29,11 +30,19 @@
     type=PathParam(),
     help="Output directory for the agent",
 )
+@click.option(
+    "-f",
+    "--framework",
+    type=EnumParam(cls=AgenticFramework),
+    help="Type of agent to scaffold, defaults to SWE",
+    default=AgenticFramework.LANGGRAPH,
+)
 @click.help_option("--help")
 def _scaffold(
     framework: AgenticFramework,
     name: t.Optional[str] = None,
     outdir: t.Optional[Path] = None,
+    type: AgentType = AgentType.SWE,
 ) -> None:
     """🤖 Scaffold agent using composio toolset."""
     try:
@@ -41,6 +50,7 @@
             framework=framework,
             name=name,
             outdir=outdir,
+            agent_type=type,
         )
         click.echo(f"🤖 Scaffolded agent @ {output}")
     except SWEKitError as e:
```
