Haystack + Declaw

pip install declaw python-dotenv haystack-ai

from haystack import component
from declaw import Sandbox

@component
class DecawCodeRunner:
    """Haystack component that executes Python code in a Declaw sandbox."""

    @component.output_types(output=str, exit_code=int)
    def run(self, code: str) -> dict:
        sbx = Sandbox.create(template="python", timeout=300)
        try:
            sbx.files.write("/tmp/code.py", code)
            result = sbx.commands.run("python3 /tmp/code.py", timeout=30)
            return {"output": result.stdout, "exit_code": result.exit_code}
        finally:
            sbx.kill()

from haystack import Pipeline
from haystack.components.generators import OpenAIGenerator
from haystack.components.builders.prompt_builder import PromptBuilder

template = (
    "Write Python code (and nothing else) to solve: {{ query }}\n"
    "Output only the Python code, no markdown fences."
)

pipeline = Pipeline()
pipeline.add_component("prompt_builder", PromptBuilder(template=template))
pipeline.add_component("llm", OpenAIGenerator(model="gpt-4o-mini"))
pipeline.add_component("code_runner", DecawCodeRunner())

pipeline.connect("prompt_builder", "llm")
pipeline.connect("llm.replies", "code_runner.code")

query = "Calculate the factorial of 15 and print it"
result = pipeline.run({"prompt_builder": {"query": query}})
print(result["code_runner"]["output"])
print(result["code_runner"]["exit_code"])

from declaw import Sandbox

code = """\
import math
n = 15
print(f"{n}! = {math.factorial(n)}")
"""

sbx = Sandbox.create(template="python", timeout=300)
try:
    sbx.files.write("/tmp/code.py", code)
    result = sbx.commands.run("python3 /tmp/code.py", timeout=30)
    print(result.stdout)
finally:
    sbx.kill()

=======================================================
Haystack + Declaw Sandbox Example
=======================================================
No OPENAI_API_KEY found -- running demo mode.

--- Component Definition ---
@component
class DecawCodeRunner:
    @component.output_types(output=str, exit_code=int)
    def run(self, code: str) -> dict:
        sbx = Sandbox.create(template="python", timeout=300)
        ...

--- Running Code Directly in Declaw Sandbox ---
Code:
import math
n = 15
print(f"{n}! = {math.factorial(n)}")

stdout: 15! = 1307674368000
stderr:
exit_code: 0

Sandbox cleaned up.

=======================================================
Done!
=======================================================