beamlit 0.0.30rc36__py3-none-any.whl → 0.0.31__py3-none-any.whl

beamlit/agents/chain.py

@@ -1,17 +1,17 @@
 import asyncio
-from dataclasses import dataclass
 import warnings
-from typing import Any, Callable
+from dataclasses import dataclass
+from typing import Callable
 
 import pydantic
 import typing_extensions as t
+from langchain_core.tools.base import BaseTool, ToolException
+
 from beamlit.api.agents import list_agents
 from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.common.settings import get_settings
 from beamlit.models import Agent, AgentChain
 from beamlit.run import RunClient
-from beamlit.common.settings import get_settings
-from langchain_core.tools.base import BaseTool, ToolException
-
 
 
 class ChainTool(BaseTool):

beamlit/agents/decorator.py

@@ -6,6 +6,10 @@ import importlib
 import os
 from logging import getLogger
 
+from langchain_core.tools import Tool
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.prebuilt import create_react_agent
+
 from beamlit.api.models import get_model
 from beamlit.authentication import new_client
 from beamlit.common.settings import init
@@ -13,9 +17,6 @@ from beamlit.errors import UnexpectedStatus
 from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.functions.remote.remote import RemoteToolkit
 from beamlit.models import Agent, AgentMetadata, AgentSpec
-from langchain_core.tools import Tool
-from langgraph.checkpoint.memory import MemorySaver
-from langgraph.prebuilt import create_react_agent
 
 from .chain import ChainToolkit
 from .chat import get_chat_model

beamlit/api/service_accounts/create_workspace_service_account.py

@@ -6,7 +6,9 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.create_workspace_service_account_body import CreateWorkspaceServiceAccountBody
-from ...models.create_workspace_service_account_response_200 import CreateWorkspaceServiceAccountResponse200
+from ...models.create_workspace_service_account_response_200 import (
+    CreateWorkspaceServiceAccountResponse200,
+)
 from ...types import Response
 
 

beamlit/api/service_accounts/delete_workspace_service_account.py

@@ -5,7 +5,9 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.delete_workspace_service_account_response_200 import DeleteWorkspaceServiceAccountResponse200
+from ...models.delete_workspace_service_account_response_200 import (
+    DeleteWorkspaceServiceAccountResponse200,
+)
 from ...types import Response
 
 

beamlit/api/service_accounts/get_workspace_service_accounts.py

@@ -5,7 +5,9 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
+from ...models.get_workspace_service_accounts_response_200_item import (
+    GetWorkspaceServiceAccountsResponse200Item,
+)
 from ...types import Response
 
 

beamlit/api/service_accounts/update_workspace_service_account.py

@@ -6,7 +6,9 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.update_workspace_service_account_body import UpdateWorkspaceServiceAccountBody
-from ...models.update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
+from ...models.update_workspace_service_account_response_200 import (
+    UpdateWorkspaceServiceAccountResponse200,
+)
 from ...types import Response
 
 

beamlit/authentication/authentication.py

@@ -1,15 +1,19 @@
-import os
 from dataclasses import dataclass
 from typing import Dict, Generator
 
-from beamlit.common.settings import Settings, get_settings
 from httpx import Auth, Request, Response
 
+from beamlit.common.settings import Settings, get_settings
+
 from ..client import AuthenticatedClient
 from .apikey import ApiKeyProvider
 from .clientcredentials import ClientCredentials
-from .credentials import (Credentials, current_context, load_credentials,
-                          load_credentials_from_settings)
+from .credentials import (
+    Credentials,
+    current_context,
+    load_credentials,
+    load_credentials_from_settings,
+)
 from .device_mode import BearerToken
 
 

beamlit/common/instrumentation.py

@@ -8,22 +8,16 @@ from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import (
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
     OTLPSpanExporter,
 )
-from opentelemetry.instrumentation.fastapi import (  # type: ignore
-    FastAPIInstrumentor,
-)
-from opentelemetry.instrumentation.httpx import (  # type: ignore
-    HTTPXClientInstrumentor,
-)
-from opentelemetry.instrumentation.logging import (  # type: ignore
-    LoggingInstrumentor,
-)
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+from opentelemetry.instrumentation.logging import LoggingInstrumentor
+from opentelemetry.metrics import NoOpMeterProvider
 from opentelemetry.sdk.metrics import MeterProvider
