beamlit 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

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,15 @@ 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()
         description = d.pop("description", UNSET)
 
@@ -57,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/time_fields.py

@@ -0,0 +1,68 @@
+from typing import Any, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="TimeFields")
+
+
+@_attrs_define
+class TimeFields:
+    """Time fields for Persistance
+
+    Attributes:
+        created_at (Union[Unset, str]): The date and time when the resource was created
+        updated_at (Union[Unset, str]): The date and time when the resource was updated
+    """
+
+    created_at: Union[Unset, str] = UNSET
+    updated_at: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        created_at = self.created_at
+
+        updated_at = self.updated_at
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if created_at is not UNSET:
+            field_dict["createdAt"] = created_at
+        if updated_at is not UNSET:
+            field_dict["updatedAt"] = updated_at
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        created_at = d.pop("createdAt", UNSET)
+
+        updated_at = d.pop("updatedAt", UNSET)
+
+        time_fields = cls(
+            created_at=created_at,
+            updated_at=updated_at,
+        )
+
+        time_fields.additional_properties = d
+        return time_fields
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

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,9 @@ 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()
         description = d.pop("description", UNSET)
 

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,9 @@ 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()
         client_id = d.pop("client_id", UNSET)
 

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,9 @@ 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()
         role = d.pop("role")
 

beamlit/models/websocket_channel.py

@@ -0,0 +1,86 @@
+from typing import Any, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="WebsocketChannel")
+
+
+@_attrs_define
+class WebsocketChannel:
+    """WebSocket connection details
+
+    Attributes:
+        created_at (Union[Unset, str]): The date and time when the resource was created
+        updated_at (Union[Unset, str]): The date and time when the resource was updated
+        connection_id (Union[Unset, str]): Unique connection ID
+        workspace (Union[Unset, str]): Workspace the connection belongs to
+    """
+
+    created_at: Union[Unset, str] = UNSET
+    updated_at: Union[Unset, str] = UNSET
+    connection_id: Union[Unset, str] = UNSET
+    workspace: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        created_at = self.created_at
+
+        updated_at = self.updated_at
+
+        connection_id = self.connection_id
+
+        workspace = self.workspace
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if created_at is not UNSET:
+            field_dict["createdAt"] = created_at
+        if updated_at is not UNSET:
+            field_dict["updatedAt"] = updated_at
+        if connection_id is not UNSET:
+            field_dict["connection_id"] = connection_id
+        if workspace is not UNSET:
+            field_dict["workspace"] = workspace
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        created_at = d.pop("createdAt", UNSET)
+
+        updated_at = d.pop("updatedAt", UNSET)
+
+        connection_id = d.pop("connection_id", UNSET)
+
+        workspace = d.pop("workspace", UNSET)
+
+        websocket_channel = cls(
+            created_at=created_at,
+            updated_at=updated_at,
+            connection_id=connection_id,
+            workspace=workspace,
+        )
+
+        websocket_channel.additional_properties = d
+        return websocket_channel
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

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,9 +73,11 @@ 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:
+            return None
         d = src_dict.copy()
         created_at = d.pop("created_at", UNSET)
 

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,9 @@ 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()
         workspace_labels = cls()
 

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,9 @@ 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()
         accepted = d.pop("accepted", UNSET)
 

beamlit/run.py

@@ -0,0 +1,49 @@
+import urllib.parse
+from typing import Any
+
+import requests
+
+from beamlit.client import AuthenticatedClient
+from beamlit.common.settings import get_settings
+
+
+class RunClient:
+    def __init__(self, client: AuthenticatedClient):
+        self.client = client
+
+    def run(
+        self,
+        resource_type: str,
+        resource_name: str,
+        environment: str,
+        method: str,
+        path: str = "",
+        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 {}
+        params = params or {}
+
+        # Build the path
+        if path:
+            path = f"{settings.workspace}/{resource_type}s/{resource_name}/{path}"
+        else:
+            path = f"{settings.workspace}/{resource_type}s/{resource_name}"
+
+        client = self.client.get_httpx_client()
+        url = urllib.parse.urljoin(settings.run_url, path)
+
+        kwargs = {
+            "headers": headers,
+            "params": {"environment": environment, **params},
+        }
+        if data:
+            kwargs["data"] = data
+        if json:
+            kwargs["json"] = json
+
+        response = client.request(method, url, **kwargs)
+        return response

beamlit/serve/app.py

@@ -0,0 +1,78 @@
+import importlib
+import os
+import sys
+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
+
+sys.path.insert(0, os.getcwd())
+sys.path.insert(0, os.path.join(os.getcwd(), "src"))
+
+
+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.19.dist-info → beamlit-0.0.20.dist-info}/METADATA

@@ -1,13 +1,22 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.19
+Version: 0.0.20
 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
-Requires-Dist: pyyaml==6.0.2
+Requires-Dist: pyyaml<6.1.0,>=6.0.2
+Requires-Dist: requests<2.33.0,>=2.32.3
 Description-Content-Type: text/markdown
 
 # beamlit