scalebox-sdk 0.1.25__py3-none-any.whl → 1.0.2__py3-none-any.whl

scalebox/code_interpreter/charts.py

@@ -1,230 +1,230 @@
-import enum
-from typing import Any, List, Optional, Tuple, Union
-
-
-class ChartType(str, enum.Enum):
-    """
-    Chart types
-    """
-
-    LINE = "line"
-    SCATTER = "scatter"
-    BAR = "bar"
-    PIE = "pie"
-    BOX_AND_WHISKER = "box_and_whisker"
-    SUPERCHART = "superchart"
-    UNKNOWN = "unknown"
-
-
-class ScaleType(str, enum.Enum):
-    """
-    Ax scale types
-    """
-
-    LINEAR = "linear"
-    DATETIME = "datetime"
-    CATEGORICAL = "categorical"
-    LOG = "log"
-    SYMLOG = "symlog"
-    LOGIT = "logit"
-    FUNCTION = "function"
-    FUNCTIONLOG = "functionlog"
-    ASINH = "asinh"
-    UNKNOWN = "unknown"
-
-
-class Chart:
-    """
-    Extracted data from a chart. It's useful for building an interactive charts or custom visualizations.
-    """
-
-    type: ChartType
-    title: str
-
-    elements: List[Any]
-
-    def __init__(self, **kwargs) -> None:
-        self._raw_data = kwargs
-        self.type = ChartType(kwargs["type"] or ChartType.UNKNOWN)
-        self.title = kwargs["title"]
-        self.elements = kwargs["elements"]
-
-    def to_dict(self) -> dict:
-        return self._raw_data
-
-
-class Chart2D(Chart):
-    x_label: Optional[str]
-    y_label: Optional[str]
-    x_unit: Optional[str]
-    y_unit: Optional[str]
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self.x_label = kwargs["x_label"]
-        self.y_label = kwargs["y_label"]
-        self.x_unit = kwargs["x_unit"]
-        self.y_unit = kwargs["y_unit"]
-
-
-class PointData:
-    label: str
-    points: List[Tuple[Union[str, float], Union[str, float]]]
-
-    def __init__(self, **kwargs) -> None:
-        self.label = kwargs["label"]
-        self.points = [(x, y) for x, y in kwargs["points"]]
-
-
-class PointChart(Chart2D):
-    x_ticks: List[Union[str, float]]
-    x_tick_labels: List[str]
-    x_scale: ScaleType
-
-    y_ticks: List[Union[str, float]]
-    y_tick_labels: List[str]
-    y_scale: ScaleType
-
-    elements: List[PointData]
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self.x_label = kwargs["x_label"]
-
-        try:
-            self.x_scale = ScaleType(kwargs.get("x_scale"))
-        except ValueError:
-            self.x_scale = ScaleType.UNKNOWN
-
-        self.x_ticks = kwargs["x_ticks"]
-        self.x_tick_labels = kwargs["x_tick_labels"]
-
-        self.y_label = kwargs["y_label"]
-
-        try:
-            self.y_scale = ScaleType(kwargs.get("y_scale"))
-        except ValueError:
-            self.y_scale = ScaleType.UNKNOWN
-
-        self.y_ticks = kwargs["y_ticks"]
-        self.y_tick_labels = kwargs["y_tick_labels"]
-
-        self.elements = [PointData(**d) for d in kwargs["elements"]]
-
-
-class LineChart(PointChart):
-    type = ChartType.LINE
-
-
-class ScatterChart(PointChart):
-    type = ChartType.SCATTER
-
-
-class BarData:
-    label: str
-    group: str
-    value: str
-
-    def __init__(self, **kwargs) -> None:
-        self.label = kwargs["label"]
-        self.value = kwargs["value"]
-        self.group = kwargs["group"]
-
-
-class BarChart(Chart2D):
-    type = ChartType.BAR
-
-    elements: List[BarData]
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self.elements = [BarData(**d) for d in kwargs["elements"]]
-
-
-class PieData:
-    label: str
-    angle: float
-    radius: float
-
-    def __init__(self, **kwargs) -> None:
-        self.label = kwargs["label"]
-        self.angle = kwargs["angle"]
-        self.radius = kwargs["radius"]
-
-
-class PieChart(Chart):
-    type = ChartType.PIE
-
-    elements: List[PieData]
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self.elements = [PieData(**d) for d in kwargs["elements"]]
-
-
-class BoxAndWhiskerData:
-    label: str
-    min: float
-    first_quartile: float
-    median: float
-    third_quartile: float
-    max: float
-    outliers: List[float]
-
-    def __init__(self, **kwargs) -> None:
-        self.label = kwargs["label"]
-        self.min = kwargs["min"]
-        self.first_quartile = kwargs["first_quartile"]
-        self.median = kwargs["median"]
-        self.third_quartile = kwargs["third_quartile"]
-        self.max = kwargs["max"]
-        self.outliers = kwargs.get("outliers") or []
-
-
-class BoxAndWhiskerChart(Chart2D):
-    type = ChartType.BOX_AND_WHISKER
-
-    elements: List[BoxAndWhiskerData]
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self.elements = [BoxAndWhiskerData(**d) for d in kwargs["elements"]]
-
-
-class SuperChart(Chart):
-    type = ChartType.SUPERCHART
-
-    elements: List[
-        Union[LineChart, ScatterChart, BarChart, PieChart, BoxAndWhiskerChart]
-    ]
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self.elements = [_deserialize_chart(g) for g in kwargs["elements"]]
-
-
-ChartTypes = Union[
-    LineChart, ScatterChart, BarChart, PieChart, BoxAndWhiskerChart, SuperChart
-]
-
-
-def _deserialize_chart(data: Optional[dict]) -> Optional[ChartTypes]:
-    if not data:
-        return None
-
-    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:
-        chart = Chart(**data)
-
-    return chart
+import enum
+from typing import Any, List, Optional, Tuple, Union
+
+
+class ChartType(str, enum.Enum):
+    """
+    Chart types
+    """
+
+    LINE = "line"
+    SCATTER = "scatter"
+    BAR = "bar"
+    PIE = "pie"
+    BOX_AND_WHISKER = "box_and_whisker"
+    SUPERCHART = "superchart"
+    UNKNOWN = "unknown"
+
+
+class ScaleType(str, enum.Enum):
+    """
+    Ax scale types
+    """
+
+    LINEAR = "linear"
+    DATETIME = "datetime"
+    CATEGORICAL = "categorical"
+    LOG = "log"
+    SYMLOG = "symlog"
+    LOGIT = "logit"
+    FUNCTION = "function"
+    FUNCTIONLOG = "functionlog"
+    ASINH = "asinh"
+    UNKNOWN = "unknown"
+
+
+class Chart:
+    """
+    Extracted data from a chart. It's useful for building an interactive charts or custom visualizations.
+    """
+
+    type: ChartType
+    title: str
+
+    elements: List[Any]
+
+    def __init__(self, **kwargs) -> None:
+        self._raw_data = kwargs
+        self.type = ChartType(kwargs["type"] or ChartType.UNKNOWN)
+        self.title = kwargs["title"]
+        self.elements = kwargs["elements"]
+
+    def to_dict(self) -> dict:
+        return self._raw_data
+
+
+class Chart2D(Chart):
+    x_label: Optional[str]
+    y_label: Optional[str]
+    x_unit: Optional[str]
+    y_unit: Optional[str]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.x_label = kwargs["x_label"]
+        self.y_label = kwargs["y_label"]
+        self.x_unit = kwargs["x_unit"]
+        self.y_unit = kwargs["y_unit"]
+
+
+class PointData:
+    label: str
+    points: List[Tuple[Union[str, float], Union[str, float]]]
+
+    def __init__(self, **kwargs) -> None:
+        self.label = kwargs["label"]
+        self.points = [(x, y) for x, y in kwargs["points"]]
+
+
+class PointChart(Chart2D):
+    x_ticks: List[Union[str, float]]
+    x_tick_labels: List[str]
+    x_scale: ScaleType
+
+    y_ticks: List[Union[str, float]]
+    y_tick_labels: List[str]
+    y_scale: ScaleType
+
+    elements: List[PointData]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.x_label = kwargs["x_label"]
+
+        try:
+            self.x_scale = ScaleType(kwargs.get("x_scale"))
+        except ValueError:
+            self.x_scale = ScaleType.UNKNOWN
+
+        self.x_ticks = kwargs["x_ticks"]
+        self.x_tick_labels = kwargs["x_tick_labels"]
+
+        self.y_label = kwargs["y_label"]
+
+        try:
+            self.y_scale = ScaleType(kwargs.get("y_scale"))
+        except ValueError:
+            self.y_scale = ScaleType.UNKNOWN
+
+        self.y_ticks = kwargs["y_ticks"]
+        self.y_tick_labels = kwargs["y_tick_labels"]
+
+        self.elements = [PointData(**d) for d in kwargs["elements"]]
+
+
+class LineChart(PointChart):
+    type = ChartType.LINE
+
+
+class ScatterChart(PointChart):
+    type = ChartType.SCATTER
+
+
+class BarData:
+    label: str
+    group: str
+    value: str
+
+    def __init__(self, **kwargs) -> None:
+        self.label = kwargs["label"]
+        self.value = kwargs["value"]
+        self.group = kwargs["group"]
+
+
+class BarChart(Chart2D):
+    type = ChartType.BAR
+
+    elements: List[BarData]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.elements = [BarData(**d) for d in kwargs["elements"]]
+
+
+class PieData:
+    label: str
+    angle: float
+    radius: float
+
+    def __init__(self, **kwargs) -> None:
+        self.label = kwargs["label"]
+        self.angle = kwargs["angle"]
+        self.radius = kwargs["radius"]
+
+
+class PieChart(Chart):
+    type = ChartType.PIE
+
+    elements: List[PieData]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.elements = [PieData(**d) for d in kwargs["elements"]]
+
+
+class BoxAndWhiskerData:
+    label: str
+    min: float
+    first_quartile: float
+    median: float
+    third_quartile: float
+    max: float
+    outliers: List[float]
+
+    def __init__(self, **kwargs) -> None:
+        self.label = kwargs["label"]
+        self.min = kwargs["min"]
+        self.first_quartile = kwargs["first_quartile"]
+        self.median = kwargs["median"]
+        self.third_quartile = kwargs["third_quartile"]
+        self.max = kwargs["max"]
+        self.outliers = kwargs.get("outliers") or []
+
+
+class BoxAndWhiskerChart(Chart2D):
+    type = ChartType.BOX_AND_WHISKER
+
+    elements: List[BoxAndWhiskerData]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.elements = [BoxAndWhiskerData(**d) for d in kwargs["elements"]]
+
+
+class SuperChart(Chart):
+    type = ChartType.SUPERCHART
+
+    elements: List[
+        Union[LineChart, ScatterChart, BarChart, PieChart, BoxAndWhiskerChart]
+    ]
+
+    def __init__(self, **kwargs) -> None:
+        super().__init__(**kwargs)
+        self.elements = [_deserialize_chart(g) for g in kwargs["elements"]]
+
+
+ChartTypes = Union[
+    LineChart, ScatterChart, BarChart, PieChart, BoxAndWhiskerChart, SuperChart
+]
+
+
+def _deserialize_chart(data: Optional[dict]) -> Optional[ChartTypes]:
+    if not data:
+        return None
+
+    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:
+        chart = Chart(**data)
+
+    return chart