-from opentelemetry.sdk.metrics.export import (
-    PeriodicExportingMetricReader,
-)
+from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.trace import NoOpTracerProvider
 from typing_extensions import Dict
 
 from .settings import get_settings
@@ -57,11 +51,20 @@ def get_resource_attributes() -> Dict[str, Any]:
     return resources_dict
 
 
-def get_metrics_exporter() -> OTLPMetricExporter:
+def get_metrics_exporter() -> OTLPMetricExporter | None:
+    settings = get_settings()
+    if settings is None:
+        raise Exception("Settings are not initialized")
+    if not settings.enable_opentelemetry:
+        # Return None or a NoOpExporter equivalent
+        return None
     return OTLPMetricExporter()
 
 
-def get_span_exporter() -> OTLPSpanExporter:
+def get_span_exporter() -> OTLPSpanExporter | None:
+    settings = get_settings()
+    if not settings.enable_opentelemetry:
+        return None
     return OTLPSpanExporter()
 
 
@@ -71,6 +74,16 @@ def instrument_app(app: FastAPI):
     settings = get_settings()
     if settings is None:
         raise Exception("Settings are not initialized")
+
+    if not settings.enable_opentelemetry:
+        # Use NoOp implementations to stub tracing and metrics
+        trace.set_tracer_provider(NoOpTracerProvider())
+        tracer = trace.get_tracer(__name__)
+
+        metrics.set_meter_provider(NoOpMeterProvider())
+        meter = metrics.get_meter(__name__)
+        return
+
     resource = Resource.create(
         {
             "service.name": settings.name,
@@ -78,24 +91,33 @@ def instrument_app(app: FastAPI):
             "service.workspace": settings.workspace,
         }
     )
-    # Set up the TracerProvider
-    trace_provider = TracerProvider(resource=resource)
-    span_processor = BatchSpanProcessor(get_span_exporter())
-    trace_provider.add_span_processor(span_processor)
-    trace.set_tracer_provider(trace_provider)
-    tracer = trace_provider.get_tracer(__name__)
-
-    metrics_exporter = PeriodicExportingMetricReader(get_metrics_exporter())
-    meter_provider = MeterProvider(
-        resource=resource, metric_readers=[metrics_exporter]
-    )
-    metrics.set_meter_provider(meter_provider)
-    meter = meter_provider.get_meter(__name__)
 
