8b46e4f7f9
# PR Description
This PR creates a new toolkit called CodeSandbox. This toolkit has two
tools:
1. `RunCode`: Creates an E2B sandbox and runs the provided code in that
sandbox. Returns the execution logs, result, and errors. Supports
Python, JavaScript, R, Java, and Bash code.
2. `CreateStaticMatplotlibChart`: Creates a sandbox, runs the provided
python code that uses matplotlib, and returns the base64 encoded image
of the chart along with any logs or errors.
- I recommend not using `tool_choice="generate"` since the return object
contains a base64 image can be a lot of tokens that will not provide
much value to a generate's response.
Example of creating a pie chart:
```python
import base64
import json
import os
from openai import OpenAI
def call_tool_with_openai(client: OpenAI) -> dict:
response = client.chat.completions.create(
messages=[
{
"role": "user",
"content": "There are 17 red apples, 4 green apples, and 10 yellow apples. Create a pie chart for this data.",
},
],
model="gpt-4o-mini",
user="you@example.com",
tools=["CodeSandbox.CreateStaticMatplotlibChart"],
tool_choice="execute",
)
return response
arcade_api_key = os.environ.get("ARCADE_API_KEY")
cloud_host = "http://localhost:9099/v1"
openai_client = OpenAI(
api_key=arcade_api_key,
base_url=cloud_host,
)
chat_result = call_tool_with_openai(openai_client)
tool_call_id = chat_result.choices[0].message.tool_calls[0].id
content = json.loads(chat_result.choices[0].message.content)
base64_image = content[tool_call_id]["value"]["base64_image"]
image_data = base64.b64decode(base64_image)
with open("output_image.png", "wb") as image_file:
image_file.write(image_data)
```
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from typing import Annotated
|
|
|
|
from e2b_code_interpreter import Sandbox
|
|
|
|
from arcade.sdk import tool
|
|
from arcade_code_sandbox.tools.models import E2BSupportedLanguage
|
|
from arcade_code_sandbox.tools.utils import get_secret
|
|
|
|
# See https://e2b.dev/docs to learn more about E2B
|
|
|
|
|
|
@tool
|
|
def run_code(
|
|
code: Annotated[str, "The code to run"],
|
|
language: Annotated[
|
|
E2BSupportedLanguage, "The language of the code"
|
|
] = E2BSupportedLanguage.PYTHON,
|
|
) -> Annotated[str, "The sandbox execution as a JSON string"]:
|
|
"""
|
|
Run code in a sandbox and return the output.
|
|
"""
|
|
api_key = get_secret("E2B_API_KEY")
|
|
|
|
with Sandbox(api_key=api_key) as sbx:
|
|
execution = sbx.run_code(code=code, language=language)
|
|
|
|
return execution.to_json()
|
|
|
|
|
|
# Note: Not recommended to use tool_choice='generate' with this tool since it contains base64 encoded image.
|
|
@tool
|
|
def create_static_matplotlib_chart(
|
|
code: Annotated[str, "The Python code to run"],
|
|
) -> Annotated[dict, "A dictionary with the following keys: base64_image, logs, error"]:
|
|
"""
|
|
Run the provided Python code to generate a static matplotlib chart. The resulting chart is returned as a base64 encoded image.
|
|
"""
|
|
api_key = get_secret("E2B_API_KEY")
|
|
|
|
with Sandbox(api_key=api_key) as sbx:
|
|
execution = sbx.run_code(code=code)
|
|
|
|
result = {
|
|
"base64_image": execution.results[0].png if execution.results else None,
|
|
"logs": execution.logs.to_json(),
|
|
"error": execution.error.to_json() if execution.error else None,
|
|
}
|
|
|
|
return result
|