beamlit 0.0.30rc36__tar.gz → 0.0.31__tar.gz

{beamlit-0.0.30rc36 → beamlit-0.0.31}/PKG-INFO

@@ -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 → beamlit-0.0.31}/pyproject.toml

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

{beamlit-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.31/src/beamlit/common/instrumentation.py

@@ -0,0 +1,123 @@
+from typing import Any
+
+from fastapi import FastAPI
+from opentelemetry import metrics, trace
+from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import (
+    OTLPMetricExporter,
+)
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+    OTLPSpanExporter,
+)
+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.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
+
+tracer: trace.Tracer | None = None
+meter: metrics.Meter | None = None
+
+
+def get_tracer() -> trace.Tracer:
+    if tracer is None:
+        raise Exception("Tracer is not initialized")
+    return tracer
+
+
+def get_meter() -> metrics.Meter:
+    if meter is None:
+        raise Exception("Meter is not initialized")
+    return meter
+
+
+def get_resource_attributes() -> Dict[str, Any]:
+    resources = Resource.create()
+    resources_dict: Dict[str, Any] = {}
+    for key in resources.attributes:
+        resources_dict[key] = resources.attributes[key]
+    settings = get_settings()
+    if settings is None:
+        raise Exception("Settings are not initialized")
+    resources_dict["workspace"] = settings.workspace
+    resources_dict["service.name"] = settings.name
+    return resources_dict
+
+
+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 | None:
+    settings = get_settings()
+    if not settings.enable_opentelemetry:
+        return None
+    return OTLPSpanExporter()
+
+
+def instrument_app(app: FastAPI):
+    global tracer
+    global meter
+    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,
+            "service.namespace": settings.workspace,
+            "service.workspace": settings.workspace,
+        }
+    )
+
+    # 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=metrics.get_meter_provider())
+    LoggingInstrumentor(tracer_provider=trace.get_tracer_provider()).instrument(
+        set_logging_format=True
+    )

{beamlit-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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-0.0.30rc36 → beamlit-0.0.31}/src/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 → beamlit-0.0.31}/src/integrationtest/agent.py

@@ -13,6 +13,7 @@ from beamlit.agents import agent
         "spec": {
             "description": "A chat agent using Beamlit to handle your tasks.",
             "model": "gpt-4o-mini",
+            "policies": ["only-us"],
         },
     },
 )

beamlit-0.0.31/src/integrationtest/functions/helloworld.py

@@ -0,0 +1,13 @@
+from beamlit.functions import function
+
+
+@function(
+    function={
+        "spec": {
+            "policies": ["only-us"],
+        }
+    }
+)
+def helloworld(query: str):
+    """A function for saying hello to the world."""
+    return "Hello from Beamlit!"