-    FastAPIInstrumentor.instrument_app(  # type: ignore
-        app=app, tracer_provider=trace_provider, meter_provider=meter_provider
+    # Set up the TracerProvider if not already set
+    if not isinstance(trace.get_tracer_provider(), TracerProvider):
+        trace_provider = TracerProvider(resource=resource)
+        span_processor = BatchSpanProcessor(get_span_exporter())
+        trace_provider.add_span_processor(span_processor)
+        trace.set_tracer_provider(trace_provider)
+        tracer = trace_provider.get_tracer(__name__)
+    else:
+        tracer = trace.get_tracer(__name__)
+
+    # Set up the MeterProvider if not already set
+    if not isinstance(metrics.get_meter_provider(), MeterProvider):
+        metrics_exporter = PeriodicExportingMetricReader(get_metrics_exporter())
+        meter_provider = MeterProvider(
+            resource=resource, metric_readers=[metrics_exporter]
+        )
+        metrics.set_meter_provider(meter_provider)
+        meter = meter_provider.get_meter(__name__)
+    else:
+        meter = metrics.get_meter(__name__)
+
+    # Only instrument the app when OpenTelemetry is enabled
+    FastAPIInstrumentor.instrument_app(
+        app=app, tracer_provider=trace.get_tracer_provider(), meter_provider=metrics.get_meter_provider()
     )
-    HTTPXClientInstrumentor().instrument(meter_provider=meter_provider)  # type: ignore
-    LoggingInstrumentor(tracer_provider=trace_provider).instrument(
+    HTTPXClientInstrumentor().instrument(meter_provider=metrics.get_meter_provider())
+    LoggingInstrumentor(tracer_provider=trace.get_tracer_provider()).instrument(
         set_logging_format=True
     )

beamlit/common/logger.py

@@ -27,5 +27,5 @@ def init(log_level: str):
     logging.getLogger("httpx").propagate = False
 
     handler = logging.StreamHandler()
-    handler.setFormatter(ColoredFormatter(f"%(levelname)s %(name)s - %(message)s"))
+    handler.setFormatter(ColoredFormatter("%(levelname)s %(name)s - %(message)s"))
     logging.basicConfig(level=log_level, handlers=[handler])

beamlit/common/settings.py

@@ -2,6 +2,16 @@ import os
 from logging import getLogger
 from typing import Tuple, Type, Union
 
+from langchain_core.language_models.chat_models import BaseChatModel
+from langgraph.graph.graph import CompiledGraph
+from pydantic import Field
+from pydantic_settings import (
+    BaseSettings,
+    PydanticBaseSettingsSource,
+    SettingsConfigDict,
+    YamlConfigSettingsSource,
+)
+
 from beamlit.api.agents import get_agent
 from beamlit.api.functions import get_function
 from beamlit.api.models import get_model
@@ -9,11 +19,6 @@ from beamlit.client import AuthenticatedClient
 from beamlit.common.logger import init as init_logger
 from beamlit.models import Agent, Function, Model
 from beamlit.types import UNSET, Unset
-from langchain_core.language_models.chat_models import BaseChatModel
-from langgraph.graph.graph import CompiledGraph
-from pydantic import Field
-from pydantic_settings import (BaseSettings, PydanticBaseSettingsSource,
-                               SettingsConfigDict, YamlConfigSettingsSource)
 
 global SETTINGS
 SETTINGS = None
@@ -62,6 +67,7 @@ class Settings(BaseSettings):
     mcp_hub_url: str = Field(default="https://mcp-hub-server.beamlit.workers.com")
     registry_url: str = Field(default="https://us.registry.beamlit.com")
     log_level: str = Field(default="INFO")
+    enable_opentelemetry: bool = Field(default=False)
     agent: SettingsAgent = SettingsAgent()
     server: SettingsServer = SettingsServer()
     authentication: SettingsAuthentication = SettingsAuthentication()
@@ -99,7 +105,6 @@ def init_agent(
     client: AuthenticatedClient,
     destination: str = f"{os.getcwd()}/src/beamlit_generated.py",
 ):
-    from beamlit.api.agents import get_agent_deployment
     from beamlit.common.generate import generate
 
     logger = getLogger(__name__)

beamlit/deploy/deploy.py

@@ -2,12 +2,20 @@ import ast
 import json
 import os
 import sys
+import uuid
 from logging import getLogger
 from typing import Literal
 
 from beamlit.common.settings import Settings, get_settings, init
-from beamlit.models import (Agent, AgentChain, AgentSpec, EnvironmentMetadata,
-                            Flavor, Function, FunctionSpec, Runtime)
+from beamlit.models import (
+    Agent,
+    AgentSpec,
+    EnvironmentMetadata,
+    Flavor,
+    Function,
+    FunctionSpec,
+    Runtime,
+)
 
 from .format import arg_to_dict, format_agent_chain, format_parameters
 from .parser import Resource, get_description, get_parameters, get_resources
@@ -15,11 +23,16 @@ from .parser import Resource, get_description, get_parameters, get_resources
 sys.path.insert(0, os.getcwd())
 sys.path.insert(0, os.path.join(os.getcwd(), "src"))
 
+random_id = str(uuid.uuid4())[:8]
+def slugify(name: str) -> str:
+    return name.lower().replace(" ", "-").replace("_", "-")
 
 def get_runtime_image(type: str, name: str) -> str:
     settings = get_settings()
     registry_url = settings.registry_url.replace("https://", "").replace("http://", "")
     image = f"{registry_url}/{settings.workspace}/{type}s/{name}"
+    # Generate a random ID to ensure unique image tags
+    image = f"{image}:{random_id}"
     return image
 
 
@@ -110,7 +123,7 @@ def get_agent_yaml(
 apiVersion: beamlit.com/v1alpha1
 kind: Agent
 metadata:
-  name: {agent.metadata.name}
+  name: {slugify(agent.metadata.name)}
   displayName: {agent.metadata.display_name or agent.metadata.name}
   environment: {settings.environment}
   workspace: {settings.workspace}
@@ -144,7 +157,7 @@ def get_function_yaml(function: Function, settings: Settings) -> str:
 apiVersion: beamlit.com/v1alpha1
 kind: Function
 metadata:
-  name: {function.metadata.name}
+  name: {slugify(function.metadata.name)}
   displayName: {function.metadata.display_name or function.metadata.name}
   environment: {settings.environment}
 spec:
@@ -244,29 +257,29 @@ def generate_beamlit_deployment(directory: str):
         # write deployment file
         agent_dir = os.path.join(agents_dir, agent.metadata.name)
         os.makedirs(agent_dir, exist_ok=True)
-        with open(os.path.join(agent_dir, f"agent.yaml"), "w") as f:
+        with open(os.path.join(agent_dir, "agent.yaml"), "w") as f:
             content = get_agent_yaml(agent, functions, settings)
             f.write(content)
         # write dockerfile for build
-        with open(os.path.join(agent_dir, f"Dockerfile"), "w") as f:
+        with open(os.path.join(agent_dir, "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:
+        with open(os.path.join(agent_dir, "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)
         os.makedirs(function_dir, exist_ok=True)
-        with open(os.path.join(function_dir, f"function.yaml"), "w") as f:
+        with open(os.path.join(function_dir, "function.yaml"), "w") as f:
             content = get_function_yaml(function, settings)
             f.write(content)
         # write dockerfile for build
-        with open(os.path.join(function_dir, f"Dockerfile"), "w") as f:
+        with open(os.path.join(function_dir, "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:
+        with open(os.path.join(function_dir, "destination.txt"), "w") as f:
             content = function.spec.runtime.image
             f.write(content)

beamlit/deploy/format.py

@@ -1,7 +1,9 @@
 
 import ast
+
 from beamlit.models import AgentChain, StoreFunctionParameter
 
+
 def arg_to_list(arg: ast.List):
     value = []
     for v in arg.elts:

beamlit/functions/decorator.py

@@ -4,11 +4,12 @@ import json
 from collections.abc import Callable
 from logging import getLogger
 
+from langchain_core.tools import create_schema_from_function
+
 from beamlit.authentication import new_client
 from beamlit.common.settings import get_settings
 from beamlit.models import Function, FunctionKit
 from beamlit.run import RunClient
-from langchain_core.tools import create_schema_from_function
 
 logger = getLogger(__name__)
 
@@ -67,10 +68,16 @@ def kit(bl_kit: FunctionKit = None, **kwargs: dict) -> Callable:
     return wrapper
 
 
-def function(*args, function: Function = None, kit=False, **kwargs: dict) -> Callable:
+def function(*args, function: Function | dict = None, kit=False, **kwargs: dict) -> Callable:
     """Create function tools with Beamlit and LangChain integration."""
     settings = get_settings()
-
+    if function is not None and not isinstance(function, dict):
+        raise Exception(
+            'function must be a dictionary, example: @function(function={"metadata": {"name": "my_function"}})'
+        )
+    if isinstance(function, dict):
+        function = Function(**function)
+        
     def wrapper(func: Callable) -> Callable:
         if function and not func.__doc__ and function.spec and function.spec.description:
             func.__doc__ = function.spec.description

beamlit/functions/mcp/mcp.py

@@ -7,11 +7,12 @@ 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 import ListToolsResult
 
+from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.common.settings import get_settings
+
 settings = get_settings()
 
 def create_dynamic_schema(name: str, schema: dict[str, t.Any]) -> type[pydantic.BaseModel]:

beamlit/functions/remote/remote.py

@@ -1,16 +1,17 @@
 import asyncio
 import warnings
 from dataclasses import dataclass
-from typing import Callable, Literal
+from typing import Callable
 
 import pydantic
 import typing_extensions as t
+from langchain_core.tools.base import BaseTool, ToolException
+
 from beamlit.api.functions import get_function
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
 from beamlit.models import Function, StoreFunctionParameter
 from beamlit.run import RunClient
-from langchain_core.tools.base import BaseTool, ToolException
 
 
 def create_dynamic_schema(name: str, parameters: list[StoreFunctionParameter]) -> type[pydantic.BaseModel]:

beamlit/models/__init__.py

@@ -32,7 +32,9 @@ 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 .get_workspace_service_accounts_response_200_item import (
+    GetWorkspaceServiceAccountsResponse200Item,
+)
 from .increase_and_rate_metric import IncreaseAndRateMetric
 from .integration_config import IntegrationConfig
 from .integration_connection import IntegrationConnection
@@ -75,8 +77,12 @@ from .resource_deployment_metrics_query_per_second_per_region_per_code import (
     ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode,
 )
 from .resource_environment_metrics import ResourceEnvironmentMetrics
-from .resource_environment_metrics_inference_per_region import ResourceEnvironmentMetricsInferencePerRegion
-from .resource_environment_metrics_query_per_region_per_code import ResourceEnvironmentMetricsQueryPerRegionPerCode
+from .resource_environment_metrics_inference_per_region import (
+    ResourceEnvironmentMetricsInferencePerRegion,
+)
+from .resource_environment_metrics_query_per_region_per_code import (
+    ResourceEnvironmentMetricsQueryPerRegionPerCode,
+)
 from .resource_log import ResourceLog
 from .resource_metrics import ResourceMetrics
 from .runtime import Runtime

beamlit/models/resource_environment_metrics.py

@@ -8,7 +8,9 @@ from ..types import UNSET, Unset
 if TYPE_CHECKING:
     from ..models.metric import Metric
     from ..models.qps import QPS
-    from ..models.resource_environment_metrics_inference_per_region import ResourceEnvironmentMetricsInferencePerRegion
+    from ..models.resource_environment_metrics_inference_per_region import (
+        ResourceEnvironmentMetricsInferencePerRegion,
+    )
     from ..models.resource_environment_metrics_query_per_region_per_code import (
         ResourceEnvironmentMetricsQueryPerRegionPerCode,
     )

beamlit/run.py

@@ -2,6 +2,7 @@ import urllib.parse
 from typing import Any
 
 import requests
+
 from beamlit.client import AuthenticatedClient
 from beamlit.common import HTTPError, get_settings
 

beamlit/serve/app.py

@@ -7,14 +7,18 @@ from logging import getLogger
 from uuid import uuid4
 
 from asgi_correlation_id import CorrelationIdMiddleware
-from beamlit.common import HTTPError, get_settings, init
-from beamlit.common.instrumentation import (get_metrics_exporter,
-                                            get_resource_attributes,
-                                            get_span_exporter, instrument_app)
 from fastapi import FastAPI, Request, Response
 from fastapi.responses import JSONResponse
 from traceloop.sdk import Traceloop
 
+from beamlit.common import HTTPError, get_settings, init
+from beamlit.common.instrumentation import (
+    get_metrics_exporter,
+    get_resource_attributes,
+    get_span_exporter,
+    instrument_app,
+)
+
 from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
 
 sys.path.insert(0, os.getcwd())
@@ -37,13 +41,14 @@ logger.info(
     f" on {settings.server.host}:{settings.server.port}"
 )
 
-Traceloop.init(
-    app_name=settings.name,
-    exporter=get_span_exporter(),
-    metrics_exporter=get_metrics_exporter(),
-    resource_attributes=get_resource_attributes(),
-    should_enrich_metrics=os.getenv("ENRICHED_METRICS", "false") == "true",
-)
+if settings.enable_opentelemetry:
+    Traceloop.init(
+        app_name=settings.name,
+        exporter=get_span_exporter(),
+        metrics_exporter=get_metrics_exporter(),
+        resource_attributes=get_resource_attributes(),
+        should_enrich_metrics=os.getenv("ENRICHED_METRICS", "false") == "true",
+    )
 
 app = FastAPI(docs_url=None, redoc_url=None)
 app.add_middleware(
@@ -55,6 +60,7 @@ app.add_middleware(AddProcessTimeHeader)
 app.add_middleware(AccessLogMiddleware)
 instrument_app(app)
 
+
 @app.get("/health")
 async def health():
     return {"status": "ok"}
@@ -67,7 +73,7 @@ async def root(request: Request):
     try:
         body = await request.json()
 
-        original_func = getattr(func, '__wrapped__', func)
+        original_func = getattr(func, "__wrapped__", func)
         if asyncio.iscoroutinefunction(func) or asyncio.iscoroutinefunction(original_func):
             response = await func(body)
         else:

{beamlit-0.0.30rc36.dist-info → beamlit-0.0.31.dist-info}/METADATA

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

{beamlit-0.0.30rc36.dist-info → beamlit-0.0.31.dist-info}/RECORD

@@ -2,12 +2,12 @@ beamlit/__init__.py,sha256=545gFC-wLLwUktWcOAjUWe_Glha40tBetRTOYSfHnbI,164
 beamlit/client.py,sha256=PnR6ybZk5dLIJPnDKAf2epHOeQC_7yL0fG4muvphHjA,12695
 beamlit/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
 beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
-beamlit/run.py,sha256=BCVQXdtIzJedxqGSWttYlJphO5ru7VNDYrUAQmCNq_c,1445
+beamlit/run.py,sha256=HtDYDjD7oVfQ8r3T5_t4qN5UDJOJfsQILi45Z21ArAg,1446
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=nf1iwQwGtCG6nDqyVhxfWoqR6dv6X3bvSpCeqkTCFaM,101
-beamlit/agents/chain.py,sha256=HzBs3nI4xaH86I_r-M-HGGp6roXsJxMx-qXl_GrJaY0,2831
+beamlit/agents/chain.py,sha256=vfCjiFHuu02uTTGicxMlFzjyICQkIjpXrBGs-7uJEsg,2826
 beamlit/agents/chat.py,sha256=gVyv4FGBdQTDhdutX8l64OUNa6Fdqaw4eCfEDRH0IPQ,3558
-beamlit/agents/decorator.py,sha256=4KxPcu6hPoB__1OdzpEJVD5N3QVa_syKd_Y6n6lVEjQ,9715
+beamlit/agents/decorator.py,sha256=RAhe7wHKJnjMpmU7KWKcM6EZjdm3pgUBhQQUP2T6GZY,9716
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
 beamlit/api/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/agents/create_agent.py,sha256=t5Pr62My2EhQlcIY71MrI73-0_q5Djr3a_Ybt9MIiQQ,3587
@@ -98,12 +98,12 @@ beamlit/api/privateclusters/update_private_cluster.py,sha256=Urb5GGuVcSwQy2WBlru
 beamlit/api/privateclusters/update_private_cluster_health.py,sha256=PxCyl5ZEIqpQ_PKIr9ErqjZcoTSKlTZS1YgTT1JwhCg,2572
 beamlit/api/service_accounts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/service_accounts/create_api_key_for_service_account.py,sha256=Ee9CXzYqVh9OTJeai8bKaclznuWpJiPd74aPHZYZdeQ,4525
-beamlit/api/service_accounts/create_workspace_service_account.py,sha256=XFAvmmjrwaenuWEPw0YBe7QeiYmXNX8WmcsYj6VYHa4,4639
+beamlit/api/service_accounts/create_workspace_service_account.py,sha256=D3NkGoPZyMi6ibj29F7xbJd5aLI2jw2vLSl1qDRGhW4,4648
 beamlit/api/service_accounts/delete_api_key_for_service_account.py,sha256=bS2OWtO7Dn8FU2uaTe9h7VCxiobti8Ka2NtTUKBd31Q,2621
-beamlit/api/service_accounts/delete_workspace_service_account.py,sha256=_FVYHH29JKQGZqCFSih2-gU92OR-6Am1CIQX6VJ_goo,4127
-beamlit/api/service_accounts/get_workspace_service_accounts.py,sha256=R67By_FV4Nv8tVT2b8mgsC_Dnwy0hla353-EOupBizY,4142
+beamlit/api/service_accounts/delete_workspace_service_account.py,sha256=XEz6-DkWZTmuaTOMooaywKvBsGilBYR-xG32RsXP5Mw,4136
+beamlit/api/service_accounts/get_workspace_service_accounts.py,sha256=6c7T_WsXCuotQomm4zkAkAkhdLEcgVyvon9aYHPiEiI,4151
 beamlit/api/service_accounts/list_api_keys_for_service_account.py,sha256=skp0PmjxfW7EGAyRcVAye_SKfryKjvsTuhSRxeIaXDo,4066
-beamlit/api/service_accounts/update_workspace_service_account.py,sha256=IWVMnChlL27mOCH44GYSA6ep5AJuLXzW2oDsqs7mYIQ,4903
+beamlit/api/service_accounts/update_workspace_service_account.py,sha256=Y4rQL8Yx8yRZSzTJz5-ESP5zQ0juVhuoX6xwpg5ZM00,4912
 beamlit/api/store/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/store/get_store_agent.py,sha256=UjvsRXyFsXZLf4utQjpiOiPM7KxlRj4C-x_ac5EXoLk,3624
 beamlit/api/store/get_store_function.py,sha256=kEbg91jJh0XAXmipjzwg931mxkMCZ820PxiJ89zYbso,3756
@@ -124,35 +124,35 @@ 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=8R-3WdQSykNjCbebAW2p8Glvw5nlAmSEZr6Ylo-vPuc,3377
+beamlit/authentication/authentication.py,sha256=ODQCc00RvCOZNaiGG5ctylNnE-JVCugePyJedJ0NYsM,3361
 beamlit/authentication/clientcredentials.py,sha256=cxZPPu--CgizwqX0pdfFQ91gJt1EFKwyy-aBB_dXX7I,3990
 beamlit/authentication/credentials.py,sha256=p_1xenabCbQuRz7BiFk7oTK4uCxAt_zoyku5o-jcKGE,5343
 beamlit/authentication/device_mode.py,sha256=tmr22gllKOZwBRub_QjF5pYa425x-nE8tQNpZ_EGR6g,3644
 beamlit/common/__init__.py,sha256=vj4_boIBVitMsaQR8BqBqE2eupOIh6MWBAYlYyCCH98,341
 beamlit/common/error.py,sha256=f9oJDFxhoHK-vpjxBgEp0NwWIk0N_THPemUI7uQxVzU,270
 beamlit/common/generate.py,sha256=LtdCju_QayRS4lZrrb_0VHqWWvTcv4Mbf-iV1TB_Qko,7522
-beamlit/common/instrumentation.py,sha256=MsBDfFcMYqGDiHHj4j5hLHE4EWxZExkhmCeFS3SKzJY,3181
-beamlit/common/logger.py,sha256=VFRbaZh93n8ZGugeeYKe88IP2nI3g2JNa7XN4j8wVJE,1116
+beamlit/common/instrumentation.py,sha256=GVYeat7qCcqzDoKSYig3s8ZCC172R9JiQIr3Evv3kik,4293
+beamlit/common/logger.py,sha256=nN_dSOl4bs13QU3Rod-w3e3jYOnlSrHx3_bs-ACY6Aw,1115
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
-beamlit/common/settings.py,sha256=2V06qodZHwPBqRDdpPyZhdCO7HeMjWr2rkEePOgmHqE,5936
+beamlit/common/settings.py,sha256=_4oCVrJZOMaTZoK2Zzo2DWqtUyBsspuOH3iAJ_nU0tw,5923
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/deploy/__init__.py,sha256=GS7l7Jtm2yKs7iNLKcfjYO-rAhUzggQ3xiYSf3oxLBY,91
-beamlit/deploy/deploy.py,sha256=AbQxSVGAPbwBn4wKPiMEF5GjtGa8q4u3DBuhnYdz4GI,9706
-beamlit/deploy/format.py,sha256=78tOoeNPJ8969AhQTeFlIwZgQ3un8gmTSMmrYbRQSds,1818
+beamlit/deploy/deploy.py,sha256=xo2fAdF6jR1cG75sXMzpqsJlG_qGAKVnMwjuDa_8sk8,9936
+beamlit/deploy/format.py,sha256=PJ8kU7Y1pwiS3tqqyvFaag9LO3jju-4ua574163VPk4,1820
 beamlit/deploy/parser.py,sha256=Ga0poCZkoRnuTw082QnTcNGCBJncoRAnVsn8-1FsaJE,6907
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
-beamlit/functions/decorator.py,sha256=NsdIWs55Yhn-TUO855DIE2AarV-euXtl7Qt5c2W3AUo,3239
+beamlit/functions/decorator.py,sha256=YNNU8MZXV2C6gIjxaYhp-FSWGLc9I_XLsIE24zQ3fr8,3541
 beamlit/functions/github/__init__.py,sha256=gYnUkeegukOfbymdabuuJkScvH-_ZJygX05BoqkPn0o,49
 beamlit/functions/github/github.py,sha256=FajzLCNkpXcwfgnC0l9rOGT2eSPLCz8-qrMzK9N_ZNc,598
 beamlit/functions/github/kit/__init__.py,sha256=jBwPqZv6C23_utukohxqXZwrlicNlI7PYPUj0Den7Cw,136
 beamlit/functions/github/kit/pull_request.py,sha256=wQVeRBakiqu-2ouflO8p1z7D5u07KNsitwyNRrp0KjM,1357
 beamlit/functions/math/__init__.py,sha256=wie4WME8jT-WpFRrtu-lDlHW31Mg6K2cwstjkUdLF3o,43
 beamlit/functions/math/math.py,sha256=CpoLJGwuvwCPGnVC8k9GYuIyvfUYPDQHKlZg3cx-z-A,1049
-beamlit/functions/mcp/mcp.py,sha256=gXzvUPAKmvGQPHUKTVzMiINukxpw1BV6H8M2iOWvIGU,4348
-beamlit/functions/remote/remote.py,sha256=x2eHh4oYkWwHuZWwg5XeylY-E9Opa6SzfN_0396ePZw,3775
+beamlit/functions/mcp/mcp.py,sha256=-LL7O35vTlcYfF1MSlEY83rBKKShJTaHB-y9MRPAEiU,4349
+beamlit/functions/remote/remote.py,sha256=AL8WhD7yXZ18h3iU4vE9E4JtJt0n_i-ZwlbBDoolnEs,3767
 beamlit/functions/search/__init__.py,sha256=5NAthQ9PBwrkNg1FpLRx4flauvv0HyWuwaVS589c1Pw,49
 beamlit/functions/search/search.py,sha256=8s9ECltq7YE17j6rTxb12uY2EQY4_eTLHmwlIMThI0w,515
-beamlit/models/__init__.py,sha256=0BldOldXzMn1A6d1bOhsU8mk7i3uRmh33_U-0IYjIRY,7803
+beamlit/models/__init__.py,sha256=CfMLR86qnFW6mz2fp6t2f91Vr62HmjY7qgTuvZT79Is,7830
 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
@@ -223,7 +223,7 @@ beamlit/models/qps.py,sha256=Us3PQTfeM5KN-QwLXkJ2XESJDIyTvD1tSJR9taClIog,1684
 beamlit/models/resource_deployment_metrics.py,sha256=TaV4-xbI79FCfzFcgDQ-A3EjJTByMKA42xgvS7pcT10,8875
 beamlit/models/resource_deployment_metrics_inference_per_second_per_region.py,sha256=phgZ9vJ3s7pYIYo14mgyav6ZrJnY3IS7y5T6To1IDjE,2624
 beamlit/models/resource_deployment_metrics_query_per_second_per_region_per_code.py,sha256=uCzjEHEy3j3j3Jcdul097N6hiKAHBhM8laQ-TxDs65c,2381
-beamlit/models/resource_environment_metrics.py,sha256=1HOUGH92Depu7fGpHiIEI9x2XBC6x8xotcvnTe9TnEY,7533
+beamlit/models/resource_environment_metrics.py,sha256=tVbEvRXdNgcu_c6bVQAkVDYiYqK3N-TsTWMWdSB3q28,7550
 beamlit/models/resource_environment_metrics_inference_per_region.py,sha256=pIc14XdNeKq50WGWU0FnmNqtFU5S8y_q3gomLA-dOgM,2580
 beamlit/models/resource_environment_metrics_inference_per_second_per_region.py,sha256=CmJx_tjhsfHS1hMHOwOkuVPypbG3xyZ_GCG3A2uPhr4,2629
 beamlit/models/resource_environment_metrics_query_per_region_per_code.py,sha256=G2qxM4jRKavyC96K9CO6AlF_7_K6aHtV4dMqT4n4vcA,2307
@@ -252,10 +252,10 @@ beamlit/models/websocket_channel.py,sha256=jg3vN7yS_oOIwGtndtIUr1LsyEA58RXLXahqS
 beamlit/models/workspace.py,sha256=l__bIpbA4oJvxXo7UbEoCcqkvu9MiNt5aXXpZ3bgwHg,4309
 beamlit/models/workspace_labels.py,sha256=WbnUY6eCTkUNdY7hhhSF-KQCl8fWFfkCf7hzCTiNp4A,1246
 beamlit/models/workspace_user.py,sha256=70CcifQWYbeWG7TDui4pblTzUe5sVK0AS19vNCzKE8g,3423
-beamlit/serve/app.py,sha256=k9THiOqcPs4lcQQhZaucLcboXCR_5AKP9AFCMNRbdPI,3559
+beamlit/serve/app.py,sha256=QzReZZjGFQWhNA1XBDq81rsn9zpEihtLZLrfZi9ThT0,3554
 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.30rc36.dist-info/METADATA,sha256=6OtNFDOop4RWU3ATSYjTG9g94ygVfovGM9Ef6HhQejU,2405
-beamlit-0.0.30rc36.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.30rc36.dist-info/RECORD,,
+beamlit-0.0.31.dist-info/METADATA,sha256=AiAjQ4okR_M97rDaam6B7Bw5tax6IW3MZ19lRHzqKmw,2401
+beamlit-0.0.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.31.dist-info/RECORD,,