beamlit 0.0.20rc5__py3-none-any.whl → 0.0.20rc7__py3-none-any.whl

beamlit/models/store_configuration.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -17,30 +17,30 @@ class StoreConfiguration:
     """Store configuration for resources (eg: agent, function, etc)
 
     Attributes:
-        available_models (Union[Unset, List[str]]): Available models for the configuration
+        available_models (Union[Unset, list[str]]): Available models for the configuration
         description (Union[Unset, str]): Store configuration description
         display_name (Union[Unset, str]): Store configuration display name
         if_ (Union[Unset, str]): Conditional rendering for the configuration, example: provider === 'openai'
         name (Union[Unset, str]): Store configuration name
-        options (Union[Unset, List['StoreConfigurationOption']]):
+        options (Union[Unset, list['StoreConfigurationOption']]):
         required (Union[Unset, bool]): Store configuration required
         secret (Union[Unset, bool]): Store configuration secret
-        type (Union[Unset, str]): Store configuration type
+        type_ (Union[Unset, str]): Store configuration type
     """
 
-    available_models: Union[Unset, List[str]] = UNSET
+    available_models: Union[Unset, list[str]] = UNSET
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     if_: Union[Unset, str] = UNSET
     name: Union[Unset, str] = UNSET
-    options: Union[Unset, List["StoreConfigurationOption"]] = UNSET
+    options: Union[Unset, list["StoreConfigurationOption"]] = UNSET
     required: Union[Unset, bool] = UNSET
     secret: Union[Unset, bool] = UNSET
-    type: Union[Unset, str] = UNSET
+    type_: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        available_models: Union[Unset, List[str]] = UNSET
+        available_models: Union[Unset, list[str]] = UNSET
         if not isinstance(self.available_models, Unset):
             available_models = self.available_models
 
@@ -52,7 +52,7 @@ class StoreConfiguration:
 
         name = self.name
 
-        options: Union[Unset, List[Dict[str, Any]]] = UNSET
+        options: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.options, Unset):
             options = []
             for options_item_data in self.options:
@@ -63,7 +63,7 @@ class StoreConfiguration:
 
         secret = self.secret
 
-        type = self.type
+        type_ = self.type_
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -84,19 +84,19 @@ class StoreConfiguration:
             field_dict["required"] = required
         if secret is not UNSET:
             field_dict["secret"] = secret
-        if type is not UNSET:
-            field_dict["type"] = type
+        if type_ is not UNSET:
+            field_dict["type"] = type_
 
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.store_configuration_option import StoreConfigurationOption
 
         if not src_dict:
             return None
         d = src_dict.copy()
-        available_models = cast(List[str], d.pop("available_models", UNSET))
+        available_models = cast(list[str], d.pop("available_models", UNSET))
 
         description = d.pop("description", UNSET)
 
@@ -117,7 +117,7 @@ class StoreConfiguration:
 
         secret = d.pop("secret", UNSET)
 
-        type = d.pop("type", UNSET)
+        type_ = d.pop("type", UNSET)
 
         store_configuration = cls(
             available_models=available_models,
@@ -128,7 +128,7 @@ class StoreConfiguration:
             options=options,
             required=required,
             secret=secret,
-            type=type,
+            type_=type_,
         )
 
         store_configuration.additional_properties = d

beamlit/models/store_configuration_option.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -43,7 +43,7 @@ class StoreConfigurationOption:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/store_function.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -24,28 +24,28 @@ class StoreFunction:
         created_by (Union[Unset, str]): The user or service account who created the resource
         updated_at (Union[Unset, str]): The date and time when the resource was updated
         updated_by (Union[Unset, str]): The user or service account who updated the resource
-        configuration (Union[Unset, List['StoreConfiguration']]): Store function configuration
+        configuration (Union[Unset, list['StoreConfiguration']]): Store function configuration
         description (Union[Unset, str]): Store function description
         display_name (Union[Unset, str]): Store function display name
         image (Union[Unset, str]): Store function image
