beamlit 0.0.47rc93__py3-none-any.whl → 0.0.48__py3-none-any.whl

beamlit/api/{model_providers/update_model_provider.py → accounts/update_account.py}

@@ -5,20 +5,20 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.model_provider import ModelProvider
+from ...models.account import Account
 from ...types import Response
 
 
 def _get_kwargs(
-    model_provider_name: str,
+    account_id: str,
     *,
-    body: ModelProvider,
+    body: Account,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
     _kwargs: dict[str, Any] = {
         "method": "put",
-        "url": f"/model_providers/{model_provider_name}",
+        "url": f"/accounts/{account_id}",
     }
 
     _body = body.to_dict()
@@ -30,9 +30,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[ModelProvider]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Account]:
     if response.status_code == 200:
-        response_200 = ModelProvider.from_dict(response.json())
+        response_200 = Account.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -41,7 +41,7 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[ModelProvider]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Account]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -51,29 +51,29 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 
 
 def sync_detailed(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Response[ModelProvider]:
-    """Update model provider
+    body: Account,
+) -> Response[Account]:
+    """Update account
 
-     Update an integration by ID.
+     Updates an account by name.
 
     Args:
-        model_provider_name (str):
-        body (ModelProvider): Model provider
+        account_id (str):
+        body (Account): Account
 
     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[ModelProvider]
+        Response[Account]
     """
 
     kwargs = _get_kwargs(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
         body=body,
     )
 
@@ -85,58 +85,58 @@ def sync_detailed(
 
 
 def sync(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Optional[ModelProvider]:
-    """Update model provider
+    body: Account,
+) -> Optional[Account]:
+    """Update account
 
-     Update an integration by ID.
+     Updates an account by name.
 
     Args:
-        model_provider_name (str):
-        body (ModelProvider): Model provider
+        account_id (str):
+        body (Account): Account
 
     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:
-        ModelProvider
+        Account
     """
 
     return sync_detailed(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
         client=client,
         body=body,
     ).parsed
 
 
 async def asyncio_detailed(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Response[ModelProvider]:
-    """Update model provider
+    body: Account,
+) -> Response[Account]:
+    """Update account
 
-     Update an integration by ID.
+     Updates an account by name.
 
     Args:
-        model_provider_name (str):
-        body (ModelProvider): Model provider
+        account_id (str):
+        body (Account): Account
 
     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[ModelProvider]
+        Response[Account]
     """
 
     kwargs = _get_kwargs(
-        model_provider_name=model_provider_name,
+        account_id=account_id,
         body=body,
     )
 
@@ -146,30 +146,30 @@ async def asyncio_detailed(
 
 
 async def asyncio(
-    model_provider_name: str,
+    account_id: str,
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Optional[ModelProvider]:
-    """Update model provider
+    body: Account,
+) -> Optional[Account]:
+    """Update account
 
-     Update an integration by ID.
+     Updates an account by name.
 
     Args:
-        model_provider_name (str):
-        body (ModelProvider): Model provider
+        account_id (str):
+        body (Account): Account
 
     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:
-        ModelProvider
+        Account
     """
 
     return (
         await asyncio_detailed(
-            model_provider_name=model_provider_name,
+            account_id=account_id,
             client=client,
             body=body,
         )

beamlit/api/agents/get_agent_metrics.py

@@ -5,7 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.resource_metrics import ResourceMetrics
+from ...models.resource_environment_metrics import ResourceEnvironmentMetrics
 from ...types import UNSET, Response, Unset
 
 
@@ -31,9 +31,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     if response.status_code == 200:
-        response_200 = ResourceMetrics.from_dict(response.json())
+        response_200 = ResourceEnvironmentMetrics.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -44,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,7 +58,7 @@ def sync_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -70,7 +70,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -90,7 +90,7 @@ def sync(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -102,7 +102,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return sync_detailed(
@@ -117,7 +117,7 @@ async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -129,7 +129,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -147,7 +147,7 @@ async def asyncio(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get agent metrics
 
     Args:
@@ -159,7 +159,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return (

beamlit/api/{model_providers/create_model_provider.py → default/create_account.py}

@@ -5,19 +5,19 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.model_provider import ModelProvider
+from ...models.account import Account
 from ...types import Response
 
 
 def _get_kwargs(
     *,
-    body: ModelProvider,
+    body: Account,
 ) -> dict[str, Any]:
     headers: dict[str, Any] = {}
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": "/model_providers",
+        "url": "/accounts",
     }
 
     _body = body.to_dict()
@@ -29,9 +29,9 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[ModelProvider]:
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Account]:
     if response.status_code == 200:
-        response_200 = ModelProvider.from_dict(response.json())
+        response_200 = Account.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -40,7 +40,7 @@ def _parse_response(*, client: Union[AuthenticatedClient, Client], response: htt
         return None
 
 
-def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[ModelProvider]:
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Account]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -52,21 +52,21 @@ def _build_response(*, client: Union[AuthenticatedClient, Client], response: htt
 def sync_detailed(
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Response[ModelProvider]:
-    """Create model provider
+    body: Account,
+) -> Response[Account]:
+    """Create account
 
-     Create an integration.
+     Creates an account.
 
     Args:
-        body (ModelProvider): Model provider
+        body (Account): Account
 
     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[ModelProvider]
+        Response[Account]
     """
 
     kwargs = _get_kwargs(
@@ -83,21 +83,21 @@ def sync_detailed(
 def sync(
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Optional[ModelProvider]:
-    """Create model provider
+    body: Account,
+) -> Optional[Account]:
+    """Create account
 
-     Create an integration.
+     Creates an account.
 
     Args:
-        body (ModelProvider): Model provider
+        body (Account): Account
 
     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:
-        ModelProvider
+        Account
     """
 
     return sync_detailed(
@@ -109,21 +109,21 @@ def sync(
 async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Response[ModelProvider]:
-    """Create model provider
+    body: Account,
+) -> Response[Account]:
+    """Create account
 
-     Create an integration.
+     Creates an account.
 
     Args:
-        body (ModelProvider): Model provider
+        body (Account): Account
 
     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[ModelProvider]
+        Response[Account]
     """
 
     kwargs = _get_kwargs(
@@ -138,21 +138,21 @@ async def asyncio_detailed(
 async def asyncio(
     *,
     client: AuthenticatedClient,
-    body: ModelProvider,
-) -> Optional[ModelProvider]:
-    """Create model provider
+    body: Account,
+) -> Optional[Account]:
+    """Create account
 
-     Create an integration.
+     Creates an account.
 
     Args:
-        body (ModelProvider): Model provider
+        body (Account): Account
 
     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:
-        ModelProvider
+        Account
     """
 
     return (

beamlit/api/functions/get_function_metrics.py

@@ -5,7 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.resource_metrics import ResourceMetrics
+from ...models.resource_environment_metrics import ResourceEnvironmentMetrics
 from ...types import UNSET, Response, Unset
 
 
@@ -31,9 +31,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     if response.status_code == 200:
-        response_200 = ResourceMetrics.from_dict(response.json())
+        response_200 = ResourceEnvironmentMetrics.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -44,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,7 +58,7 @@ def sync_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -70,7 +70,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -90,7 +90,7 @@ def sync(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -102,7 +102,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return sync_detailed(
@@ -117,7 +117,7 @@ async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -129,7 +129,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -147,7 +147,7 @@ async def asyncio(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get function metrics
 
     Args:
@@ -159,7 +159,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return (

beamlit/api/models/get_model_metrics.py

@@ -5,7 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.resource_metrics import ResourceMetrics
+from ...models.resource_environment_metrics import ResourceEnvironmentMetrics
 from ...types import UNSET, Response, Unset
 
 
@@ -31,9 +31,9 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     if response.status_code == 200:
-        response_200 = ResourceMetrics.from_dict(response.json())
+        response_200 = ResourceEnvironmentMetrics.from_dict(response.json())
 
         return response_200
     if client.raise_on_unexpected_status:
@@ -44,7 +44,7 @@ def _parse_response(
 
 def _build_response(
     *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -58,7 +58,7 @@ def sync_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -72,7 +72,7 @@ def sync_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -92,7 +92,7 @@ def sync(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -106,7 +106,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return sync_detailed(
@@ -121,7 +121,7 @@ async def asyncio_detailed(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Response[ResourceMetrics]:
+) -> Response[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -135,7 +135,7 @@ async def asyncio_detailed(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[ResourceMetrics]
+        Response[ResourceEnvironmentMetrics]
     """
 
     kwargs = _get_kwargs(
@@ -153,7 +153,7 @@ async def asyncio(
     *,
     client: AuthenticatedClient,
     environment: Union[Unset, str] = UNSET,
-) -> Optional[ResourceMetrics]:
+) -> Optional[ResourceEnvironmentMetrics]:
     """Get model metrics
 
      Returns metrics for a model by name.
@@ -167,7 +167,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        ResourceMetrics
+        ResourceEnvironmentMetrics
     """
 
     return (

beamlit/functions/common.py

@@ -4,7 +4,6 @@ import asyncio
 import importlib.util
 import os
 import traceback
-
 from logging import getLogger
 from typing import Union
 

beamlit/functions/mcp/mcp.py

@@ -6,13 +6,14 @@ import pydantic
 import pydantic_core
 import requests
 import typing_extensions as t
-from beamlit.authentication.authentication import AuthenticatedClient
-from beamlit.common.settings import get_settings
 from langchain_core.tools.base import BaseTool, BaseToolkit, ToolException
 from mcp.types import CallToolResult, ListToolsResult
 from pydantic.json_schema import JsonSchemaValue
 from pydantic_core import core_schema as cs
 
+from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.common.settings import get_settings
+
 settings = get_settings()
 
 TYPE_MAP = {