mirror of
https://github.com/johndoe6345789/AutoMetabuilder.git
synced 2026-04-24 13:54:59 +00:00
Fix remaining imports from utils in engine and tools modules
Co-authored-by: johndoe6345789 <224850594+johndoe6345789@users.noreply.github.com>
This commit is contained in:
@@ -1,5 +1,12 @@
|
||||
"""Build workflow runtime context."""
|
||||
from ..utils.model_resolver import resolve_model_name
|
||||
import os
|
||||
|
||||
DEFAULT_MODEL = "openai/gpt-4o"
|
||||
|
||||
|
||||
def resolve_model_name(prompt: dict) -> str:
|
||||
"""Resolve model name from env or prompt."""
|
||||
return os.environ.get("LLM_MODEL", prompt.get("model", DEFAULT_MODEL))
|
||||
|
||||
|
||||
def build_workflow_context(parts: dict) -> dict:
|
||||
|
||||
@@ -1,6 +1,42 @@
|
||||
"""Run a task inside Docker."""
|
||||
import subprocess
|
||||
import os
|
||||
from ..utils.docker_utils import run_command_in_docker
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger("autometabuilder.docker")
|
||||
|
||||
|
||||
def run_command_in_docker(image: str, command: str, volumes: dict = None, workdir: str = None):
|
||||
"""
|
||||
Run a command inside a Docker container.
|
||||
|
||||
:param image: Docker image to use.
|
||||
:param command: Command to execute.
|
||||
:param volumes: Dictionary of volume mappings {host_path: container_path}.
|
||||
:param workdir: Working directory inside the container.
|
||||
:return: Standard output of the command.
|
||||
"""
|
||||
docker_command = ["docker", "run", "--rm"]
|
||||
|
||||
if volumes:
|
||||
for host_path, container_path in volumes.items():
|
||||
docker_command.extend(["-v", f"{os.path.abspath(host_path)}:{container_path}"])
|
||||
|
||||
if workdir:
|
||||
docker_command.extend(["-w", workdir])
|
||||
|
||||
docker_command.append(image)
|
||||
docker_command.extend(["sh", "-c", command])
|
||||
|
||||
logger.info(f"Executing in Docker ({image}): {command}")
|
||||
result = subprocess.run(docker_command, capture_output=True, text=True, check=False)
|
||||
|
||||
output = result.stdout
|
||||
if result.stderr:
|
||||
output += "\n" + result.stderr
|
||||
|
||||
logger.info(output)
|
||||
return output
|
||||
|
||||
|
||||
def run_docker_task(image: str, command: str, workdir: str = "/workspace") -> str:
|
||||
|
||||
Reference in New Issue
Block a user