mini-swe-agent 1.4.0__py3-none-any.whl → 1.4.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mini-swe-agent
3
- Version: 1.4.0
3
+ Version: 1.4.1
4
4
  Summary: Nano SWE Agent - A simple AI software engineering agent
5
5
  Author-email: Kilian Lieret <kilian.lieret@posteo.de>, "Carlos E. Jimenez" <carlosej@princeton.edu>
6
6
  License: MIT License
@@ -1,10 +1,10 @@
1
- mini_swe_agent-1.4.0.dist-info/licenses/LICENSE.md,sha256=D3luWPkdHAe7LBsdD4vzqDAXw6Xewb3G-uczss0uh1s,1094
2
- minisweagent/__init__.py,sha256=3w6Aygz0dJetlfY8zRfpQ4rZecN_3XvpkNLJubSeGjA,1787
1
+ mini_swe_agent-1.4.1.dist-info/licenses/LICENSE.md,sha256=D3luWPkdHAe7LBsdD4vzqDAXw6Xewb3G-uczss0uh1s,1094
2
+ minisweagent/__init__.py,sha256=Fao4tu0VXB-yMYWz02SQVn7_hu-zj9am7sgP6swBWzs,1787
3
3
  minisweagent/__main__.py,sha256=FIyAOiw--c3FQ2g240FOM1FdL0lk_PxSpixu0pQ7WFo,194
4
4
  minisweagent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  minisweagent/agents/__init__.py,sha256=cpjJLzg1IGxLM-tZpoMJV9S33ye13XtdBO0x7DU_Lrk,48
6
- minisweagent/agents/default.py,sha256=6TZUfKch6e_7m05BXmjqUjV-k068s6ldg7T6zJBbQW8,5438
7
- minisweagent/agents/interactive.py,sha256=_DCBabdwdIR4gAojT_TaQW2MSFtmBq997mwmiDGYdRA,7327
6
+ minisweagent/agents/default.py,sha256=9LxgX3KWHFGkGE4FjGOZwK5oA8ykzg9KgmpgyqlEuoo,5460
7
+ minisweagent/agents/interactive.py,sha256=7HW2cffaV5f66DIjxvtIbL8mo_S5aZSwgNLSmHp6VC0,7450
8
8
  minisweagent/agents/interactive_textual.py,sha256=-GrhQT3nh_bn4qyWFRMf6F7xEqMsRnQ-Ry5n3JJBjsc,16077
9
9
  minisweagent/config/README.md,sha256=tPruhnQDhZ8ugc1FNPKk9tVMRltmmIjdYgvHCmN-3Hs,354
10
10
  minisweagent/config/__init__.py,sha256=UfORdQID1Ek_dduZlybUsIKJjihImkSqNU5tIjpw0hk,694
@@ -22,7 +22,7 @@ minisweagent/environments/extra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm
22
22
  minisweagent/environments/extra/swerex_docker.py,sha256=MOhhFdX1sAk_U0g-GOxohfjrldzO4YfrUnHq8qJff7c,1502
23
23
  minisweagent/models/__init__.py,sha256=J4bnvfMByTVG0cL_6p51sm8gdargXhARfbG5c0UZ8Z4,2890
24
24
  minisweagent/models/anthropic.py,sha256=D8nHvvbgzPjla0He8p0O9kaXASPWg1Sai0pHsAj_Yn8,855
25
- minisweagent/models/litellm_model.py,sha256=jMWsFYPPoXgBaN6Ypa5MR6A9hOpvacnYEXZkzuAkgNc,2283
25
+ minisweagent/models/litellm_model.py,sha256=2-Q7gEoKK-2kbUsfDxrTBxIQasEqw1raE_1Xk28XXUc,2289
26
26
  minisweagent/models/test_models.py,sha256=oB3jmZUire5TkVT8ebUCD3jLuLhPIbcTiTqdIix85Yw,1174
27
27
  minisweagent/models/utils/cache_control.py,sha256=mG9cE56HQaUwXfoqvXoH6LcbMV_G1vlEE1aBBpikXYg,1608
28
28
  minisweagent/models/utils/key_per_thread.py,sha256=Vlxt--rapNNYCgIHrMCu1WVAkuiVIhC_awbarkbnkZQ,644
