execution-agent 0.1.0__tar.gz → 0.1.1__tar.gz

{execution_agent-0.1.0/src/execution_agent.egg-info → execution_agent-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: execution-agent
-Version: 0.1.0
+Version: 0.1.1
 Summary: Automated project building and test execution inside Docker containers
 Author: Islem Bouzenia
 License: MIT License
@@ -64,7 +64,7 @@ Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Requires-Dist: mini-swe-agent
+Requires-Dist: litellm
 Requires-Dist: docker
 Requires-Dist: pydantic>=2.0
 Requires-Dist: pyyaml

{execution_agent-0.1.0 → execution_agent-0.1.1}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "execution-agent"
-version = "0.1.0"
+version = "0.1.1"
 description = "Automated project building and test execution inside Docker containers"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -29,7 +29,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "mini-swe-agent",
+    "litellm",
     "docker",
     "pydantic >= 2.0",
     "pyyaml",

{execution_agent-0.1.0 → execution_agent-0.1.1}/src/execution_agent/__init__.py

@@ -2,7 +2,7 @@
 
 from pathlib import Path
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 package_dir = Path(__file__).resolve().parent
 prompt_files_dir = package_dir / "prompt_files"

execution_agent-0.1.1/src/execution_agent/llm.py

@@ -0,0 +1,26 @@
+"""Lightweight LLM wrapper using litellm for plain text completions."""
+
+from __future__ import annotations
+
+import litellm
+
+
+class LitellmModel:
+    """Simple litellm wrapper that returns plain text responses.
+
+    Compatible with the .query() interface used throughout execution_agent.
+    """
+
+    def __init__(self, model_name: str, model_kwargs: dict | None = None):
+        self.model_name = model_name
+        self.model_kwargs = model_kwargs or {}
+
+    def query(self, messages: list[dict[str, str]], **kwargs) -> dict:
+        merged = {**self.model_kwargs, **kwargs}
+        response = litellm.completion(
+            model=self.model_name,
+            messages=messages,
+            **merged,
+        )
+        content = response.choices[0].message.content or ""
+        return {"content": content}

{execution_agent-0.1.0 → execution_agent-0.1.1}/src/execution_agent/main.py

@@ -53,11 +53,8 @@ signal.signal(signal.SIGINT, _cleanup_on_shutdown)
 signal.signal(signal.SIGTERM, _cleanup_on_shutdown)
 atexit.register(_cleanup_on_shutdown)
 
-import os as _os
-_os.environ.setdefault("MSWEA_SILENT_STARTUP", "1")
-
 from execution_agent.prompt_logging import install_cycle_prompt_logging
-from minisweagent.models.litellm_model import LitellmModel
+from execution_agent.llm import LitellmModel
 
 from execution_agent.agent import ExecutionAgent
 from execution_agent.env import ExecutionEnvironment

{execution_agent-0.1.0 → execution_agent-0.1.1/src/execution_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: execution-agent
-Version: 0.1.0
+Version: 0.1.1
 Summary: Automated project building and test execution inside Docker containers
 Author: Islem Bouzenia
 License: MIT License
@@ -64,7 +64,7 @@ Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Requires-Dist: mini-swe-agent
+Requires-Dist: litellm
 Requires-Dist: docker
 Requires-Dist: pydantic>=2.0
 Requires-Dist: pyyaml

{execution_agent-0.1.0 → execution_agent-0.1.1}/src/execution_agent.egg-info/SOURCES.txt

@@ -11,6 +11,7 @@ src/execution_agent/docker_helpers_static.py
 src/execution_agent/env.py
 src/execution_agent/exceptions.py
 src/execution_agent/exit_artifacts.py
+src/execution_agent/llm.py
 src/execution_agent/main.py
 src/execution_agent/prompt_logging.py
 src/execution_agent/repetition.py

{execution_agent-0.1.0 → execution_agent-0.1.1}/src/execution_agent.egg-info/requires.txt

@@ -1,4 +1,4 @@
-mini-swe-agent
+litellm
 docker
 pydantic>=2.0
 pyyaml