beamlit 0.0.29rc32__py3-none-any.whl → 0.0.30rc33__py3-none-any.whl

beamlit/common/__init__.py

@@ -1,6 +1,7 @@
+from .error import HTTPError
 from .logger import init as init_logger
 from .secrets import Secret
-from .settings import Settings, get_settings, init_agent
+from .settings import Settings, get_settings, init, init_agent
 from .utils import copy_folder
 
 __all__ = [
@@ -8,6 +9,8 @@ __all__ = [
     "Settings",
     "get_settings",
     "init_agent",
+    "init",
     "copy_folder",
     "init_logger",
+    "HTTPError",
 ]

beamlit/common/error.py

@@ -0,0 +1,9 @@
+class HTTPError(Exception):
+    def __init__(self, status_code: int, message: str):
+        self.status_code = status_code
+        self.message = message
+        super().__init__(self.message)
+
+    def __str__(self):
+        return f"{self.status_code} {self.message}"
+

beamlit/run.py

@@ -3,7 +3,7 @@ from typing import Any
 
 import requests
 from beamlit.client import AuthenticatedClient
-from beamlit.common.settings import get_settings
+from beamlit.common import HTTPError, get_settings
 
 
 class RunClient:
@@ -45,4 +45,6 @@ class RunClient:
             kwargs["json"] = json
 
         response = client.request(method, url, **kwargs)
+        if response.status_code >= 400:
+            raise HTTPError(response.status_code, response.text)
         return response

beamlit/serve/app.py

@@ -7,13 +7,10 @@ from logging import getLogger
 from uuid import uuid4
 
 from asgi_correlation_id import CorrelationIdMiddleware
-from beamlit.common.settings import get_settings, init
-from beamlit.common.instrumentation import (
-    get_metrics_exporter,
-    get_resource_attributes,
-    get_span_exporter,
-    instrument_app,
-)
+from beamlit.common import HTTPError, get_settings, init
+from beamlit.common.instrumentation import (get_metrics_exporter,
+                                            get_resource_attributes,
+                                            get_span_exporter, instrument_app)
 from fastapi import FastAPI, Request, Response
 from fastapi.responses import JSONResponse
 from traceloop.sdk import Traceloop
@@ -90,10 +87,17 @@ async def root(request: Request):
         content = {"error": str(e)}
         if settings.environment == "development":
             content["traceback"] = str(traceback.format_exc())
-        logger.error(f"{content}")
+        logger.error(str(traceback.format_exc()))
         return JSONResponse(status_code=400, content=content)
+    except HTTPError as e:
+        content = {"error": e.message, "status_code": e.status_code}
+        if settings.environment == "development":
+            content["traceback"] = str(traceback.format_exc())
+        logger.error(f"{e.status_code} {str(traceback.format_exc())}")
+        return JSONResponse(status_code=e.status_code, content=content)
     except Exception as e:
         content = {"error": f"Internal server error, {e}"}
         if settings.environment == "development":
             content["traceback"] = str(traceback.format_exc())
+        logger.error(str(traceback.format_exc()))
         return JSONResponse(status_code=500, content=content)

{beamlit-0.0.29rc32.dist-info → beamlit-0.0.30rc33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.29rc32
+Version: 0.0.30rc33
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.29rc32.dist-info → beamlit-0.0.30rc33.dist-info}/RECORD

@@ -2,7 +2,7 @@ beamlit/__init__.py,sha256=545gFC-wLLwUktWcOAjUWe_Glha40tBetRTOYSfHnbI,164
 beamlit/client.py,sha256=PnR6ybZk5dLIJPnDKAf2epHOeQC_7yL0fG4muvphHjA,12695
 beamlit/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
 beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
-beamlit/run.py,sha256=RLwMv5f_S8lw7Wq7O6DvEcOl0nGYh769qjGl_SmR9Z0,1338
+beamlit/run.py,sha256=BCVQXdtIzJedxqGSWttYlJphO5ru7VNDYrUAQmCNq_c,1445
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=nf1iwQwGtCG6nDqyVhxfWoqR6dv6X3bvSpCeqkTCFaM,101
 beamlit/agents/chain.py,sha256=HzBs3nI4xaH86I_r-M-HGGp6roXsJxMx-qXl_GrJaY0,2831
@@ -128,7 +128,8 @@ beamlit/authentication/authentication.py,sha256=8R-3WdQSykNjCbebAW2p8Glvw5nlAmSE
 beamlit/authentication/clientcredentials.py,sha256=cxZPPu--CgizwqX0pdfFQ91gJt1EFKwyy-aBB_dXX7I,3990
 beamlit/authentication/credentials.py,sha256=p_1xenabCbQuRz7BiFk7oTK4uCxAt_zoyku5o-jcKGE,5343
 beamlit/authentication/device_mode.py,sha256=tmr22gllKOZwBRub_QjF5pYa425x-nE8tQNpZ_EGR6g,3644
-beamlit/common/__init__.py,sha256=yDoMJDKj-xjTGl7U1YI59KpWxiOV65HSiUulgO8xdTA,277
+beamlit/common/__init__.py,sha256=vj4_boIBVitMsaQR8BqBqE2eupOIh6MWBAYlYyCCH98,341
+beamlit/common/error.py,sha256=f9oJDFxhoHK-vpjxBgEp0NwWIk0N_THPemUI7uQxVzU,270
 beamlit/common/generate.py,sha256=LtdCju_QayRS4lZrrb_0VHqWWvTcv4Mbf-iV1TB_Qko,7522
 beamlit/common/instrumentation.py,sha256=MsBDfFcMYqGDiHHj4j5hLHE4EWxZExkhmCeFS3SKzJY,3181
 beamlit/common/logger.py,sha256=VFRbaZh93n8ZGugeeYKe88IP2nI3g2JNa7XN4j8wVJE,1116
@@ -251,10 +252,10 @@ beamlit/models/websocket_channel.py,sha256=jg3vN7yS_oOIwGtndtIUr1LsyEA58RXLXahqS
 beamlit/models/workspace.py,sha256=l__bIpbA4oJvxXo7UbEoCcqkvu9MiNt5aXXpZ3bgwHg,4309
 beamlit/models/workspace_labels.py,sha256=WbnUY6eCTkUNdY7hhhSF-KQCl8fWFfkCf7hzCTiNp4A,1246
 beamlit/models/workspace_user.py,sha256=70CcifQWYbeWG7TDui4pblTzUe5sVK0AS19vNCzKE8g,3423
-beamlit/serve/app.py,sha256=OpwPjRdyHZK6J-ziPwhiRDGGa2mvCrFVcBFE6alJVOM,3071
+beamlit/serve/app.py,sha256=k9THiOqcPs4lcQQhZaucLcboXCR_5AKP9AFCMNRbdPI,3559
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=Mu4T4_9OvHybjA0ApzZFpgi2C8f3X1NbUk-76v634XM,631
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.29rc32.dist-info/METADATA,sha256=SX1BoOg-Mqw3GzEHwX9bdwJPkkQ01jE1AHgRdGKXjvY,2405
-beamlit-0.0.29rc32.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.29rc32.dist-info/RECORD,,
+beamlit-0.0.30rc33.dist-info/METADATA,sha256=NQdt4gFJboZxWip3Z_i19Tkn759fbWvjZgi0ffzBYrQ,2405
+beamlit-0.0.30rc33.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.30rc33.dist-info/RECORD,,