@@ -39,8 +39,8 @@ minisweagent/run/extra/swebench_single.py,sha256=L3Kk4G65o3MCPLMEwGNIs77-AFf6Lfc
39
39
  minisweagent/run/extra/utils/batch_progress.py,sha256=u__khJ-fipZLxTJu43LamGAtPUCqEZYEi8J7SfH7X6A,6211
40
40
  minisweagent/run/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
41
41
  minisweagent/run/utils/save.py,sha256=q7omf7zYHg73k8-Iyp9w5YVSYvDAacRrh4X9L_4VhNM,942
42
- mini_swe_agent-1.4.0.dist-info/METADATA,sha256=tM-cskbK7S2xIRUegGm39qXgPT3tOfpfZOlOZfk-lcg,13459
43
- mini_swe_agent-1.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
44
- mini_swe_agent-1.4.0.dist-info/entry_points.txt,sha256=d1_yRbTaGjs1UXHa6JQK0sKDGBIVGm8oeW0k2kfbJgQ,182
45
- mini_swe_agent-1.4.0.dist-info/top_level.txt,sha256=zKF4t8bFpV87fdVABZt2Da-vnb4Vkh_CxkwQx5YT4Ew,13
46
- mini_swe_agent-1.4.0.dist-info/RECORD,,
42
+ mini_swe_agent-1.4.1.dist-info/METADATA,sha256=wbuaTIHIypAE-u6bBYi2uXnmtXRv96vVmognFS6WG2U,13459
43
+ mini_swe_agent-1.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
44
+ mini_swe_agent-1.4.1.dist-info/entry_points.txt,sha256=d1_yRbTaGjs1UXHa6JQK0sKDGBIVGm8oeW0k2kfbJgQ,182
45
+ mini_swe_agent-1.4.1.dist-info/top_level.txt,sha256=zKF4t8bFpV87fdVABZt2Da-vnb4Vkh_CxkwQx5YT4Ew,13
46
+ mini_swe_agent-1.4.1.dist-info/RECORD,,
minisweagent/__init__.py CHANGED
@@ -8,7 +8,7 @@ This file provides:
8
8
  unless you want the static type checking.
9
9
  """
10
10
 
11
- __version__ = "1.4.0"
11
+ __version__ = "1.4.1"
12
12
 
13
13
  import os
14
14
  from pathlib import Path
@@ -72,7 +72,7 @@ class DefaultAgent:
72
72
  def run(self, task: str) -> tuple[str, str]:
73
73
  """Run step() until agent is finished. Return exit status & message"""
74
74
  self.messages = []
75
- self.add_message("system", self.config.system_template)
75
+ self.add_message("system", self.render_template(self.config.system_template))
76
76
  self.add_message("user", self.render_template(self.config.instance_template, task=task))
77
77
  while True:
78
78
  try:
@@ -35,8 +35,8 @@ class InteractiveAgentConfig(AgentConfig):
35
35
  class InteractiveAgent(DefaultAgent):
36
36
  _MODE_COMMANDS_MAPPING = {"/u": "human", "/c": "confirm", "/y": "yolo"}
37
37
 
38
- def __init__(self, *args, **kwargs):
39
- super().__init__(*args, config_class=InteractiveAgentConfig, **kwargs)
38
+ def __init__(self, *args, config_class=InteractiveAgentConfig, **kwargs):
39
+ super().__init__(*args, config_class=config_class, **kwargs)
40
40
  self.cost_last_confirmed = 0.0
41
41
 
42
42
  def add_message(self, role: str, content: str):
@@ -59,7 +59,9 @@ class InteractiveAgent(DefaultAgent):
59
59
  case "/y" | "/c": # Just go to the super query, which queries the LM for the next action
60
60
  pass
61
61
  case _:
62
- return {"content": f"\n```bash\n{command}\n```"}
62
+ msg = {"content": f"\n```bash\n{command}\n```"}
63
+ self.add_message("assistant", msg["content"])
64
+ return msg
63
65
  try:
64
66
  with console.status("Waiting for the LM to respond..."):
65
67
  return super().query()
@@ -31,7 +31,7 @@ class LitellmModel:
31
31
  self.cost = 0.0
32
32
  self.n_calls = 0
33
33
  if self.config.litellm_model_registry is not None:
34
- litellm.utils.register_model(json.loads(self.config.litellm_model_registry.read_text()))
34
+ litellm.utils.register_model(json.loads(Path(self.config.litellm_model_registry).read_text()))
35
35
 
36
36
  @retry(
37
37
  stop=stop_after_attempt(10),