scalebox/code_interpreter/code_interpreter_async.py

@@ -365,6 +365,8 @@ class AsyncSandbox(BaseAsyncSandbox):
             # headers = {
             #     "Authorization": "Bearer root",
             # }
-            await client.destroy_context(destroy_context_request,extra_headers=self.connection_config.headers)
+            await client.destroy_context(
+                destroy_context_request, extra_headers=self.connection_config.headers
+            )
         except Exception as e:
             logger.warning(f"Failed to destroy context {context.id}: {e}")

scalebox/code_interpreter/code_interpreter_sync.py

@@ -312,7 +312,9 @@ class Sandbox(BaseSandbox):
             # headers = {
             #     "Authorization": "Bearer root",
             # }
-            client.destroy_context(destroy_context_request,extra_headers=self.connection_config.headers)
+            client.destroy_context(
+                destroy_context_request, extra_headers=self.connection_config.headers
+            )
 
         except Exception as e:
             logger.warning(f"Failed to destroy context {context.id}: {e}")

scalebox/code_interpreter/constants.py

@@ -1,3 +1,3 @@
-DEFAULT_TEMPLATE = "code-interpreter"
-JUPYTER_PORT = 32000
-DEFAULT_TIMEOUT = 300
+DEFAULT_TEMPLATE = "code-interpreter"
+JUPYTER_PORT = 32000
+DEFAULT_TIMEOUT = 300

scalebox/code_interpreter/exceptions.py

@@ -1,13 +1,13 @@
-from ..exceptions import TimeoutException
-
-
-def format_request_timeout_error() -> Exception:
-    return TimeoutException(
-        f"Request timed out — the 'request_timeout' option can be used to increase this timeout",
-    )
-
-
-def format_execution_timeout_error() -> Exception:
-    return TimeoutException(
-        f"Execution timed out — the 'timeout' option can be used to increase this timeout",
-    )
+from ..exceptions import TimeoutException
+
+
+def format_request_timeout_error() -> Exception:
+    return TimeoutException(
+        f"Request timed out — the 'request_timeout' option can be used to increase this timeout",
+    )
+
+
+def format_execution_timeout_error() -> Exception:
+    return TimeoutException(
+        f"Execution timed out — the 'timeout' option can be used to increase this timeout",
+    )