Swekit refactoring (working swe-agent in langgraph and crewai, pr-review-agent in langgraph)

python/composio/tools/local/codeanalysis/tool.py

@@ -18,7 +18,7 @@ class CodeAnalysisTool(LocalTool, autoload=True):
         "tree_sitter==0.21.3",
         "deeplake>3.9,<4",
         "sentence-transformers",
-        "tokenizers>=0.19,<0.20",
+        "tokenizers>=0.20,<0.21",
         "tree_sitter_languages",
         "git+https://github.com/DataDog/jedi.git@92d0c807b0dcd115b1ffd0a4ed21e44db127c2fb#egg=jedi",
         "PyJWT",  # deeplake/client/client.py:41

python/swe/README.md

@@ -98,9 +98,12 @@ Before getting started, ensure you have the following set up:
 1. Scaffold your agent using:
 
    ```
-   swekit scaffold crewai -o <path>
+   swekit scaffold <type> -f <framework> -o <path>
    ```
 
+   - `<type>` can be `swe` or `pr_review` depending on the use-case
+   - `<framework>` can be `crewai` or `langgraph`. Support for more frameworks coming soon!
+
    This creates a new agent in `<path>/agent` with four key files:
 
    - `main.py`: Entry point to run the agent on your issue

python/swe/swekit/cli.py

@@ -17,7 +17,7 @@ def swekit() -> None:
 
 
 @swekit.command(name="scaffold")
-@click.argument("framework", type=EnumParam(cls=AgenticFramework))
+@click.argument("type", type=EnumParam(cls=AgentType))
 @click.option(
     "-n",
     "--name",
@@ -31,11 +31,11 @@ def swekit() -> None:
     help="Output directory for the agent",
 )
 @click.option(
-    "-t",
-    "--type",
-    type=EnumParam(cls=AgentType),
+    "-f",
+    "--framework",
+    type=EnumParam(cls=AgenticFramework),
     help="Type of agent to scaffold, defaults to SWE",
-    default=AgentType.SWE,
+    default=AgenticFramework.LANGGRAPH,
 )
 @click.help_option("--help")
 def _scaffold(

python/swe/swekit/scaffold/_scaffold.py

@@ -23,14 +23,13 @@ class AgenticFramework(Enum):
     LANGGRAPH = "langgraph"
     CAMELAI = "camelai"
     AUTOGEN = "autogen"
-    SWEAGENT = "swe-agent"
 
     def load_templates(self, agent_type: AgentType) -> t.Dict:
         """Load template string."""
         if agent_type == AgentType.SWE:
             return {
                 file.name.replace(".template", ".py"): file.read_text(encoding="utf-8")
-                for file in (TEMPLATES_PATH / self.value).glob("*.template")
+                for file in (TEMPLATES_PATH / "swe" / self.value).glob("*.template")
             }
         elif agent_type == AgentType.PR_REVIEW:
             return {