e2b-code-interpreter 0.0.11b46__tar.gz → 1.0.0__tar.gz

e2b_code_interpreter-1.0.0/PKG-INFO

@@ -0,0 +1,150 @@
+Metadata-Version: 2.1
+Name: e2b-code-interpreter
+Version: 1.0.0
+Summary: E2B Code Interpreter - Stateful code execution
+Home-page: https://e2b.dev/
+License: Apache-2.0
+Author: e2b
+Author-email: hello@e2b.dev
+Requires-Python: >=3.8,<4.0
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: attrs (>=21.3.0)
+Requires-Dist: e2b (>=1.0.0,<2.0.0)
+Requires-Dist: httpx (>=0.20.0,<0.28.0)
+Project-URL: Bug Tracker, https://github.com/e2b-dev/code-interpreter/issues
+Project-URL: Repository, https://github.com/e2b-dev/e2b-code-interpreter/tree/python
+Description-Content-Type: text/markdown
+
+<p align="center">
+  <img width="100" src="/readme-assets/logo-circle.png" alt="e2b logo">
+</p>
+
+
+<h1 align="center">
+  Code interpreter extension for Python
+</h1>
+
+<h4 align="center">
+  <a href="https://pypi.org/project/e2b/">
+    <img alt="Last 1 month downloads for the Python SDK" loading="lazy" width="200" height="20" decoding="async" data-nimg="1"
+    style="color:transparent;width:auto;height:100%" src="https://img.shields.io/pypi/dm/e2b?label=PyPI%20Downloads">
+  </a>
+  <a href="https://www.npmjs.com/package/e2b">
+    <img alt="Last 1 month downloads for the Python SDK" loading="lazy" width="200" height="20" decoding="async" data-nimg="1"
+    style="color:transparent;width:auto;height:100%" src="https://img.shields.io/npm/dm/e2b?label=NPM%20Downloads">
+  </a>
+</h4>
+
+The repository contains a template and modules for the code interpreter sandbox. It is based on the Jupyter server and implements the Jupyter Kernel messaging protocol. This allows for sharing context between code executions and improves support for plotting charts and other display-able data.
+
+## Key Features
+
+- **Stateful Execution**: Unlike traditional sandboxes that treat each code execution independently, this package maintains context across executions.
+- **Displaying Charts & Data**: Implements parts of the [Jupyter Kernel messaging protocol](https://jupyter-client.readthedocs.io/en/latest/messaging.html), which support for interactive features like plotting charts, rendering DataFrames, etc.
+
+## Installation
+
+```sh
+pip install e2b-code-interpreter
+```
+
+## Examples
+
+### Minimal example with the sharing context
+
+```python
+from e2b_code_interpreter import Sandbox
+
+with Sandbox() as sandbox:
+    sandbox.run_code()("x = 1")
+
+    execution = sandbox.run_code()("x+=1; x")
+    print(execution.text)  # outputs 2
+```
+
+### Get charts and any display-able data
+
+```python
+import base64
+import io
+
+from matplotlib import image as mpimg, pyplot as plt
+
+from e2b_code_interpreter import Sandbox
+
+code = """
+import matplotlib.pyplot as plt
+import numpy as np
+
+x = np.linspace(0, 20, 100)
+y = np.sin(x)
+
+plt.plot(x, y)
+plt.show()
+"""
+
+with Sandbox() as sandbox:
+    # you can install dependencies in "jupyter notebook style"
+    sandbox.run_code("!pip install matplotlib")
+
+    # plot random chart
+    execution = sandbox.run_code(code)
+
+# there's your image
+image = execution.results[0].png
+
+# example how to show the image / prove it works
+i = base64.b64decode(image)
+i = io.BytesIO(i)
+i = mpimg.imread(i, format='PNG')
+
+plt.imshow(i, interpolation='nearest')
+plt.show()
+```
+
+### Streaming code output
+
+```python
+from e2b_code_interpreter import Sandbox
+
+code = """
+import time
+import pandas as pd
+
+print("hello")
+time.sleep(3)
+data = pd.DataFrame(data=[[1, 2], [3, 4]], columns=["A", "B"])
+display(data.head(10))
+time.sleep(3)
+print("world")
+"""
+
+with Sandbox() as sandbox:
+    sandbox.run_code(code, on_stdout=print, on_stderr=print, on_result=(lambda result: print(result.text)))
+```
+
+### More resources
+- Check out the [JavaScript/TypeScript](https://e2b.dev/docs/hello-world/js) and [Python](https://e2b.dev/docs/hello-world/py) "Hello World" guides to learn how to use our SDK.
+
+- See [E2B documentation](https://e2b.dev/docs) to get started.
+
+- Visit our [Cookbook](https://github.com/e2b-dev/e2b-cookbook/tree/main) to get inspired by examples with different LLMs and AI frameworks.
+
+___
+
+<div align='center'>
+<!-- <a href="https://e2b.dev/docs" target="_blank">
+<img src="https://img.shields.io/badge/docs-%2300acee.svg?color=143D52&style=for-the-badge&logo=x&logoColor=white" alt=docs style="margin-bottom: 5px;"/></a>  -->
+<a href="https://twitter.com/e2b_dev" target="_blank">
+<img src="https://img.shields.io/badge/x (twitter)-%2300acee.svg?color=000000&style=for-the-badge&logo=x&logoColor=white" alt=linkedin style="margin-bottom: 5px;"/></a> 
+<a href="https://discord.com/invite/U7KEcGErtQ" target="_blank">
+<img src="https://img.shields.io/badge/discord -%2300acee.svg?color=143D52&style=for-the-badge&logo=discord&logoColor=white" alt=discord style="margin-bottom: 5px;"/></a> 
+<a href="https://www.linkedin.com/company/e2b-dev/" target="_blank">
+<img src="https://img.shields.io/badge/linkedin-%2300acee.svg?color=000000&style=for-the-badge&logo=linkedin&logoColor=white" alt=linkedin style="margin-bottom: 5px;"/></a> 
+</div align='center'>

e2b_code_interpreter-1.0.0/README.md

@@ -0,0 +1,127 @@
+<p align="center">
+  <img width="100" src="/readme-assets/logo-circle.png" alt="e2b logo">
+</p>
+
+
+<h1 align="center">
+  Code interpreter extension for Python
+</h1>
+
+<h4 align="center">
+  <a href="https://pypi.org/project/e2b/">
+    <img alt="Last 1 month downloads for the Python SDK" loading="lazy" width="200" height="20" decoding="async" data-nimg="1"
+    style="color:transparent;width:auto;height:100%" src="https://img.shields.io/pypi/dm/e2b?label=PyPI%20Downloads">
+  </a>
+  <a href="https://www.npmjs.com/package/e2b">
+    <img alt="Last 1 month downloads for the Python SDK" loading="lazy" width="200" height="20" decoding="async" data-nimg="1"
+    style="color:transparent;width:auto;height:100%" src="https://img.shields.io/npm/dm/e2b?label=NPM%20Downloads">
+  </a>
+</h4>
+
+The repository contains a template and modules for the code interpreter sandbox. It is based on the Jupyter server and implements the Jupyter Kernel messaging protocol. This allows for sharing context between code executions and improves support for plotting charts and other display-able data.
+
+## Key Features
+
+- **Stateful Execution**: Unlike traditional sandboxes that treat each code execution independently, this package maintains context across executions.
+- **Displaying Charts & Data**: Implements parts of the [Jupyter Kernel messaging protocol](https://jupyter-client.readthedocs.io/en/latest/messaging.html), which support for interactive features like plotting charts, rendering DataFrames, etc.
+
+## Installation
+
+```sh
+pip install e2b-code-interpreter
+```
+
+## Examples
+
+### Minimal example with the sharing context
+
+```python
+from e2b_code_interpreter import Sandbox
+
+with Sandbox() as sandbox:
+    sandbox.run_code()("x = 1")
+
+    execution = sandbox.run_code()("x+=1; x")
+    print(execution.text)  # outputs 2
+```
+
+### Get charts and any display-able data
+
+```python
+import base64
+import io
+
+from matplotlib import image as mpimg, pyplot as plt
+
+from e2b_code_interpreter import Sandbox
+
+code = """
+import matplotlib.pyplot as plt
+import numpy as np
+
+x = np.linspace(0, 20, 100)
+y = np.sin(x)
+
+plt.plot(x, y)
+plt.show()
+"""
+
+with Sandbox() as sandbox:
+    # you can install dependencies in "jupyter notebook style"
+    sandbox.run_code("!pip install matplotlib")
+
+    # plot random chart
+    execution = sandbox.run_code(code)
+
+# there's your image
+image = execution.results[0].png
+
+# example how to show the image / prove it works
+i = base64.b64decode(image)
+i = io.BytesIO(i)
+i = mpimg.imread(i, format='PNG')
+
+plt.imshow(i, interpolation='nearest')
+plt.show()
+```
+
+### Streaming code output
+
+```python
+from e2b_code_interpreter import Sandbox
+
+code = """
+import time
+import pandas as pd
+
+print("hello")
+time.sleep(3)
+data = pd.DataFrame(data=[[1, 2], [3, 4]], columns=["A", "B"])
+display(data.head(10))
+time.sleep(3)
+print("world")
+"""
+
+with Sandbox() as sandbox:
+    sandbox.run_code(code, on_stdout=print, on_stderr=print, on_result=(lambda result: print(result.text)))
+```
+
+### More resources
+- Check out the [JavaScript/TypeScript](https://e2b.dev/docs/hello-world/js) and [Python](https://e2b.dev/docs/hello-world/py) "Hello World" guides to learn how to use our SDK.
+
+- See [E2B documentation](https://e2b.dev/docs) to get started.
+
+- Visit our [Cookbook](https://github.com/e2b-dev/e2b-cookbook/tree/main) to get inspired by examples with different LLMs and AI frameworks.
+
+___
+
+<div align='center'>
+<!-- <a href="https://e2b.dev/docs" target="_blank">
+<img src="https://img.shields.io/badge/docs-%2300acee.svg?color=143D52&style=for-the-badge&logo=x&logoColor=white" alt=docs style="margin-bottom: 5px;"/></a>  -->
+<a href="https://twitter.com/e2b_dev" target="_blank">
+<img src="https://img.shields.io/badge/x (twitter)-%2300acee.svg?color=000000&style=for-the-badge&logo=x&logoColor=white" alt=linkedin style="margin-bottom: 5px;"/></a> 
+<a href="https://discord.com/invite/U7KEcGErtQ" target="_blank">
+<img src="https://img.shields.io/badge/discord -%2300acee.svg?color=143D52&style=for-the-badge&logo=discord&logoColor=white" alt=discord style="margin-bottom: 5px;"/></a> 
+<a href="https://www.linkedin.com/company/e2b-dev/" target="_blank">
+<img src="https://img.shields.io/badge/linkedin-%2300acee.svg?color=000000&style=for-the-badge&logo=linkedin&logoColor=white" alt=linkedin style="margin-bottom: 5px;"/></a> 
+</div align='center'>

{e2b_code_interpreter-0.0.11b46 → e2b_code_interpreter-1.0.0}/e2b_code_interpreter/__init__.py

@@ -2,6 +2,7 @@ from e2b import *
 from .code_interpreter_sync import Sandbox
 from .code_interpreter_async import AsyncSandbox
 from .models import (
+    Context,
     Execution,
     ExecutionError,
     Result,

e2b_code_interpreter-0.0.11b46/e2b_code_interpreter/graphs.py → e2b_code_interpreter-1.0.0/e2b_code_interpreter/charts.py

@@ -2,9 +2,9 @@ import enum
 from typing import List, Tuple, Any, Optional, Union
 
 
-class GraphType(str, enum.Enum):
+class ChartType(str, enum.Enum):
     """
-    Graph types
+    Chart types
     """
 
     LINE = "line"
@@ -12,7 +12,7 @@ class GraphType(str, enum.Enum):
     BAR = "bar"
     PIE = "pie"
     BOX_AND_WHISKER = "box_and_whisker"
-    SUPERGRAPH = "supergraph"
+    SUPERCHART = "superchart"
     UNKNOWN = "unknown"
 
 
@@ -33,23 +33,23 @@ class ScaleType(str, enum.Enum):
     UNKNOWN = "unknown"
 
 
-class Graph:
+class Chart:
     """
-    Extracted data from a graph. It's useful for building an interactive graphs or custom visualizations.
+    Extracted data from a chart. It's useful for building an interactive charts or custom visualizations.
     """
 
-    type: GraphType
+    type: ChartType
     title: str
 
     elements: List[Any]
 
     def __init__(self, **kwargs):
-        self.type = GraphType(kwargs["type"] or GraphType.UNKNOWN)
+        self.type = ChartType(kwargs["type"] or ChartType.UNKNOWN)
         self.title = kwargs["title"]
         self.elements = kwargs["elements"]
 
 
-class Graph2D(Graph):
+class Chart2D(Chart):
     x_label: Optional[str]
     y_label: Optional[str]
     x_unit: Optional[str]
@@ -72,7 +72,7 @@ class PointData:
         self.points = [(x, y) for x, y in kwargs["points"]]
 
 
-class PointGraph(Graph2D):
+class PointChart(Chart2D):
     x_ticks: List[Union[str, float]]
     x_tick_labels: List[str]
     x_scale: ScaleType
@@ -108,12 +108,12 @@ class PointGraph(Graph2D):
         self.elements = [PointData(**d) for d in kwargs["elements"]]
 
 
-class LineGraph(PointGraph):
-    type = GraphType.LINE
+class LineChart(PointChart):
+    type = ChartType.LINE
 
 
-class ScatterGraph(PointGraph):
-    type = GraphType.SCATTER
+class ScatterChart(PointChart):
+    type = ChartType.SCATTER
 
 
 class BarData:
@@ -127,8 +127,8 @@ class BarData:
         self.group = kwargs["group"]
 
 
-class BarGraph(Graph2D):
-    type = GraphType.BAR
+class BarChart(Chart2D):
+    type = ChartType.BAR
 
     elements: List[BarData]
 
@@ -148,8 +148,8 @@ class PieData:
         self.radius = kwargs["radius"]
 
 
-class PieGraph(Graph):
-    type = GraphType.PIE
+class PieChart(Chart):
+    type = ChartType.PIE
 
     elements: List[PieData]
 
@@ -175,8 +175,8 @@ class BoxAndWhiskerData:
         self.max = kwargs["max"]
 
 
-class BoxAndWhiskerGraph(Graph2D):
-    type = GraphType.BOX_AND_WHISKER
+class BoxAndWhiskerChart(Chart2D):
+    type = ChartType.BOX_AND_WHISKER
 
     elements: List[BoxAndWhiskerData]
 
@@ -185,40 +185,40 @@ class BoxAndWhiskerGraph(Graph2D):
         self.elements = [BoxAndWhiskerData(**d) for d in kwargs["elements"]]
 
 
-class SuperGraph(Graph):
-    type = GraphType.SUPERGRAPH
+class SuperChart(Chart):
+    type = ChartType.SUPERCHART
 
     elements: List[
-        Union[LineGraph, ScatterGraph, BarGraph, PieGraph, BoxAndWhiskerGraph]
+        Union[LineChart, ScatterChart, BarChart, PieChart, BoxAndWhiskerChart]
     ]
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
-        self.elements = [deserialize_graph(g) for g in kwargs["elements"]]
+        self.elements = [_deserialize_chart(g) for g in kwargs["elements"]]
 
 
-GraphTypes = Union[
-    LineGraph, ScatterGraph, BarGraph, PieGraph, BoxAndWhiskerGraph, SuperGraph
+ChartTypes = Union[
+    LineChart, ScatterChart, BarChart, PieChart, BoxAndWhiskerChart, SuperChart
 ]
 
 
-def deserialize_graph(data: Optional[dict]) -> Optional[GraphTypes]:
+def _deserialize_chart(data: Optional[dict]) -> Optional[ChartTypes]:
     if not data:
         return None
 
-    if data["type"] == GraphType.LINE:
-        graph = LineGraph(**data)
-    elif data["type"] == GraphType.SCATTER:
-        graph = ScatterGraph(**data)
-    elif data["type"] == GraphType.BAR:
-        graph = BarGraph(**data)
-    elif data["type"] == GraphType.PIE:
-        graph = PieGraph(**data)
-    elif data["type"] == GraphType.BOX_AND_WHISKER:
-        graph = BoxAndWhiskerGraph(**data)
-    elif data["type"] == GraphType.SUPERGRAPH:
-        graph = SuperGraph(**data)
+    if data["type"] == ChartType.LINE:
+        chart = LineChart(**data)
+    elif data["type"] == ChartType.SCATTER:
+        chart = ScatterChart(**data)
+    elif data["type"] == ChartType.BAR:
+        chart = BarChart(**data)
+    elif data["type"] == ChartType.PIE:
+        chart = PieChart(**data)
+    elif data["type"] == ChartType.BOX_AND_WHISKER:
+        chart = BoxAndWhiskerChart(**data)
+    elif data["type"] == ChartType.SUPERCHART:
+        chart = SuperChart(**data)
     else:
-        graph = Graph(**data)
+        chart = Chart(**data)
 
-    return graph
+    return chart