beamlit 0.0.29rc32__py3-none-any.whl → 0.0.30__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/common/settings.py

@@ -42,7 +42,7 @@ class SettingsServer(BaseSettings):
     module: str = Field(default="main.main")
     port: int = Field(default=80)
     host: str = Field(default="0.0.0.0")
-
+    directory: str = Field(default="src")
 
 class Settings(BaseSettings):
     model_config = SettingsConfigDict(

beamlit/deploy/deploy.py

@@ -174,15 +174,15 @@ def dockerfile(
     Returns:
         str: Dockerfile content
     """
+    settings = get_settings()
     if type == "agent":
         module = f"{resource.module.__file__.split('/')[-1].replace('.py', '')}.{resource.module.__name__}"
     else:
-        module = f"functions.{resource.module.__name__}.{resource.func.__name__}"
+        module = f"functions.{resource.module.__file__.split('/')[-1].replace('.py', '')}.{resource.module.__name__}"
     cmd = ["bl", "serve", "--port", "80", "--module", module]
     if type == "agent":
         cmd.append("--remote")
     cmd_str = ",".join([f'"{c}"' for c in cmd])
-
     return f"""
 FROM python:3.12-slim
 
@@ -201,7 +201,7 @@ RUN uv sync --no-cache
 
 COPY README.m[d] /beamlit/README.md
 COPY LICENS[E] /beamlit/LICENSE
-COPY src /beamlit/src
+COPY {settings.server.directory} /beamlit/src
 
 ENV PATH="/beamlit/.venv/bin:$PATH"
 
@@ -226,11 +226,11 @@ def generate_beamlit_deployment(directory: str):
     logger.info(f"Importing server module: {settings.server.module}")
     functions: list[tuple[Resource, Function]] = []
     agents: list[tuple[Resource, Agent]] = []
-    for resource in get_resources("agent"):
+    for resource in get_resources("agent", settings.server.directory):
         agent = get_beamlit_deployment_from_resource(resource)
         if agent:
             agents.append((resource, agent))
-    for resource in get_resources("function"):
+    for resource in get_resources("function", settings.server.directory):
         function = get_beamlit_deployment_from_resource(resource)
         if function:
             functions.append((resource, function))
@@ -251,6 +251,10 @@ def generate_beamlit_deployment(directory: str):
         with open(os.path.join(agent_dir, f"Dockerfile"), "w") as f:
             content = dockerfile("agent", resource, agent)
             f.write(content)
+        # write destination docker
+        with open(os.path.join(agent_dir, f"destination.txt"), "w") as f:
+            content = agent.spec.runtime.image
+            f.write(content)
     for resource, function in functions:
         # write deployment file
         function_dir = os.path.join(functions_dir, function.metadata.name)
@@ -262,3 +266,7 @@ def generate_beamlit_deployment(directory: str):
         with open(os.path.join(function_dir, f"Dockerfile"), "w") as f:
             content = dockerfile("function", resource, function)
             f.write(content)
+        # write destination docker
+        with open(os.path.join(function_dir, f"destination.txt"), "w") as f:
+            content = function.spec.runtime.image
+            f.write(content)

beamlit/deploy/parser.py

@@ -7,6 +7,7 @@ from typing import Callable, Literal
 
 from beamlit.models import StoreFunctionParameter
 
+
 @dataclass
 class Resource:
     type: Literal["agent", "function"]
@@ -16,7 +17,7 @@ class Resource:
     func: Callable
 
 
-def get_resources(from_decorator, dir="src") -> list[Resource]:
+def get_resources(from_decorator, dir) -> list[Resource]:
     """
     Scans through Python files in a directory to find functions decorated with a specific decorator.
 

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.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.29rc32
+Version: 0.0.30
 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.30.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,17 +128,18 @@ 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
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
-beamlit/common/settings.py,sha256=bxgQxMV5ncNqDGcWS_Wj3nzOF8FgAmC6alMP2fOdEDU,5895
+beamlit/common/settings.py,sha256=2V06qodZHwPBqRDdpPyZhdCO7HeMjWr2rkEePOgmHqE,5936
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/deploy/__init__.py,sha256=GS7l7Jtm2yKs7iNLKcfjYO-rAhUzggQ3xiYSf3oxLBY,91
-beamlit/deploy/deploy.py,sha256=on1i93SdECKrLVRMm3V2BRW5JeolPPq1dJHa4Evp0ns,9188
+beamlit/deploy/deploy.py,sha256=AbQxSVGAPbwBn4wKPiMEF5GjtGa8q4u3DBuhnYdz4GI,9706
 beamlit/deploy/format.py,sha256=78tOoeNPJ8969AhQTeFlIwZgQ3un8gmTSMmrYbRQSds,1818
-beamlit/deploy/parser.py,sha256=uT-bezLX6yjyxr1ogm1GXIT_MeREqHDUBlKiyav5qg0,6912
+beamlit/deploy/parser.py,sha256=Ga0poCZkoRnuTw082QnTcNGCBJncoRAnVsn8-1FsaJE,6907
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
 beamlit/functions/decorator.py,sha256=NsdIWs55Yhn-TUO855DIE2AarV-euXtl7Qt5c2W3AUo,3239
 beamlit/functions/github/__init__.py,sha256=gYnUkeegukOfbymdabuuJkScvH-_ZJygX05BoqkPn0o,49
@@ -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.30.dist-info/METADATA,sha256=UNYwevCSKK872KofAvX_-dHOIuFA_9LcrmF4lgT4mFg,2401
+beamlit-0.0.30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.30.dist-info/RECORD,,