-        kit (Union[Unset, List['StoreFunctionKit']]): Store function kit
+        kit (Union[Unset, list['StoreFunctionKit']]): Store function kit
         labels (Union[Unset, StoreFunctionLabels]): Store function labels
         name (Union[Unset, str]): Store function name
-        parameters (Union[Unset, List['StoreFunctionParameter']]): Store function parameters
+        parameters (Union[Unset, list['StoreFunctionParameter']]): Store function parameters
     """
 
     created_at: Union[Unset, str] = UNSET
     created_by: Union[Unset, str] = UNSET
     updated_at: Union[Unset, str] = UNSET
     updated_by: Union[Unset, str] = UNSET
-    configuration: Union[Unset, List["StoreConfiguration"]] = UNSET
+    configuration: Union[Unset, list["StoreConfiguration"]] = UNSET
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
     image: Union[Unset, str] = UNSET
-    kit: Union[Unset, List["StoreFunctionKit"]] = UNSET
+    kit: Union[Unset, list["StoreFunctionKit"]] = UNSET
     labels: Union[Unset, "StoreFunctionLabels"] = UNSET
     name: Union[Unset, str] = UNSET
-    parameters: Union[Unset, List["StoreFunctionParameter"]] = UNSET
+    parameters: Union[Unset, list["StoreFunctionParameter"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -57,7 +57,7 @@ class StoreFunction:
 
         updated_by = self.updated_by
 
-        configuration: Union[Unset, List[Dict[str, Any]]] = UNSET
+        configuration: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.configuration, Unset):
             configuration = []
             for configuration_item_data in self.configuration:
@@ -70,20 +70,20 @@ class StoreFunction:
 
         image = self.image
 
-        kit: Union[Unset, List[Dict[str, Any]]] = UNSET
+        kit: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.kit, Unset):
             kit = []
             for kit_item_data in self.kit:
                 kit_item = kit_item_data.to_dict()
                 kit.append(kit_item)
 
-        labels: Union[Unset, Dict[str, Any]] = UNSET
+        labels: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
 
         name = self.name
 
-        parameters: Union[Unset, List[Dict[str, Any]]] = UNSET
+        parameters: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.parameters, Unset):
             parameters = []
             for parameters_item_data in self.parameters:
@@ -121,7 +121,7 @@ class StoreFunction:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.store_configuration import StoreConfiguration
         from ..models.store_function_kit import StoreFunctionKit
         from ..models.store_function_labels import StoreFunctionLabels

beamlit/models/store_function_configuration.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -55,7 +55,7 @@ class StoreFunctionConfiguration:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/store_function_kit.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -20,13 +20,13 @@ class StoreFunctionKit:
         description (Union[Unset, str]): Description of the function kit, very important for the agent to work with your
             kit
         name (Union[Unset, str]): The kit name, very important for the agent to work with your kit
-        parameters (Union[Unset, List['StoreFunctionParameter']]): Kit parameters, for your kit to be callable with an
+        parameters (Union[Unset, list['StoreFunctionParameter']]): Kit parameters, for your kit to be callable with an
             Agent
     """
 
     description: Union[Unset, str] = UNSET
     name: Union[Unset, str] = UNSET
