mini-swe-agent 1.17.5__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

{mini_swe_agent-1.17.5.dist-info → mini_swe_agent-2.0.0a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.17.5
+Version: 2.0.0a1
 Summary: Nano SWE Agent - A simple AI software engineering agent
 Author-email: Kilian Lieret <kilian.lieret@posteo.de>, "Carlos E. Jimenez" <carlosej@princeton.edu>
 License: MIT License
@@ -40,6 +40,7 @@ License-File: LICENSE.md
 Requires-Dist: pyyaml
 Requires-Dist: requests
 Requires-Dist: jinja2
+Requires-Dist: pydantic>=2.0
 Requires-Dist: litellm>=1.75.5
 Requires-Dist: tenacity
 Requires-Dist: rich
@@ -48,12 +49,16 @@ Requires-Dist: typer
 Requires-Dist: platformdirs
 Requires-Dist: textual
 Requires-Dist: prompt_toolkit
+Requires-Dist: datasets
 Requires-Dist: openai!=1.100.0,!=1.100.1
 Provides-Extra: full
 Requires-Dist: mini-swe-agent[dev]; extra == "full"
 Requires-Dist: swe-rex>=1.4.0; extra == "full"
+Requires-Dist: mini-swe-agent[modal]; extra == "full"
+Provides-Extra: modal
+Requires-Dist: modal; extra == "modal"
+Requires-Dist: boto3; extra == "modal"
 Provides-Extra: dev
-Requires-Dist: datasets; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
@@ -67,14 +72,16 @@ Requires-Dist: mkdocs-material; extra == "dev"
 Requires-Dist: mkdocs-glightbox; extra == "dev"
 Requires-Dist: mkdocs-redirects; extra == "dev"
 Requires-Dist: portkey-ai; extra == "dev"
+Requires-Dist: swe-rex; extra == "dev"
 Dynamic: license-file
 
 <div align="center">
 <a href="https://mini-swe-agent.com/latest/"><img src="https://github.com/SWE-agent/mini-swe-agent/raw/main/docs/assets/mini-swe-agent-banner.svg" alt="mini-swe-agent banner" style="height: 7em"/></a>
 </div>
 
-# The 100 line AI agent that solves GitHub issues & more
+# The minimal AI software engineering agent
 
+📣 [New tutorial on building minimal AI agents](https://minimal-agent.com/)<br/>
 📣 [Gemini 3 Pro reaches 74% on SWE-bench verified with mini-swe-agent!](https://x.com/KLieret/status/1991164693839270372)<br/>
 📣 [New blogpost: Randomly switching between GPT-5 and Sonnet 4 boosts performance](https://www.swebench.com/SWE-bench/blog/2025/08/19/mini-roulette/)
 
@@ -82,9 +89,12 @@ Dynamic: license-file
 [![Slack](https://img.shields.io/badge/Slack-4A154B?style=for-the-badge&logo=slack&logoColor=white)](https://join.slack.com/t/swe-bench/shared_invite/zt-36pj9bu5s-o3_yXPZbaH2wVnxnss1EkQ)
 [![PyPI - Version](https://img.shields.io/pypi/v/mini-swe-agent?style=for-the-badge&logo=python&logoColor=white&labelColor=black&color=deeppink)](https://pypi.org/project/mini-swe-agent/)
 
-In 2024, [SWE-bench](https://github.com/swe-bench/SWE-bench) & [SWE-agent](https://github.com/swe-agent/swe-agent) helped kickstart the coding agent revolution.
+> [!WARNING]
+> This is **mini-swe-agent v2**. For the previous version, check out the [v1 branch](https://github.com/SWE-agent/mini-swe-agent/tree/v1).
 
-We now ask: **What if SWE-agent was 100x smaller, and still worked nearly as well?**
+In 2024, we built [SWE-bench](https://github.com/swe-bench/SWE-bench) & [SWE-agent](https://github.com/swe-agent/swe-agent) and helped kickstart the coding agent revolution.
+
+We now ask: **What if our agent was 100x smaller, and still worked nearly as well?**
 
 The `mini` agent is for
 
@@ -94,11 +104,11 @@ The `mini` agent is for
 
 Here's some details:
 
-- **Minimal**: Just [100 lines of python](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/agents/default.py) (+100 total for [env](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/environments/local.py),
-[model](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/models/litellm_model.py), [script](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/run/hello_world.py)) — no fancy dependencies!
-- **Performant:** Scores >74% on the [SWE-bench verified benchmark](https://www.swebench.com/) benchmark; starts faster than Claude Code
+- **Minimal**: Just some 100 lines of python for the [agent class](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/agents/default.py) (and a bit more for the [environment](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/environments/local.py),
+[model](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/models/litellm_model.py), and [run script](https://github.com/SWE-agent/mini-swe-agent/blob/main/src/minisweagent/run/hello_world.py)) — no fancy dependencies!
+- **Performant:** Scores >74% on the [SWE-bench verified benchmark](https://www.swebench.com/) benchmark; starts much faster than Claude Code
 - **Deployable:** In addition to local envs, you can use **docker**, **podman**, **singularity**, **apptainer**, and more
-- **Cutting edge:** Built by the Princeton & Stanford team behind [SWE-bench](https://swebench.com) and [SWE-agent](https://swe-agent.com).
+- Built by the Princeton & Stanford team behind [SWE-bench](https://swebench.com), [SWE-agent](https://swe-agent.com), and more (see below)
 - **Widely adopted:** In use by Meta, NVIDIA, Essential AI, Anyscale, and others
 - **Tested:** [![Codecov](https://img.shields.io/codecov/c/github/swe-agent/mini-swe-agent?style=flat-square)](https://codecov.io/gh/SWE-agent/mini-swe-agent)
 
@@ -110,7 +120,7 @@ Here's some details:
 However, one year later, as LMs have become more capable, a lot of this is not needed at all to build a useful agent!
 In fact, the `mini` agent
 
-- **Does not have any tools other than bash** — it doesn't even use the tool-calling interface of the LMs.
+- **Does not have any tools other than bash** — it doesn't even need to use the tool-calling interface of the LMs.
   This means that you can run it with literally any model. When running in sandboxed environments you also don't need to take care
   of installing a single package — all it needs is bash.
 - **Has a completely linear history** — every step of the agent just appends to the messages and that's it.
@@ -139,7 +149,7 @@ The `mini` agent wants to be a hackable tool, not a black box.
 
 Unlike other agents (including our own [swe-agent](https://swe-agent.com/latest/)), it is radically simpler, because it:
 
-- **Does not have any tools other than bash** — it doesn't even use the tool-calling interface of the LMs.
+- **Does not have any tools other than bash** — it doesn't even need to use the tool-calling interface of the LMs.
   Instead of implementing custom tools for every specific thing the agent might want to do, the focus is fully on the LM utilizing the shell to its full potential.
   Want it to do something specific like opening a PR?
   Just tell the LM to figure it out rather than spending time to implement it in the agent.
@@ -176,57 +186,40 @@ What you get with both
 <table>
 <tr>
 <td width="50%">
-<a href="https://mini-swe-agent.com/latest/usage/mini/"><strong>Simple UI</strong></a> (<code>mini</code>)
+<a href="https://mini-swe-agent.com/latest/usage/mini/"><strong>CLI</strong></a> (<code>mini</code>)
 </td>
 <td>
-<a href="https://mini-swe-agent.com/latest/usage/mini_v/"><strong>Visual UI</strong></a> (<code>mini -v</code>)
+<a href="https://mini-swe-agent.com/latest/usage/swebench/"><strong>Batch inference</strong></a>
 </td>
 </tr>
 <tr>
 <td width="50%">
 
-  ![mini](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/mini.gif?raw=true)
+![mini](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/mini.gif?raw=true)
 
 </td>
 <td>
 
-  ![miniv](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/mini2.gif?raw=true)
+![swebench](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/swebench.gif?raw=true)
 
 </td>
 </tr>
 <tr>
-  <td>
-    <a href="https://mini-swe-agent.com/latest/usage/swebench/"><strong>Batch inference</strong></a>
-  </td>
-  <td>
-    <a href="https://mini-swe-agent.com/latest/usage/inspector/"><strong>Trajectory browser</strong></a>
-  </td>
-<tr>
-<tr>
-
 <td>
-
-![swebench](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/swebench.gif?raw=true)
-
+<a href="https://mini-swe-agent.com/latest/usage/inspector/"><strong>Trajectory browser</strong></a>
 </td>
-
-<td>
-
-![inspector](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/inspector.gif?raw=true)
-
-</td>
-
-</tr>
 <td>
 <a href="https://mini-swe-agent.com/latest/advanced/cookbook/"><strong>Python bindings</strong></a>
 </td>
-<td>
-<a href="https://mini-swe-agent.com"><strong>More in the docs</strong></a>
-</td>
 </tr>
 <tr>
 <td>
 
+![inspector](https://github.com/SWE-agent/swe-agent-media/blob/main/media/mini/gif/inspector.gif?raw=true)
+
+</td>
+<td>
+
 ```python
 agent = DefaultAgent(
     LitellmModel(model_name=...),
@@ -234,15 +227,6 @@ agent = DefaultAgent(
 )
 agent.run("Write a sudoku game")
 ```
-</td>
-<td>
-
-* [Quick start](https://mini-swe-agent.com/latest/quickstart/)
-* [`mini`](https://mini-swe-agent.com/latest/usage/mini/)
-* [FAQ](https://mini-swe-agent.com/latest/faq/)
-* [Global configuration](https://mini-swe-agent.com/latest/advanced/global_configuration/)
-* [Yaml configuration files](https://mini-swe-agent.com/latest/advanced/yaml_configuration/)
-* [Power up](https://mini-swe-agent.com/latest/advanced/cookbook/)
 
 </td>
 </tr>
@@ -253,16 +237,16 @@ agent.run("Write a sudoku game")
 **Option 1:** If you just want to try out the CLI (package installed in anonymous virtual environment)
 
 ```bash
-pip install uv && uvx mini-swe-agent [-v]
+pip install uv && uvx mini-swe-agent
 # or
-pip install pipx && pipx ensurepath && pipx run mini-swe-agent [-v]
+pip install pipx && pipx ensurepath && pipx run mini-swe-agent
 ```
 
 **Option 2:** Install CLI & python bindings in current environment
 
 ```bash
 pip install mini-swe-agent
-mini -v  # run the CLI
+mini  # run the CLI
 ```
 
 **Option 3:** Install from source (developer setup)
@@ -270,13 +254,13 @@ mini -v  # run the CLI
 ```bash
 git clone https://github.com/SWE-agent/mini-swe-agent.git
 cd mini-swe-agent && pip install -e .
-mini [-v]  # run the CLI
+mini  # run the CLI
 ```
 
 Read more in our [documentation](https://mini-swe-agent.com/latest/):
 
 * [Quick start guide](https://mini-swe-agent.com/latest/quickstart/)
-* More on [`mini`](https://mini-swe-agent.com/latest/usage/mini/) and [`mini -v`](https://mini-swe-agent.com/latest/usage/mini_v/)
+* [Using the `mini` CLI](https://mini-swe-agent.com/latest/usage/mini/)
 * [Global configuration](https://mini-swe-agent.com/latest/advanced/global_configuration/)
 * [Yaml configuration files](https://mini-swe-agent.com/latest/advanced/yaml_configuration/)
 * [Power up with the cookbook](https://mini-swe-agent.com/latest/advanced/cookbook/)

mini_swe_agent-2.0.0a1.dist-info/RECORD

@@ -0,0 +1,70 @@
+mini_swe_agent-2.0.0a1.dist-info/licenses/LICENSE.md,sha256=D3luWPkdHAe7LBsdD4vzqDAXw6Xewb3G-uczss0uh1s,1094
+minisweagent/__init__.py,sha256=zlvjrkG3KEpusrSrfjcbgi9YK6hDVnTBcqpttTBub2U,2263
+minisweagent/__main__.py,sha256=FIyAOiw--c3FQ2g240FOM1FdL0lk_PxSpixu0pQ7WFo,194
+minisweagent/exceptions.py,sha256=ETVDXEet3ykSMBVmqH4egdG3gcz4_yPN5JRgjqcNeow,619
+minisweagent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/agents/__init__.py,sha256=cpjJLzg1IGxLM-tZpoMJV9S33ye13XtdBO0x7DU_Lrk,48
+minisweagent/agents/default.py,sha256=W7LvWCqs22U8nVEEhTCDxa8ok0fHNofOZiVYfVmuHRo,6361
+minisweagent/agents/interactive.py,sha256=_ZRyhb61rmuNWJQKMFNChicrXpzCk7iu-zrizy29G04,10129
+minisweagent/config/README.md,sha256=FDC6xu91wny_g4ps5BMMunTliCZRgZTfncU0krPxrHo,250
+minisweagent/config/__init__.py,sha256=czwg5fZboB84Y96zFyyrBA9SNiwoAfxIzDiZnOHvwG8,1868
+minisweagent/config/default.yaml,sha256=Chn6Jo6vcjc8mx6TgQJvk3sq02oIH3j7IjITa6BqCZA,5318
+minisweagent/config/inspector.tcss,sha256=8fXckGLTWmO1vP3aU4DuathviZjbakF7tkCX9HswyLA,494
+minisweagent/config/mini.yaml,sha256=ToT8BEZOrEl3R1-zVIXVQswBiW4lQfhuu6BzSF25rko,4700
+minisweagent/config/mini_textbased.yaml,sha256=0lgaAIMf0HDG0g679ENORXX6oOGlHbDz0IN7yt-5yds,5185
+minisweagent/config/benchmarks/__init__.py,sha256=e1MoAlDn_wc9HnXNoncf1P-B4DQ-iRf6n7Q_txjZGRI,52
+minisweagent/config/benchmarks/swebench.yaml,sha256=tN7WgvDk-HmPSrqFBZruvHRXKwua0HUfnPqFBuZH0KY,6210
+minisweagent/config/benchmarks/swebench_backticks.yaml,sha256=t2CaRixM7tY_B_LWGPxawDAapOMMbscAGjwB8eB8lFs,8592
+minisweagent/config/benchmarks/swebench_modal.yaml,sha256=ogb0euCQSJ1ksfPHkSzezZRr2JAD1uRVAFl6HrBKqiE,1406
+minisweagent/config/benchmarks/swebench_xml.yaml,sha256=S2k5IuKnPFrqGo-gsWHelKke6802WM1Vm6eezZ61Ui8,8738
+minisweagent/environments/__init__.py,sha256=PI6o8gWW8q6eghkQKKchtjudSZ7nPJBiyt5IATz8aNU,1357
+minisweagent/environments/docker.py,sha256=IdOJEisAjnt0PR5kAgf9oLCNkBDPYyPd0C8HQrM7lqY,6427
+minisweagent/environments/local.py,sha256=kzhuEsGiTYwwoS8X3VmkiNNPIY6WmeaLhOdbq-kBn70,3018
+minisweagent/environments/singularity.py,sha256=BTtnr-kHj0k21YeBNsHILrb0GNXig_-qRUjdo8Ieh6k,5362
+minisweagent/environments/extra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/environments/extra/bubblewrap.py,sha256=h3Lu-8SxBspdj59B20kOWPlBMkTX4QPYb_zcYh1vhNM,5220
+minisweagent/environments/extra/swerex_docker.py,sha256=LyAEVCR6rymdhR1tsChp_uiOElTAtPHnnywqiZtln9g,3198
+minisweagent/environments/extra/swerex_modal.py,sha256=LTLbWRUI1wI0jvhBheElG-qM3T_ASAryP3DcAGAcM1s,3464
+minisweagent/models/__init__.py,sha256=6xQerjRtUa9Ae0-vhorX6Gn6T_8nVBU4H6jvhFYW81E,4466
+minisweagent/models/litellm_model.py,sha256=f7ZiOD_7ZxevX3XdwfkHMdxvmcD_8S3eG4_70YXFigM,6868
+minisweagent/models/litellm_response_model.py,sha256=VyYdPRlW0Oq22wb-ir1Fx_jzWigbJxn6Z_AVNacvtp4,3190
+minisweagent/models/litellm_textbased_model.py,sha256=OeQm--3QltRm0HulkxkNS__Or3LE7ra8_MTEmWagChE,2007
+minisweagent/models/openrouter_model.py,sha256=5E0sGX69KM0QOSyKFObLnbuJtX7_6bEd896FiBYGpBA,7321
+minisweagent/models/openrouter_response_model.py,sha256=ngAiM4lpyz0DA9Yy3QAB5M1grcss-789kYIj1PfFc-Q,4897
+minisweagent/models/openrouter_textbased_model.py,sha256=AO80uN8owW0nwmoO3GIbWpxYOHm6KU7RZepFeHHq0W4,3101
+minisweagent/models/portkey_model.py,sha256=4uEeuJGLcOAG2UiLmlqywgtxOp0QG6YN8R_kfJSzfdk,9535
+minisweagent/models/portkey_response_model.py,sha256=tP1wJN4CU-9BsMM3qm5SyoT1iFzVu6Mk5UoJgt84TD4,6607
+minisweagent/models/requesty_model.py,sha256=Ir5kudErHDOI0XVRxAuGz7jnpqwg9rx3G5F4Jixjnq8,6727
+minisweagent/models/test_models.py,sha256=RDMGhXOfR-VUuQ-nPM8Bm8LiuA5MKHwCX2PqT1Jjsj4,10728
+minisweagent/models/extra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/models/extra/roulette.py,sha256=dKqMG0OJCXAxlEOCt0YUV4dZAuuYyKp2EGCgfY-y4J8,2147
+minisweagent/models/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/models/utils/actions_text.py,sha256=7Lka_-heukp6_ovEJh1mmFm9JDuDPFRvTCVXTH0HlXk,2166
+minisweagent/models/utils/actions_toolcall.py,sha256=oqEHq6-NX-BFoFxvsdqaXqBeViK5ufKBvqNiqCTy2k8,3637
+minisweagent/models/utils/actions_toolcall_response.py,sha256=dD_IDUmfe-sqw3VxtDHuFwPV2P8U-9RMGbTN05ja0VM,4234
+minisweagent/models/utils/anthropic_utils.py,sha256=E1XFRtfSF1AvSutbqHUT9Vcg-NTSwSR7j8XXkweNFw4,1189
+minisweagent/models/utils/cache_control.py,sha256=7buJUhldHZ2-9F7XytkQpQsc3sCBdd502__JQU89pbo,2365
+minisweagent/models/utils/content_string.py,sha256=thJAlIMF-MiJufmoKoP1Bx-zojYLg2DM-Fv0fRhcEp0,2941
+minisweagent/models/utils/openai_multimodal.py,sha256=xJjsLpfAv0XCEqEN6fXRstJNTeRqBf0Orql1wavj-B8,1865
+minisweagent/models/utils/retry.py,sha256=-53Z2coCU6KyrIaQf-pMNP51xVr21EN4HhqwIimhrBg,860
+minisweagent/run/__init__.py,sha256=WIoYgHVl7iZF2YncrfV3IttupG6P5KogroKHKECka3A,38
+minisweagent/run/hello_world.py,sha256=4dKyWeVxYWfjBmzkBtkeUdZ7Qy3UnUc0LhMG0l9c8vY,1157
+minisweagent/run/mini.py,sha256=CAirxzjssE4oGI_llmquuahF4Y09yeKiodsRCgmmv5w,4094
+minisweagent/run/benchmarks/__init__.py,sha256=x7CKf8dhQnID87rP3_1KfuDRnmLI2GnNq3F0SWLvkKw,66
+minisweagent/run/benchmarks/swebench.py,sha256=XLGoXG4LkQH4Lhh5hE2Q6PEVDMS0tKmv2yR97bm_bMk,12663
+minisweagent/run/benchmarks/swebench_single.py,sha256=ec8jdgLf-QoBmQEoYIH49tFSOTmb34AFCXcETN1ST5o,4005
+minisweagent/run/benchmarks/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/run/benchmarks/utils/batch_progress.py,sha256=PgCHttoXcg6UyNmTzYDVdMS59Bp4oJbWJIk5E8PaF9k,6808
+minisweagent/run/extra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/run/utilities/__init__.py,sha256=T7dF84NeQoKgJwPURNy3F4CCrxici-7AwGiyV0ZsDlo,79
+minisweagent/run/utilities/config.py,sha256=JF2BYoUm4bpRJmkn5mV95I8FaHJUYfMXPheQ9RnQwP8,3916
+minisweagent/run/utilities/inspector.py,sha256=fa19C-u_qnhwFbByFgx338FSaPbBR6aACRBCVCoG1Rs,10174
+minisweagent/run/utilities/mini_extra.py,sha256=VmIifsg6whXZ7h88tP5zzkKITFOSMAztpxr8FMCmyBg,1596
+minisweagent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minisweagent/utils/log.py,sha256=ruDMNKMrVC9NPvCeHwO3QYz5jsVNUGQB2dRAEAPAWp8,996
+minisweagent/utils/serialize.py,sha256=ZWE2TgAkJdF6H42r0zIMjKtYrwCLBgRoLpG4TZuocPs,761
+mini_swe_agent-2.0.0a1.dist-info/METADATA,sha256=zAzJ5rogTph7-11VIXaZv6JxAprubfcRJmWMAu0qKLE,14474
+mini_swe_agent-2.0.0a1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mini_swe_agent-2.0.0a1.dist-info/entry_points.txt,sha256=h9itmW6xlkZJd9rqSmvsLAEHHG2f6DWoGCGXoHwBErE,202
+mini_swe_agent-2.0.0a1.dist-info/top_level.txt,sha256=zKF4t8bFpV87fdVABZt2Da-vnb4Vkh_CxkwQx5YT4Ew,13
+mini_swe_agent-2.0.0a1.dist-info/RECORD,,

mini_swe_agent-2.0.0a1.dist-info/entry_points.txt

@@ -0,0 +1,5 @@
+[console_scripts]
+mini = minisweagent.run.mini:app
+mini-e = minisweagent.run.utilities.mini_extra:main
+mini-extra = minisweagent.run.utilities.mini_extra:main
+mini-swe-agent = minisweagent.run.mini:app

minisweagent/__init__.py

@@ -8,7 +8,7 @@ This file provides:
   unless you want the static type checking.
 """
 
-__version__ = "1.17.5"
+__version__ = "2.0.0a1"
 
 import os
 from pathlib import Path
@@ -17,32 +17,18 @@ from typing import Any, Protocol
 import dotenv
 from platformdirs import user_config_dir
 from rich.console import Console
-from rich.panel import Panel
 
 from minisweagent.utils.log import logger
 
 package_dir = Path(__file__).resolve().parent
 
+
 global_config_dir = Path(os.getenv("MSWEA_GLOBAL_CONFIG_DIR") or user_config_dir("mini-swe-agent"))
 global_config_dir.mkdir(parents=True, exist_ok=True)
 global_config_file = Path(global_config_dir) / ".env"
 
 if not os.getenv("MSWEA_SILENT_STARTUP"):
-    console = Console()
-    console.print(
-        Panel(
-            "[bold]mini-swe-agent v2.0 is coming soon.[/bold]\n\n"
-            "It will be even more flexible, extensible and performant. "
-            "However, [bold]breaking changes[/bold] were necessary.\n\n"
-            "To stay with the current major version for now, you can pin your dependency: "
-            "[bold white]mini-swe-agent~=1.0[/bold white]",
-            border_style="bold red",
-            title="BREAKING CHANGES AHEAD",
-            title_align="left",
-            expand=False,
-        )
-    )
-    console.print(
+    Console().print(
         f"👋 This is [bold green]mini-swe-agent[/bold green] version [bold green]{__version__}[/bold green].\n"
         f"Loading global config from [bold green]'{global_config_file}'[/bold green]"
     )
@@ -57,12 +43,18 @@ class Model(Protocol):
     """Protocol for language models."""
 
     config: Any
-    cost: float
-    n_calls: int
 
     def query(self, messages: list[dict[str, str]], **kwargs) -> dict: ...
 
-    def get_template_vars(self) -> dict[str, Any]: ...
+    def format_message(self, **kwargs) -> dict: ...
+
+    def format_observation_messages(
+        self, message: dict, outputs: list[dict], template_vars: dict | None = None
+    ) -> list[dict]: ...
+
+    def get_template_vars(self, **kwargs) -> dict[str, Any]: ...
+
+    def serialize(self) -> dict: ...
 
 
 class Environment(Protocol):
@@ -70,20 +62,21 @@ class Environment(Protocol):
 
     config: Any
 
-    def execute(self, command: str, cwd: str = "") -> dict[str, str]: ...
+    def execute(self, action: dict, cwd: str = "") -> dict[str, Any]: ...
+
+    def get_template_vars(self, **kwargs) -> dict[str, Any]: ...
 
-    def get_template_vars(self) -> dict[str, Any]: ...
+    def serialize(self) -> dict: ...
 
 
 class Agent(Protocol):
     """Protocol for agents."""
 
-    model: Model
-    env: Environment
-    messages: list[dict[str, str]]
     config: Any
 
-    def run(self, task: str, **kwargs) -> tuple[str, str]: ...
+    def run(self, task: str, **kwargs) -> dict: ...
+
+    def save(self, path: Path | None, *extra_dicts) -> dict: ...
 
 
 __all__ = [