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

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.py

@@ -0,0 +1,120 @@
+import importlib
+import time
+import traceback
+from contextlib import asynccontextmanager
+from logging import getLogger
+from uuid import uuid4
+
+import uvicorn
+from asgi_correlation_id import CorrelationIdMiddleware
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import JSONResponse
+from starlette.middleware.base import BaseHTTPMiddleware
+
+from beamlit.common.settings import get_settings, init
+
+main_agent = None
+
+
+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
+
+
+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
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    try:
+        is_main = __name__ == "main"
+        if not is_main:
+            init()
+
+        logger = getLogger(__name__)
+        settings = get_settings()
+
+        # Import the agent
+        global main_agent
+        main_agent = importlib.import_module(".".join(settings.agent_module.split(".")[0:-1]))
+        # Log the server is running
+        if is_main:
+            logger.info(f"Server running on http://{settings.host}:{settings.port}")
+        yield
+    except Exception as e:
+        logger = getLogger(__name__)
+        logger.error(f"Error initializing agent: {e}", exc_info=True)
+        raise e
+
+
+app = FastAPI(lifespan=lifespan, 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:
+        func = getattr(main_agent, settings.agent_module.split(".")[-1])
+        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)
+
+
+def main():
+    settings = init()
+    uvicorn.run(
+        f"{__name__}:app",
+        host=settings.host,
+        port=settings.port,
+        log_level="critical",
+        reload=settings.environment != "production",
+    )
+
+
+if __name__ == "__main__":
+    main()

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

@@ -1,11 +1,15 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.20rc5
+Version: 0.0.20rc6
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
+Requires-Dist: asgi-correlation-id==4.3.4
 Requires-Dist: attrs>=21.3.0
+Requires-Dist: fastapi[standard]==0.115.4
 Requires-Dist: httpx<0.28.0,>=0.20.0
+Requires-Dist: langchain-core==0.3.13
+Requires-Dist: langgraph==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