-    parameters: Union[Unset, List["StoreFunctionParameter"]] = UNSET
+    parameters: Union[Unset, list["StoreFunctionParameter"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -34,7 +34,7 @@ class StoreFunctionKit:
 
         name = self.name
 
-        parameters: Union[Unset, List[Dict[str, Any]]] = UNSET
+        parameters: Union[Unset, list[dict[str, Any]]] = UNSET
         if not isinstance(self.parameters, Unset):
             parameters = []
             for parameters_item_data in self.parameters:
@@ -54,7 +54,7 @@ class StoreFunctionKit:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.store_function_parameter import StoreFunctionParameter
 
         if not src_dict:

beamlit/models/store_function_labels.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,7 +19,7 @@ class StoreFunctionLabels:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/store_function_parameter.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -16,13 +16,13 @@ class StoreFunctionParameter:
         description (Union[Unset, str]): Store function parameter description
         name (Union[Unset, str]): Store function parameter name
         required (Union[Unset, bool]): Store function parameter required
-        type (Union[Unset, str]): Store function parameter type
+        type_ (Union[Unset, str]): Store function parameter type
     """
 
     description: Union[Unset, str] = UNSET
     name: Union[Unset, str] = UNSET
     required: Union[Unset, bool] = UNSET
-    type: Union[Unset, str] = UNSET
+    type_: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -32,7 +32,7 @@ class StoreFunctionParameter:
 
         required = self.required
 
-        type = self.type
+        type_ = self.type_
 
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -43,13 +43,13 @@ class StoreFunctionParameter:
             field_dict["name"] = name
         if required is not UNSET:
             field_dict["required"] = required
-        if type is not UNSET:
-            field_dict["type"] = type
+        if type_ is not UNSET:
+            field_dict["type"] = type_
 
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()
@@ -59,13 +59,13 @@ class StoreFunctionParameter:
 
         required = d.pop("required", UNSET)
 
-        type = d.pop("type", UNSET)
+        type_ = d.pop("type", UNSET)
 
         store_function_parameter = cls(
             description=description,
             name=name,
             required=required,
-            type=type,
+            type_=type_,
         )
 
         store_function_parameter.additional_properties = d

beamlit/models/update_workspace_service_account_body.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -36,7 +36,7 @@ class UpdateWorkspaceServiceAccountBody:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/update_workspace_service_account_response_200.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -54,7 +54,7 @@ class UpdateWorkspaceServiceAccountResponse200:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/update_workspace_user_role_body.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -30,7 +30,7 @@ class UpdateWorkspaceUserRoleBody:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/workspace.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -46,7 +46,7 @@ class Workspace:
 
         display_name = self.display_name
 
-        labels: Union[Unset, Dict[str, Any]] = UNSET
+        labels: Union[Unset, dict[str, Any]] = UNSET
         if not isinstance(self.labels, Unset):
             labels = self.labels.to_dict()
 
@@ -73,7 +73,7 @@ class Workspace:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         from ..models.workspace_labels import WorkspaceLabels
 
         if not src_dict:

beamlit/models/workspace_labels.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar
+from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,7 +19,7 @@ class WorkspaceLabels:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/models/workspace_user.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, Union
+from typing import Any, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -67,7 +67,7 @@ class WorkspaceUser:
         return field_dict
 
     @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
         if not src_dict:
             return None
         d = src_dict.copy()

beamlit/run.py

@@ -1,8 +1,8 @@
 import urllib.parse
-from contextlib import contextmanager
-from typing import Any, Callable, Dict, List, Optional
+from typing import Any
 
 import requests
+
 from beamlit.client import AuthenticatedClient
 from beamlit.common.settings import get_settings
 
@@ -18,10 +18,10 @@ class RunClient:
         environment: str,
         method: str,
         path: str = "",
-        headers: Optional[Dict[str, str]] = None,
-        json: Optional[Dict[str, Any]] = None,
-        data: Optional[str] = None,
-        params: Optional[Dict[str, str]] = None,
+        headers: dict[str, str] | None = None,
+        json: dict[str, Any] | None = None,
+        data: str | None = None,
+        params: dict[str, str] | None = None,
     ) -> requests.Response:
         settings = get_settings()
         headers = headers or {}

beamlit/serve/app.py

@@ -0,0 +1,73 @@
+import importlib
+import traceback
+from logging import getLogger
+from uuid import uuid4
+
+from asgi_correlation_id import CorrelationIdMiddleware
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import JSONResponse
+
+from beamlit.common.settings import get_settings, init
+
+from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
+
+
+def import_module():
+    settings = get_settings()
+    main_module = importlib.import_module(".".join(settings.server.module.split(".")[0:-1]))
+    func = getattr(main_module, settings.server.module.split(".")[-1])
+    return func
+
+
+settings = init()
+logger = getLogger(__name__)
+logger.info(f"Importing server module: {settings.server.module}")
+func = import_module()
+logger.info(
+    f"Running server with environment {settings.environment}"
+    f" on {settings.server.host}:{settings.server.port}"
+)
+
+app = FastAPI(docs_url=None, redoc_url=None)
+app.add_middleware(
+    CorrelationIdMiddleware,
+    header_name="x-beamlit-request-id",
+    generator=lambda: str(uuid4()),
+)
+app.add_middleware(AddProcessTimeHeader)
+app.add_middleware(AccessLogMiddleware)
+
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}
+
+
+@app.post("/")
+async def root(request: Request):
+    settings = get_settings()
+    logger = getLogger(__name__)
+    try:
+        body = await request.json()
+        response = await func(body)
+        if isinstance(response, Response):
+            return response
+        if type(response) is str:
+            return Response(
+                content=response,
+                headers={"Content-Type": "text/plain"},
+                media_type="text/plain",
+                status_code=200,
+            )
+        return JSONResponse(status_code=200, content=response)
+    except ValueError as e:
+        content = {"error": str(e)}
+        if settings.environment == "development":
+            content["traceback"] = str(traceback.format_exc())
+        logger.error(f"{content}")
+        return JSONResponse(status_code=400, content=content)
+    except Exception as e:
+        content = {"error": f"Internal server error, {e}"}
+        if settings.environment == "development":
+            content["traceback"] = str(traceback.format_exc())
+        return JSONResponse(status_code=500, content=content)

beamlit/serve/middlewares/__init__.py

@@ -0,0 +1,4 @@
+from .accesslog import AccessLogMiddleware
+from .processtime import AddProcessTimeHeader
+
+__all__ = ["AccessLogMiddleware", "AddProcessTimeHeader"]

beamlit/serve/middlewares/accesslog.py

@@ -0,0 +1,14 @@
+from logging import getLogger
+
+from starlette.middleware.base import BaseHTTPMiddleware
+
+
+class AccessLogMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request, call_next):
+        logger = getLogger(__name__)
+        response = await call_next(request)
+        process_time = response.headers.get("X-Process-Time")
+        rid_header = response.headers.get("X-Request-Id")
+        request_id = rid_header or response.headers.get("X-Beamlit-Request-Id")
+        logger.info(f"{request.method} {request.url.path} {response.status_code} {process_time}ms rid={request_id}")
+        return response

beamlit/serve/middlewares/processtime.py

@@ -0,0 +1,12 @@
+import time
+
+from starlette.middleware.base import BaseHTTPMiddleware
+
+
+class AddProcessTimeHeader(BaseHTTPMiddleware):
+    async def dispatch(self, request, call_next):
+        start_time = time.perf_counter()
+        response = await call_next(request)
+        process_time = (time.perf_counter() - start_time) * 1000
+        response.headers["X-Process-Time"] = f"{process_time:.2f}"
+        return response

{beamlit-0.0.20rc5.dist-info → beamlit-0.0.20rc7.dist-info}/METADATA

@@ -1,11 +1,17 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.20rc5
+Version: 0.0.20rc7
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
+Requires-Dist: asgi-correlation-id<5.0.0,>=4.3.4
 Requires-Dist: attrs>=21.3.0
+Requires-Dist: fastapi[standard]<0.116.0,>=0.115.4
 Requires-Dist: httpx<0.28.0,>=0.20.0
+Requires-Dist: langchain-community<0.4.0,>=0.3.3
+Requires-Dist: langchain-core<0.4.0,>=0.3.13
+Requires-Dist: langchain-openai<0.3.0,>=0.2.4
+Requires-Dist: langgraph<0.3.0,>=0.2.40
 Requires-Dist: pydantic-settings<2.7.0,>=2.6.1
 Requires-Dist: pydantic<2.11.0,>=2.10.3
 Requires-Dist: python-dateutil>=2.8.0