beamlit 0.0.29rc31__tar.gz → 0.0.30__tar.gz

{beamlit-0.0.29rc31 → beamlit-0.0.30}/PKG-INFO

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

{beamlit-0.0.29rc31 → beamlit-0.0.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.29.pre-31"
+version = "0.0.30"
 description = "Add your description here"
 readme = "README.md"
 authors = [

beamlit-0.0.30/src/beamlit/api/agents/get_agent_trace_ids.py

@@ -0,0 +1,171 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.trace_ids_response import TraceIdsResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    agent_name: str,
+    *,
+    environment: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/agents/{agent_name}/traces",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[TraceIdsResponse]:
+    if response.status_code == 200:
+        response_200 = TraceIdsResponse.from_dict(response.json())
+
+        return response_200
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[TraceIdsResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    agent_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[TraceIdsResponse]:
+    """Get agent trace IDs
+
+    Args:
+        agent_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[TraceIdsResponse]
+    """
+
+    kwargs = _get_kwargs(
+        agent_name=agent_name,
+        environment=environment,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    agent_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[TraceIdsResponse]:
+    """Get agent trace IDs
+
+    Args:
+        agent_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        TraceIdsResponse
+    """
+
+    return sync_detailed(
+        agent_name=agent_name,
+        client=client,
+        environment=environment,
+    ).parsed
+
+
+async def asyncio_detailed(
+    agent_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[TraceIdsResponse]:
+    """Get agent trace IDs
+
+    Args:
+        agent_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[TraceIdsResponse]
+    """
+
+    kwargs = _get_kwargs(
+        agent_name=agent_name,
+        environment=environment,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    agent_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[TraceIdsResponse]:
+    """Get agent trace IDs
+
+    Args:
+        agent_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        TraceIdsResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            agent_name=agent_name,
+            client=client,
+            environment=environment,
+        )
+    ).parsed

beamlit-0.0.30/src/beamlit/api/functions/get_function_trace_ids.py

@@ -0,0 +1,171 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.trace_ids_response import TraceIdsResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    function_name: str,
+    *,
+    environment: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/functions/{function_name}/traces",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[TraceIdsResponse]:
+    if response.status_code == 200:
+        response_200 = TraceIdsResponse.from_dict(response.json())
+
+        return response_200
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[TraceIdsResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    function_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[TraceIdsResponse]:
+    """Get function trace IDs
+
+    Args:
+        function_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[TraceIdsResponse]
+    """
+
+    kwargs = _get_kwargs(
+        function_name=function_name,
+        environment=environment,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    function_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[TraceIdsResponse]:
+    """Get function trace IDs
+
+    Args:
+        function_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        TraceIdsResponse
+    """
+
+    return sync_detailed(
+        function_name=function_name,
+        client=client,
+        environment=environment,
+    ).parsed
+
+
+async def asyncio_detailed(
+    function_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[TraceIdsResponse]:
+    """Get function trace IDs
+
+    Args:
+        function_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[TraceIdsResponse]
+    """
+
+    kwargs = _get_kwargs(
+        function_name=function_name,
+        environment=environment,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    function_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[TraceIdsResponse]:
+    """Get function trace IDs
+
+    Args:
+        function_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        TraceIdsResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            function_name=function_name,
+            client=client,
+            environment=environment,
+        )
+    ).parsed

beamlit-0.0.30/src/beamlit/api/models/get_model_trace_ids.py

@@ -0,0 +1,171 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.trace_ids_response import TraceIdsResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    model_name: str,
+    *,
+    environment: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["environment"] = environment
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/models/{model_name}/traces",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[TraceIdsResponse]:
+    if response.status_code == 200:
+        response_200 = TraceIdsResponse.from_dict(response.json())
+
+        return response_200
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[TraceIdsResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    model_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[TraceIdsResponse]:
+    """Get model trace IDs
+
+    Args:
+        model_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[TraceIdsResponse]
+    """
+
+    kwargs = _get_kwargs(
+        model_name=model_name,
+        environment=environment,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    model_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[TraceIdsResponse]:
+    """Get model trace IDs
+
+    Args:
+        model_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        TraceIdsResponse
+    """
+
+    return sync_detailed(
+        model_name=model_name,
+        client=client,
+        environment=environment,
+    ).parsed
+
+
+async def asyncio_detailed(
+    model_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[TraceIdsResponse]:
+    """Get model trace IDs
+
+    Args:
+        model_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[TraceIdsResponse]
+    """
+
+    kwargs = _get_kwargs(
+        model_name=model_name,
+        environment=environment,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    model_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Optional[TraceIdsResponse]:
+    """Get model trace IDs
+
+    Args:
+        model_name (str):
+        environment (Union[Unset, str]):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        TraceIdsResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            model_name=model_name,
+            client=client,
+            environment=environment,
+        )
+    ).parsed

{beamlit-0.0.29rc31 → beamlit-0.0.30}/src/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-0.0.30/src/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-0.0.29rc31 → beamlit-0.0.30}/src/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-0.0.29rc31 → beamlit-0.0.30}/src/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-0.0.29rc31 → beamlit-0.0.30}/src/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-0.0.29rc31 → beamlit-0.0.30}/src/beamlit/models/__init__.py

@@ -93,6 +93,7 @@ from .store_function_kit import StoreFunctionKit
 from .store_function_labels import StoreFunctionLabels
 from .store_function_parameter import StoreFunctionParameter
 from .time_fields import TimeFields
+from .trace_ids_response import TraceIdsResponse
 from .update_workspace_service_account_body import UpdateWorkspaceServiceAccountBody
 from .update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
 from .update_workspace_user_role_body import UpdateWorkspaceUserRoleBody
@@ -191,6 +192,7 @@ __all__ = (
     "StoreFunctionLabels",
     "StoreFunctionParameter",
     "TimeFields",
+    "TraceIdsResponse",
     "UpdateWorkspaceServiceAccountBody",
     "UpdateWorkspaceServiceAccountResponse200",
     "UpdateWorkspaceUserRoleBody",