beamlit 0.0.27__py3-none-any.whl → 0.0.28__py3-none-any.whl

beamlit/api/agents/get_agent_trace_ids.py

@@ -0,0 +1,106 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+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[Any]:
+    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[Any]:
+    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[Any]:
+    """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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        agent_name=agent_name,
+        environment=environment,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    agent_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[Any]:
+    """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[Any]
+    """
+
+    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)

beamlit/api/default/get_trace.py

@@ -0,0 +1,150 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.get_trace_response_200 import GetTraceResponse200
+from ...types import Response
+
+
+def _get_kwargs(
+    trace_id: str,
+) -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": f"/traces/{trace_id}",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[GetTraceResponse200]:
+    if response.status_code == 200:
+        response_200 = GetTraceResponse200.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[GetTraceResponse200]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+) -> Response[GetTraceResponse200]:
+    """Get trace by ID
+
+    Args:
+        trace_id (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[GetTraceResponse200]
+    """
+
+    kwargs = _get_kwargs(
+        trace_id=trace_id,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+) -> Optional[GetTraceResponse200]:
+    """Get trace by ID
+
+    Args:
+        trace_id (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:
+        GetTraceResponse200
+    """
+
+    return sync_detailed(
+        trace_id=trace_id,
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+) -> Response[GetTraceResponse200]:
+    """Get trace by ID
+
+    Args:
+        trace_id (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[GetTraceResponse200]
+    """
+
+    kwargs = _get_kwargs(
+        trace_id=trace_id,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+) -> Optional[GetTraceResponse200]:
+    """Get trace by ID
+
+    Args:
+        trace_id (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:
+        GetTraceResponse200
+    """
+
+    return (
+        await asyncio_detailed(
+            trace_id=trace_id,
+            client=client,
+        )
+    ).parsed

beamlit/api/default/get_trace_ids.py

@@ -0,0 +1,233 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.get_trace_ids_response_200 import GetTraceIdsResponse200
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    workload_id: Union[Unset, str] = UNSET,
+    workload_type: Union[Unset, str] = UNSET,
+    environment: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+    start_time: Union[Unset, str] = UNSET,
+    end_time: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["workloadId"] = workload_id
+
+    params["workloadType"] = workload_type
+
+    params["environment"] = environment
+
+    params["limit"] = limit
+
+    params["startTime"] = start_time
+
+    params["endTime"] = end_time
+
+    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": "/traces",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[GetTraceIdsResponse200]:
+    if response.status_code == 200:
+        response_200 = GetTraceIdsResponse200.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[GetTraceIdsResponse200]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+    workload_id: Union[Unset, str] = UNSET,
+    workload_type: Union[Unset, str] = UNSET,
+    environment: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+    start_time: Union[Unset, str] = UNSET,
+    end_time: Union[Unset, str] = UNSET,
+) -> Response[GetTraceIdsResponse200]:
+    """Get trace IDs
+
+    Args:
+        workload_id (Union[Unset, str]):
+        workload_type (Union[Unset, str]):
+        environment (Union[Unset, str]):
+        limit (Union[Unset, str]):
+        start_time (Union[Unset, str]):
+        end_time (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[GetTraceIdsResponse200]
+    """
+
+    kwargs = _get_kwargs(
+        workload_id=workload_id,
+        workload_type=workload_type,
+        environment=environment,
+        limit=limit,
+        start_time=start_time,
+        end_time=end_time,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+    workload_id: Union[Unset, str] = UNSET,
+    workload_type: Union[Unset, str] = UNSET,
+    environment: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+    start_time: Union[Unset, str] = UNSET,
+    end_time: Union[Unset, str] = UNSET,
+) -> Optional[GetTraceIdsResponse200]:
+    """Get trace IDs
+
+    Args:
+        workload_id (Union[Unset, str]):
+        workload_type (Union[Unset, str]):
+        environment (Union[Unset, str]):
+        limit (Union[Unset, str]):
+        start_time (Union[Unset, str]):
+        end_time (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:
+        GetTraceIdsResponse200
+    """
+
+    return sync_detailed(
+        client=client,
+        workload_id=workload_id,
+        workload_type=workload_type,
+        environment=environment,
+        limit=limit,
+        start_time=start_time,
+        end_time=end_time,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+    workload_id: Union[Unset, str] = UNSET,
+    workload_type: Union[Unset, str] = UNSET,
+    environment: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+    start_time: Union[Unset, str] = UNSET,
+    end_time: Union[Unset, str] = UNSET,
+) -> Response[GetTraceIdsResponse200]:
+    """Get trace IDs
+
+    Args:
+        workload_id (Union[Unset, str]):
+        workload_type (Union[Unset, str]):
+        environment (Union[Unset, str]):
+        limit (Union[Unset, str]):
+        start_time (Union[Unset, str]):
+        end_time (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[GetTraceIdsResponse200]
+    """
+
+    kwargs = _get_kwargs(
+        workload_id=workload_id,
+        workload_type=workload_type,
+        environment=environment,
+        limit=limit,
+        start_time=start_time,
+        end_time=end_time,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+    workload_id: Union[Unset, str] = UNSET,
+    workload_type: Union[Unset, str] = UNSET,
+    environment: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+    start_time: Union[Unset, str] = UNSET,
+    end_time: Union[Unset, str] = UNSET,
+) -> Optional[GetTraceIdsResponse200]:
+    """Get trace IDs
+
+    Args:
+        workload_id (Union[Unset, str]):
+        workload_type (Union[Unset, str]):
+        environment (Union[Unset, str]):
+        limit (Union[Unset, str]):
+        start_time (Union[Unset, str]):
+        end_time (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:
+        GetTraceIdsResponse200
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            workload_id=workload_id,
+            workload_type=workload_type,
+            environment=environment,
+            limit=limit,
+            start_time=start_time,
+            end_time=end_time,
+        )
+    ).parsed

beamlit/api/default/get_trace_logs.py

@@ -0,0 +1,186 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.get_trace_logs_response_200 import GetTraceLogsResponse200
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    trace_id: str,
+    *,
+    span_id: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+) -> dict[str, Any]:
+    params: dict[str, Any] = {}
+
+    params["spanId"] = span_id
+
+    params["limit"] = limit
+
+    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"/traces/{trace_id}/logs",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[GetTraceLogsResponse200]:
+    if response.status_code == 200:
+        response_200 = GetTraceLogsResponse200.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[GetTraceLogsResponse200]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+    span_id: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+) -> Response[GetTraceLogsResponse200]:
+    """Get trace logs
+
+    Args:
+        trace_id (str):
+        span_id (Union[Unset, str]):
+        limit (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[GetTraceLogsResponse200]
+    """
+
+    kwargs = _get_kwargs(
+        trace_id=trace_id,
+        span_id=span_id,
+        limit=limit,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+    span_id: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+) -> Optional[GetTraceLogsResponse200]:
+    """Get trace logs
+
+    Args:
+        trace_id (str):
+        span_id (Union[Unset, str]):
+        limit (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:
+        GetTraceLogsResponse200
+    """
+
+    return sync_detailed(
+        trace_id=trace_id,
+        client=client,
+        span_id=span_id,
+        limit=limit,
+    ).parsed
+
+
+async def asyncio_detailed(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+    span_id: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+) -> Response[GetTraceLogsResponse200]:
+    """Get trace logs
+
+    Args:
+        trace_id (str):
+        span_id (Union[Unset, str]):
+        limit (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[GetTraceLogsResponse200]
+    """
+
+    kwargs = _get_kwargs(
+        trace_id=trace_id,
+        span_id=span_id,
+        limit=limit,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    trace_id: str,
+    *,
+    client: AuthenticatedClient,
+    span_id: Union[Unset, str] = UNSET,
+    limit: Union[Unset, str] = UNSET,
+) -> Optional[GetTraceLogsResponse200]:
+    """Get trace logs
+
+    Args:
+        trace_id (str):
+        span_id (Union[Unset, str]):
+        limit (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:
+        GetTraceLogsResponse200
+    """
+
+    return (
+        await asyncio_detailed(
+            trace_id=trace_id,
+            client=client,
+            span_id=span_id,
+            limit=limit,
+        )
+    ).parsed

beamlit/api/functions/get_function_trace_ids.py

@@ -0,0 +1,106 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+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[Any]:
+    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[Any]:
+    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[Any]:
+    """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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        function_name=function_name,
+        environment=environment,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    function_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[Any]:
+    """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[Any]
+    """
+
+    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)

beamlit/api/models/get_model_trace_ids.py

@@ -0,0 +1,106 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+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[Any]:
+    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[Any]:
+    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[Any]:
+    """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[Any]
+    """
+
+    kwargs = _get_kwargs(
+        model_name=model_name,
+        environment=environment,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    model_name: str,
+    *,
+    client: AuthenticatedClient,
+    environment: Union[Unset, str] = UNSET,
+) -> Response[Any]:
+    """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[Any]
+    """
+
+    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)

beamlit/authentication/authentication.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass
 from typing import Dict, Generator
+import os
 
 from httpx import Auth, Request, Response
 
@@ -26,8 +27,13 @@ class PublicProvider(Auth):
 class RunClientWithCredentials:
     credentials: Credentials
     workspace: str
-    api_url: str = "https://api.beamlit.dev/v0"
-    run_url: str = "https://run.beamlit.dev/v0"
+    api_url: str = "https://api.beamlit.com/v0"
+    run_url: str = "https://run.beamlit.com/v0"
+
+    def __post_init__(self):
+        if os.getenv('BL_ENV') == 'dev':
+            self.api_url = "https://api.beamlit.dev/v0"
+            self.run_url = "https://run.beamlit.dev/v0"
 
 
 def new_client_from_settings(settings: Settings):

beamlit/client.py

@@ -1,3 +1,4 @@
+import os
 import ssl
 from typing import Any, Optional, Union
 
@@ -37,8 +38,13 @@ class Client:
 
     """
 
+    # Determine the base URL based on the environment
+    default_base_url = "https://api.beamlit.com/v0"
+    if os.getenv("BL_ENV") == "dev":
+        default_base_url = "https://api.beamlit.dev/v0"
+
     raise_on_unexpected_status: bool = field(default=True, kw_only=True)
-    _base_url: str = field(alias="base_url", default="https://api.beamlit.dev/v0")
+    _base_url: str = field(alias="base_url", default=default_base_url)
     _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
     _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
     _provider: httpx.Auth = field(default=None, alias="provider")
@@ -169,8 +175,13 @@ class AuthenticatedClient:
         provider: AuthProvider to use for authentication
     """
 
+    # Determine the base URL based on the environment
+    default_base_url = "https://api.beamlit.com/v0"
+    if os.getenv("BL_ENV") == "dev":
+        default_base_url = "https://api.beamlit.dev/v0"
+
     raise_on_unexpected_status: bool = field(default=True, kw_only=True)
-    _base_url: str = field(alias="base_url", default="https://api.beamlit.dev/v0")
+    _base_url: str = field(alias="base_url", default=default_base_url)
     _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
     _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
     _provider: httpx.Auth = field(default=None, alias="provider")

beamlit/common/settings.py

@@ -62,15 +62,23 @@ class Settings(BaseSettings):
     remote: bool = Field(default=False)
     type: str = Field(default="agent")
     name: str = Field(default="beamlit-agent")
-    base_url: str = Field(default="https://api.beamlit.dev/v0")
-    run_url: str = Field(default="https://run.beamlit.dev")
-    mcp_hub_url: str = Field(default="https://mcp-hub-server.beamlit.workers.dev")
-    registry_url: str = Field(default="https://serverless-registry-production.beamlit.workers.dev")
+    base_url: str = Field(default="https://api.beamlit.com/v0")
+    run_url: str = Field(default="https://run.beamlit.com")
+    mcp_hub_url: str = Field(default="https://mcp-hub-server.beamlit.workers.com")
+    registry_url: str = Field(default="https://serverless-registry-production.beamlit.workers.com")
     log_level: str = Field(default="INFO")
     agent: SettingsAgent = SettingsAgent()
     server: SettingsServer = SettingsServer()
     authentication: SettingsAuthentication = SettingsAuthentication()
 
+    def __init__(self, **data):
+        super().__init__(**data)
+        if os.getenv('BL_ENV') == 'dev':
+            self.base_url = "https://api.beamlit.dev/v0"
+            self.run_url = "https://run.beamlit.dev"
+            self.mcp_hub_url = "https://mcp-hub-server.beamlit.workers.dev"
+            self.registry_url = "https://serverless-registry-production.beamlit.workers.dev"
+
     @classmethod
     def settings_customise_sources(
         cls,

beamlit/models/__init__.py

@@ -29,6 +29,9 @@ from .function_kit import FunctionKit
 from .function_metadata import FunctionMetadata
 from .function_release import FunctionRelease
 from .function_spec import FunctionSpec
+from .get_trace_ids_response_200 import GetTraceIdsResponse200
+from .get_trace_logs_response_200 import GetTraceLogsResponse200
+from .get_trace_response_200 import GetTraceResponse200
 from .get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
 from .increase_and_rate_metric import IncreaseAndRateMetric
 from .integration_config import IntegrationConfig
@@ -128,6 +131,9 @@ __all__ = (
     "FunctionMetadata",
     "FunctionRelease",
     "FunctionSpec",
+    "GetTraceIdsResponse200",
+    "GetTraceLogsResponse200",
+    "GetTraceResponse200",
     "GetWorkspaceServiceAccountsResponse200Item",
     "IncreaseAndRateMetric",
     "IntegrationConfig",

beamlit/models/get_trace_ids_response_200.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="GetTraceIdsResponse200")
+
+
+@_attrs_define
+class GetTraceIdsResponse200:
+    """ """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        get_trace_ids_response_200 = cls()
+
+        get_trace_ids_response_200.additional_properties = d
+        return get_trace_ids_response_200
+
+    @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/get_trace_logs_response_200.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="GetTraceLogsResponse200")
+
+
+@_attrs_define
+class GetTraceLogsResponse200:
+    """ """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        get_trace_logs_response_200 = cls()
+
+        get_trace_logs_response_200.additional_properties = d
+        return get_trace_logs_response_200
+
+    @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/get_trace_response_200.py

@@ -0,0 +1,45 @@
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="GetTraceResponse200")
+
+
+@_attrs_define
+class GetTraceResponse200:
+    """ """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: dict[str, Any]) -> T:
+        if not src_dict:
+            return None
+        d = src_dict.copy()
+        get_trace_response_200 = cls()
+
+        get_trace_response_200.additional_properties = d
+        return get_trace_response_200
+
+    @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-0.0.27.dist-info → beamlit-0.0.28.dist-info}/METADATA

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

{beamlit-0.0.27.dist-info → beamlit-0.0.28.dist-info}/RECORD

@@ -1,5 +1,5 @@
 beamlit/__init__.py,sha256=545gFC-wLLwUktWcOAjUWe_Glha40tBetRTOYSfHnbI,164
-beamlit/client.py,sha256=OdRbs5VVHF32HUd5RMcnkhe8YxamnAmgGlxO6pm1Xac,12431
+beamlit/client.py,sha256=SmJ46egRqKwRrLXCHmwrik0D_dD6s46-87YeVhBgxSU,12817
 beamlit/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
 beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
 beamlit/run.py,sha256=RLwMv5f_S8lw7Wq7O6DvEcOl0nGYh769qjGl_SmR9Z0,1338
@@ -17,12 +17,17 @@ beamlit/api/agents/get_agent.py,sha256=IBMiNb36CyNKKyW-RvMSakmOaGrP2hSm3HRa_Gm_c
 beamlit/api/agents/get_agent_environment_logs.py,sha256=Fdd_mvlJXO17BQHbnl0YpUbXcX-1BsuZI2WKz6cgacA,3759
 beamlit/api/agents/get_agent_history.py,sha256=sDKZQhul8wrSbuRY8WNI6jRNYgFcYtCnaU2fgR1owM8,3846
 beamlit/api/agents/get_agent_metrics.py,sha256=IRdex5XAekCHSep6T7KQHB9T1J1f9egDx-MaiNynRVU,4344
+beamlit/api/agents/get_agent_trace_ids.py,sha256=5VXdgE7wveS-K-kcJpk2_SzKPh5vcs9kOkYLO74Nx6M,2729
 beamlit/api/agents/list_agent_history.py,sha256=ZMTG5PSSkfd4OLmVHDIvDZy13bElrhQivF7QtBDLK9w,3775
 beamlit/api/agents/list_agents.py,sha256=d6j_LM-8--2nfTHFjueRkoimHf02RRMAOWTpt8anJGg,4101
 beamlit/api/agents/put_agent_history.py,sha256=lt1_9yFW_vEgeS_jlh-4EumgbTZCdcZYy9GbA91gewU,4590
 beamlit/api/agents/update_agent.py,sha256=No9iJkjUFJBCHDth1_vpuDfl2Ev3us1OWHXrrRzzE4E,3937
 beamlit/api/configurations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/configurations/get_configuration.py,sha256=BtazLwvVe1OGjt0dJ-NVVipqAqZt1HwemWTGK2iuN9w,3197
+beamlit/api/default/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+beamlit/api/default/get_trace.py,sha256=2x-qpeu8jpfbq_c4-aAHgliwBQfagLEWEgM77eB_ZiY,3621
+beamlit/api/default/get_trace_ids.py,sha256=qxT085F7pt4GI0Ir7fpFA19IIR9_NFy47fYL9bPAtJo,6712
+beamlit/api/default/get_trace_logs.py,sha256=R69_j9GFAwkujsNMp4VTxoZuzw8jFni56l0bRxACT4w,4770
 beamlit/api/environments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/environments/create_environment.py,sha256=ocD_8MlOivWlo9ICeSRkHuZfndLHCrROfiJvCynsADU,4321
 beamlit/api/environments/delete_environment.py,sha256=bkDsO1akkb1To4TDz1XvKiB4XI34phBOJLj72KXMW6M,3819
@@ -37,6 +42,7 @@ beamlit/api/functions/delete_function.py,sha256=dzCBAL50Yg18bDUpcC1COjwFstnfBpqt
 beamlit/api/functions/get_function.py,sha256=U4dXy47eKQh7spED7hyyHOepj6bU2U6QFJ0a2RS2y_8,4301
 beamlit/api/functions/get_function_environment_logs.py,sha256=Ia7bDcx8k7qCBhxsm8jdDSbYGKwdTDYhkAcgv25Zz-o,3816
 beamlit/api/functions/get_function_metrics.py,sha256=8FC7OCyj2QTXKRHyY7BPKfF2EAzXw0rg-r8yM19fQSc,4413
+beamlit/api/functions/get_function_trace_ids.py,sha256=MaTT6TGTJZTNkBn6Li9KfMrhpTaIM_GnTBbg-Bp_jiw,2768
 beamlit/api/functions/list_functions.py,sha256=Z9PaBzpRCv4cfEMSiBaVLnKzRoWCBys4jOXXBWOzEU8,4167
 beamlit/api/functions/update_function.py,sha256=sH-Oy2epz-X-59_eDnazzeeUsZMVNqG5J8VPe6nYJkg,4084
 beamlit/api/history/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -71,6 +77,7 @@ beamlit/api/models/delete_model.py,sha256=4uENeuBKoIooCfniM1uZFjScqgHzlEDxl7aLjA
 beamlit/api/models/get_model.py,sha256=sTE7fGpJ91svBMSKy7PGySqSOVy5g1YH3oHjhWbMr9s,4285
 beamlit/api/models/get_model_environment_logs.py,sha256=Xi4c3I0y7y_JbqUDeZEH64oLom9DZ1Uk735j47QvDT0,3939
 beamlit/api/models/get_model_metrics.py,sha256=06BcFFYS1Ml0tifIbDos9EqH6gSgGnwssKq75vhO5eU,4516
+beamlit/api/models/get_model_trace_ids.py,sha256=7LuYPCmmt98BwrMGVZBK5E4izXA7iGZ21pHmx03-YG4,2729
 beamlit/api/models/list_models.py,sha256=Keqg_qyTTGB-aJNA6JiMGnLdNwUSLIkzr08sdhhXxo4,4297
 beamlit/api/models/release_model.py,sha256=ik1HHjOUVnaVJEvbbSS1ByQ2TMzkkUbNiGUXmlTiwBo,3893
 beamlit/api/models/update_model.py,sha256=odMblGfUK6EAJHpu5mWUtpSNjFB8NvyTgqDp0JUygDA,4521
@@ -116,7 +123,7 @@ beamlit/api/workspaces/update_workspace.py,sha256=qa5DV2UJSUYuB_ibALb4E9ghKpT1Ha
 beamlit/api/workspaces/update_workspace_user_role.py,sha256=Yn9iuJ4tKtauzBiJyU4-wYUMS9g98X2Om8zs7UkzrY8,4917
 beamlit/authentication/__init__.py,sha256=wiXqRbc7E-ulrH_ueA9duOGFvXeo7-RvhSD1XbFogMo,1020
 beamlit/authentication/apikey.py,sha256=KNBTgdi0VBzBAAmSwU2X1QoB58vRbg8wkXb8-GTZCQo,657
-beamlit/authentication/authentication.py,sha256=6N4928VcgHQz5jF0aJwvIsPSGglIjhtgvAhfHHVy4H8,3118
+beamlit/authentication/authentication.py,sha256=a29-iD92T8_q8Kpx25VXEyyASlCaFsSg0s8y-SWPtns,3311
 beamlit/authentication/clientcredentials.py,sha256=6kbfTjwUkXUArJX8XZLe9ZzbEicQc19tSXBvsTpiXMk,3954
 beamlit/authentication/credentials.py,sha256=p_1xenabCbQuRz7BiFk7oTK4uCxAt_zoyku5o-jcKGE,5343
 beamlit/authentication/device_mode.py,sha256=tmr22gllKOZwBRub_QjF5pYa425x-nE8tQNpZ_EGR6g,3644
@@ -125,7 +132,7 @@ beamlit/common/generate.py,sha256=LtdCju_QayRS4lZrrb_0VHqWWvTcv4Mbf-iV1TB_Qko,75
 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=igwhNY6DPm0tiSW-_2fCkllHdTyXVthVAT6p3kggJxo,5435
+beamlit/common/settings.py,sha256=Z-UBDQKp_pAusWgwmoIwK5nl_z9xfOYYEeCCuYafr0M,5821
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/deploy/__init__.py,sha256=GS7l7Jtm2yKs7iNLKcfjYO-rAhUzggQ3xiYSf3oxLBY,91
 beamlit/deploy/deploy.py,sha256=rXpiAqlO36Ul1eadBEiwiil-6R2DkbaYHzw-qcqZ0g4,9183
@@ -143,7 +150,7 @@ beamlit/functions/mcp/mcp.py,sha256=uGqNOwEgIKdvtVPmCf9Nf_YI_k7zioLvc6jVOpszrg4,
 beamlit/functions/remote/remote.py,sha256=W60VNSNfBcVed8HVifOjcWN20StmhDjSV06AT35STaY,3276
 beamlit/functions/search/__init__.py,sha256=5NAthQ9PBwrkNg1FpLRx4flauvv0HyWuwaVS589c1Pw,49
 beamlit/functions/search/search.py,sha256=8s9ECltq7YE17j6rTxb12uY2EQY4_eTLHmwlIMThI0w,515
-beamlit/models/__init__.py,sha256=NY93_IfysVaEENSQ0pIXYDxS6cEteOlfiY4xyZHKIZY,7458
+beamlit/models/__init__.py,sha256=O16oX1-oBJqB5LZEHAfu_hZZZh6_PX2LqZpVl4fNQRs,7730
 beamlit/models/acl.py,sha256=tH67gsl_BMaviSbTaaIkO1g9cWZgJ6VgAnYVjQSzGZY,3952
 beamlit/models/agent.py,sha256=oGZBwd2Hy-i6q_up4WQ0IvOmxqouR5I1Gk8vXvfLKvc,3384
 beamlit/models/agent_chain.py,sha256=8PN8wVSayS-LoBN2nahZsOmr6r3t62H_LPDK_8fnkM8,2255
@@ -173,6 +180,9 @@ beamlit/models/function_kit.py,sha256=VrwV4EOrEqs8QgyaI7MyevRCCt2fhsTkOzfQVWXojt
 beamlit/models/function_metadata.py,sha256=7OKXazGomwAupuMTrROFD67o4D-JCBB33gznzLUj0HI,4608
 beamlit/models/function_release.py,sha256=T8SuLZBBdgGDvUg3sVluocvrKTvNxzZ6Wq3kjK4lYk4,1955
 beamlit/models/function_spec.py,sha256=8nTfXLCGqEwCdOHjhF3UH0gwu_CGQZperdhljkxcs-U,10452
+beamlit/models/get_trace_ids_response_200.py,sha256=m2uE11a9wE5c7xPTDVd4CuubQc2cPYJNaYpbcj-V1rU,1275
+beamlit/models/get_trace_logs_response_200.py,sha256=NIFtg8qcE26_oJxoRYkt1KSV2JSYUxdUu45Gxs0Qli0,1280
+beamlit/models/get_trace_response_200.py,sha256=1QePQMknfEP7DjPXb3ulf6v17430R_MJ4SPovdYK0NI,1257
 beamlit/models/get_workspace_service_accounts_response_200_item.py,sha256=ATEojS0VhRmL6a9JPWT3-fl3plosONbIWa5Jpoe4Pag,2981
 beamlit/models/increase_and_rate_metric.py,sha256=bRQ05Qi9TMYhYjtN90PeJaMEkNOdgC0A_7aQQBrWCDs,2071
 beamlit/models/integration_config.py,sha256=lq8yZR1th2qrJmGAk5aEcuvuPy4zAI9XyNol5YjeGVY,1258
@@ -243,6 +253,6 @@ beamlit/serve/app.py,sha256=OpwPjRdyHZK6J-ziPwhiRDGGa2mvCrFVcBFE6alJVOM,3071
 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.27.dist-info/METADATA,sha256=qXtSoaL19EpudttIjKmrB795nRP2ud-Y2-vlDeWXfgs,2401
-beamlit-0.0.27.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.27.dist-info/RECORD,,
+beamlit-0.0.28.dist-info/METADATA,sha256=3SqDJjs2GCjatxPbgjFDr3YHDvSHb2KC0f1Jwg41rdo,2401
+beamlit-0.0.28.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.28.dist-info/RECORD,,