beamlit 0.0.34rc78__tar.gz → 0.0.34rc80__tar.gz

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.34rc78
+Version: 0.0.34rc80
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
@@ -15,9 +15,27 @@ Requires-Dist: langgraph<0.3.0,>=0.2.40
 Requires-Dist: mcp>=1.1.2
 Requires-Dist: opentelemetry-api>=1.28.2
 Requires-Dist: opentelemetry-exporter-otlp>=1.28.2
+Requires-Dist: opentelemetry-instrumentation-anthropic>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-chromadb>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-cohere>=0.35.0
 Requires-Dist: opentelemetry-instrumentation-fastapi>=0.49b2
+Requires-Dist: opentelemetry-instrumentation-groq>=0.35.0
 Requires-Dist: opentelemetry-instrumentation-httpx>=0.49b2
+Requires-Dist: opentelemetry-instrumentation-lancedb>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-langchain>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-llamaindex>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-marqo>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-milvus>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-mistralai>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-ollama>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-openai>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-pinecone>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-qdrant>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-replicate>=0.35.0
 Requires-Dist: opentelemetry-instrumentation-system-metrics>=0.50b0
+Requires-Dist: opentelemetry-instrumentation-together>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-watsonx>=0.35.0
+Requires-Dist: opentelemetry-instrumentation-weaviate>=0.35.0
 Requires-Dist: opentelemetry-sdk>=1.28.2
 Requires-Dist: pydantic-settings<2.7.0,>=2.6.1
 Requires-Dist: pydantic<2.11.0,>=2.10.3
@@ -25,7 +43,6 @@ Requires-Dist: pyjwt>=2.10.1
 Requires-Dist: python-dateutil>=2.8.0
 Requires-Dist: pyyaml<6.1.0,>=6.0.2
 Requires-Dist: requests<2.33.0,>=2.32.3
-Requires-Dist: traceloop-sdk>=0.33.12
 Description-Content-Type: text/markdown
 
 # beamlit

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.34.pre-78"
+version = "0.0.34.pre-80"
 description = "Add your description here"
 readme = "README.md"
 authors = [
@@ -26,7 +26,6 @@ dependencies = [
     "fastapi[standard]>=0.115.4,<0.116.0",
     "asgi-correlation-id>=4.3.4,<5.0.0",
     "mcp>=1.1.2",
-    "traceloop-sdk>=0.33.12",
     "opentelemetry-sdk>=1.28.2",
     "opentelemetry-exporter-otlp>=1.28.2",
     "opentelemetry-api>=1.28.2",
@@ -34,6 +33,24 @@ dependencies = [
     "opentelemetry-instrumentation-httpx>=0.49b2",
     "opentelemetry-instrumentation-system-metrics>=0.50b0",
     "pyjwt>=2.10.1",
+    "opentelemetry-instrumentation-anthropic>=0.35.0",
+    "opentelemetry-instrumentation-chromadb>=0.35.0",
+    "opentelemetry-instrumentation-cohere>=0.35.0",
+    "opentelemetry-instrumentation-groq>=0.35.0",
+    "opentelemetry-instrumentation-lancedb>=0.35.0",
+    "opentelemetry-instrumentation-langchain>=0.35.0",
+    "opentelemetry-instrumentation-llamaindex>=0.35.0",
+    "opentelemetry-instrumentation-marqo>=0.35.0",
+    "opentelemetry-instrumentation-milvus>=0.35.0",
+    "opentelemetry-instrumentation-mistralai>=0.35.0",
+    "opentelemetry-instrumentation-ollama>=0.35.0",
+    "opentelemetry-instrumentation-openai>=0.35.0",
+    "opentelemetry-instrumentation-pinecone>=0.35.0",
+    "opentelemetry-instrumentation-qdrant>=0.35.0",
+    "opentelemetry-instrumentation-replicate>=0.35.0",
+    "opentelemetry-instrumentation-together>=0.35.0",
+    "opentelemetry-instrumentation-watsonx>=0.35.0",
+    "opentelemetry-instrumentation-weaviate>=0.35.0",
 ]
 [dependency-groups]
 dev = [

beamlit-0.0.34rc80/src/beamlit/api/workspaces/workspace_quotas_request.py

@@ -0,0 +1,97 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...types import Response
+
+
+def _get_kwargs(
+    workspace_name: str,
+) -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "put",
+        "url": f"/workspaces/{workspace_name}/quotas",
+    }
+
+    return _kwargs
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == 200:
+        return None
+    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(
+    workspace_name: str,
+    *,
+    client: AuthenticatedClient,
+) -> Response[Any]:
+    """Request workspace quotas
+
+     Send a request to update quotas for a workspace.
+
+    Args:
+        workspace_name (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(
+        workspace_name=workspace_name,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    workspace_name: str,
+    *,
+    client: AuthenticatedClient,
+) -> Response[Any]:
+    """Request workspace quotas
+
+     Send a request to update quotas for a workspace.
+
+    Args:
+        workspace_name (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(
+        workspace_name=workspace_name,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/common/instrumentation.py

@@ -1,5 +1,6 @@
+import importlib
 import logging
-from typing import Any
+from typing import Any, Optional, Type
 
 from fastapi import FastAPI
 from opentelemetry import _logs, metrics, trace
@@ -13,8 +14,9 @@ from opentelemetry.exporter.otlp.proto.http.metric_exporter import (
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
     OTLPSpanExporter,
 )
-from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
-from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+from opentelemetry.instrumentation.fastapi import (  # type: ignore
+    FastAPIInstrumentor,
+)
 from opentelemetry.metrics import NoOpMeterProvider
 from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
 from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
@@ -34,6 +36,8 @@ tracer: trace.Tracer | None = None
 meter: metrics.Meter | None = None
 logger: LoggerProvider | None = None
 
+log = logging.getLogger(__name__)
+
 
 def auth_headers() -> Dict[str, str]:
     settings = get_settings()
@@ -82,6 +86,121 @@ def get_log_exporter() -> OTLPLogExporter | None:
     return OTLPLogExporter(headers=auth_headers())
 
 
+def _import_class(module_path: str, class_name: str) -> Optional[Type]:  # type: ignore
+    """Dynamically import a class from a module path."""
+    try:
+        module = importlib.import_module(module_path)
+        return getattr(module, class_name)
+    except (ImportError, AttributeError) as e:
+        log.error(f"Could not import {class_name} from {module_path}: {str(e)}")
+        return None
+
+
+# Define mapping of instrumentor info: (module path, class name, required package)
+INSTRUMENTOR_CONFIGS = {
+    "httpx": (
+        "opentelemetry.instrumentation.httpx",
+        "HTTPXClientInstrumentor",
+        "httpx",
+    ),
+    "anthropic": (
+        "opentelemetry.instrumentation.anthropic",
+        "AnthropicInstrumentor",
+        "anthropic",
+    ),
+    "chroma": (
+        "opentelemetry.instrumentation.chroma",
+        "ChromaInstrumentor",
+        "chromadb",
+    ),
+    "cohere": (
+        "opentelemetry.instrumentation.cohere",
+        "CohereInstrumentor",
+        "cohere",
+    ),
+    "groq": ("opentelemetry.instrumentation.groq", "GroqInstrumentor", "groq"),
+    "lance": (
+        "opentelemetry.instrumentation.lance",
+        "LanceInstrumentor",
+        "pylance",
+    ),
+    "langchain": (
+        "opentelemetry.instrumentation.langchain",
+        "LangchainInstrumentor",
+        "langchain",
+    ),
+    "llama_index": (
+        "opentelemetry.instrumentation.llama_index",
+        "LlamaIndexInstrumentor",
+        "llama_index",
+    ),
+    "marqo": (
+        "opentelemetry.instrumentation.marqo",
+        "MarqoInstrumentor",
+        "marqo",
+    ),
+    "milvus": (
+        "opentelemetry.instrumentation.milvus",
+        "MilvusInstrumentor",
+        "pymilvus",
+    ),
+    "mistralai": (
+        "opentelemetry.instrumentation.mistralai",
+        "MistralAiInstrumentor",
+        "mistralai",
+    ),
+    "ollama": (
+        "opentelemetry.instrumentation.ollama",
+        "OllamaInstrumentor",
+        "ollama",
+    ),
+    "openai": (
+        "opentelemetry.instrumentation.openai",
+        "OpenAIInstrumentor",
+        "openai",
+    ),
+    "pinecone": (
+        "opentelemetry.instrumentation.pinecone",
+        "PineconeInstrumentor",
+        "pinecone",
+    ),
+    "qdrant": (
+        "opentelemetry.instrumentation.qdrant",
+        "QdrantInstrumentor",
+        "qdrant_client",
+    ),
+    "replicate": (
+        "opentelemetry.instrumentation.replicate",
+        "ReplicateInstrumentor",
+        "replicate",
+    ),
+    "together": (
+        "opentelemetry.instrumentation.together",
+        "TogetherAiInstrumentor",
+        "together",
+    ),
+    "watsonx": (
+        "opentelemetry.instrumentation.watsonx",
+        "WatsonxInstrumentor",
+        "ibm_watson_machine_learning",
+    ),
+    "weaviate": (
+        "opentelemetry.instrumentation.weaviate",
+        "WeaviateInstrumentor",
+        "weaviate",
+    ),
+}
+
+
+def _is_package_installed(package_name: str) -> bool:
+    """Check if a package is installed."""
+    try:
+        importlib.import_module(package_name)
+        return True
+    except (ImportError, ModuleNotFoundError):
+        return False
+
+
 def instrument_app(app: FastAPI):
     global tracer
     global meter
@@ -106,7 +225,7 @@ def instrument_app(app: FastAPI):
     # 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())
+        span_processor = BatchSpanProcessor(get_span_exporter())  # type: ignore
         trace_provider.add_span_processor(span_processor)
         trace.set_tracer_provider(trace_provider)
         tracer = trace_provider.get_tracer(__name__)
@@ -115,7 +234,7 @@ def instrument_app(app: FastAPI):
 
     # Set up the MeterProvider if not already set
     if not isinstance(metrics.get_meter_provider(), MeterProvider):
-        metrics_exporter = PeriodicExportingMetricReader(get_metrics_exporter())
+        metrics_exporter = PeriodicExportingMetricReader(get_metrics_exporter())  # type: ignore
         meter_provider = MeterProvider(
             resource=resource, metric_readers=[metrics_exporter]
         )
@@ -128,7 +247,7 @@ def instrument_app(app: FastAPI):
         logger_provider = LoggerProvider(resource=resource)
         set_logger_provider(logger_provider)
         logger_provider.add_log_record_processor(
-            BatchLogRecordProcessor(get_log_exporter())
+            BatchLogRecordProcessor(get_log_exporter())  # type: ignore
         )
         handler = LoggingHandler(
             level=logging.NOTSET, logger_provider=logger_provider
@@ -138,8 +257,27 @@ def instrument_app(app: FastAPI):
         logger_provider = _logs.get_logger_provider()
 
     # Only instrument the app when OpenTelemetry is enabled
-    FastAPIInstrumentor.instrument_app(app)
-    HTTPXClientInstrumentor().instrument()
+    FastAPIInstrumentor.instrument_app(app)  # type: ignore
+
+    for name, (
+        module_path,
+        class_name,
+        required_package,
+    ) in INSTRUMENTOR_CONFIGS.items():
+        if _is_package_installed(required_package):
+            instrumentor_class = _import_class(module_path, class_name)  # type: ignore
+            if instrumentor_class:
+                try:
+                    instrumentor_class().instrument()
+                    log.info(f"Successfully instrumented {name}")
+                except Exception as e:
+                    log.error(f"Failed to instrument {name}: {str(e)}")
+            else:
+                log.error(f"Could not load instrumentor for {name}")
+        else:
+            log.debug(
+                f"Skipping {name} instrumentation - required package '{required_package}' not installed"
+            )
 
 
 def shutdown_instrumentation():

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/deploy/deploy.py

@@ -227,7 +227,7 @@ def clean_auto_generated(
                         except yaml.YAMLError:
                             continue
 
-def generate_beamlit_deployment(directory: str):
+def generate_beamlit_deployment(directory: str, name: str):
     """
     Generates all necessary deployment files for Beamlit agents and functions.
 
@@ -247,6 +247,8 @@ def generate_beamlit_deployment(directory: str):
     agents: list[tuple[Resource, Agent]] = []
     for resource in get_resources("agent", settings.server.directory):
         agent = get_beamlit_deployment_from_resource(resource)
+        if name and agent.metadata.name != name:
+            agent.metadata.name = slugify(name)
         if agent:
             agents.append((resource, agent))
     for resource in get_resources("function", settings.server.directory):

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/models/__init__.py

@@ -59,7 +59,6 @@ from .metrics_rps_per_code import MetricsRpsPerCode
 from .model import Model
 from .model_metadata import ModelMetadata
 from .model_private_cluster import ModelPrivateCluster
-from .model_provider import ModelProvider
 from .model_release import ModelRelease
 from .model_spec import ModelSpec
 from .owner_fields import OwnerFields
@@ -188,7 +187,6 @@ __all__ = (
     "Model",
     "ModelMetadata",
     "ModelPrivateCluster",
-    "ModelProvider",
     "ModelRelease",
     "ModelSpec",
     "OwnerFields",

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/models/agent_spec.py

@@ -32,6 +32,7 @@ class AgentSpec:
         policies (Union[Unset, list[str]]):
         private_clusters (Union[Unset, ModelPrivateCluster]): Private cluster where the model deployment is deployed
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
+        sandbox (Union[Unset, bool]): Sandbox mode
         serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
         agent_chain (Union[Unset, list['AgentChain']]): Agent chain
         description (Union[Unset, str]): Agent description
@@ -49,6 +50,7 @@ class AgentSpec:
     policies: Union[Unset, list[str]] = UNSET
     private_clusters: Union[Unset, "ModelPrivateCluster"] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
     agent_chain: Union[Unset, list["AgentChain"]] = UNSET
     description: Union[Unset, str] = UNSET
@@ -92,6 +94,8 @@ class AgentSpec:
         if self.runtime and not isinstance(self.runtime, Unset):
             runtime = self.runtime.to_dict()
 
+        sandbox = self.sandbox
+
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
         if self.serverless_config and not isinstance(self.serverless_config, Unset):
             serverless_config = self.serverless_config.to_dict()
@@ -136,6 +140,8 @@ class AgentSpec:
             field_dict["privateClusters"] = private_clusters
         if runtime is not UNSET:
             field_dict["runtime"] = runtime
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if serverless_config is not UNSET:
             field_dict["serverlessConfig"] = serverless_config
         if agent_chain is not UNSET:
@@ -208,6 +214,8 @@ class AgentSpec:
         else:
             runtime = Runtime.from_dict(_runtime)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _serverless_config = d.pop("serverlessConfig", UNSET)
         serverless_config: Union[Unset, ServerlessConfig]
         if isinstance(_serverless_config, Unset):
@@ -246,6 +254,7 @@ class AgentSpec:
             policies=policies,
             private_clusters=private_clusters,
             runtime=runtime,
+            sandbox=sandbox,
             serverless_config=serverless_config,
             agent_chain=agent_chain,
             description=description,

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/models/core_spec.py

@@ -30,6 +30,7 @@ class CoreSpec:
         policies (Union[Unset, list[str]]):
         private_clusters (Union[Unset, ModelPrivateCluster]): Private cluster where the model deployment is deployed
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
+        sandbox (Union[Unset, bool]): Sandbox mode
         serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
     """
 
@@ -41,6 +42,7 @@ class CoreSpec:
     policies: Union[Unset, list[str]] = UNSET
     private_clusters: Union[Unset, "ModelPrivateCluster"] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -78,6 +80,8 @@ class CoreSpec:
         if self.runtime and not isinstance(self.runtime, Unset):
             runtime = self.runtime.to_dict()
 
+        sandbox = self.sandbox
+
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
         if self.serverless_config and not isinstance(self.serverless_config, Unset):
             serverless_config = self.serverless_config.to_dict()
@@ -101,6 +105,8 @@ class CoreSpec:
             field_dict["privateClusters"] = private_clusters
         if runtime is not UNSET:
             field_dict["runtime"] = runtime
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if serverless_config is not UNSET:
             field_dict["serverlessConfig"] = serverless_config
 
@@ -159,6 +165,8 @@ class CoreSpec:
         else:
             runtime = Runtime.from_dict(_runtime)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _serverless_config = d.pop("serverlessConfig", UNSET)
         serverless_config: Union[Unset, ServerlessConfig]
         if isinstance(_serverless_config, Unset):
@@ -175,6 +183,7 @@ class CoreSpec:
             policies=policies,
             private_clusters=private_clusters,
             runtime=runtime,
+            sandbox=sandbox,
             serverless_config=serverless_config,
         )
 

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/models/function_spec.py

@@ -32,6 +32,7 @@ class FunctionSpec:
         policies (Union[Unset, list[str]]):
         private_clusters (Union[Unset, ModelPrivateCluster]): Private cluster where the model deployment is deployed
         runtime (Union[Unset, Runtime]): Set of configurations for a deployment
+        sandbox (Union[Unset, bool]): Sandbox mode
         serverless_config (Union[Unset, ServerlessConfig]): Configuration for a serverless deployment
         description (Union[Unset, str]): Function description, very important for the agent function to work with an LLM
         kit (Union[Unset, list['FunctionKit']]): The kit of the function deployment
@@ -48,6 +49,7 @@ class FunctionSpec:
     policies: Union[Unset, list[str]] = UNSET
     private_clusters: Union[Unset, "ModelPrivateCluster"] = UNSET
     runtime: Union[Unset, "Runtime"] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     serverless_config: Union[Unset, "ServerlessConfig"] = UNSET
     description: Union[Unset, str] = UNSET
     kit: Union[Unset, list["FunctionKit"]] = UNSET
@@ -89,6 +91,8 @@ class FunctionSpec:
         if self.runtime and not isinstance(self.runtime, Unset):
             runtime = self.runtime.to_dict()
 
+        sandbox = self.sandbox
+
         serverless_config: Union[Unset, dict[str, Any]] = UNSET
         if self.serverless_config and not isinstance(self.serverless_config, Unset):
             serverless_config = self.serverless_config.to_dict()
@@ -130,6 +134,8 @@ class FunctionSpec:
             field_dict["privateClusters"] = private_clusters
         if runtime is not UNSET:
             field_dict["runtime"] = runtime
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if serverless_config is not UNSET:
             field_dict["serverlessConfig"] = serverless_config
         if description is not UNSET:
@@ -198,6 +204,8 @@ class FunctionSpec:
         else:
             runtime = Runtime.from_dict(_runtime)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _serverless_config = d.pop("serverlessConfig", UNSET)
         serverless_config: Union[Unset, ServerlessConfig]
         if isinstance(_serverless_config, Unset):
@@ -232,6 +240,7 @@ class FunctionSpec:
             policies=policies,
             private_clusters=private_clusters,
             runtime=runtime,
+            sandbox=sandbox,
             serverless_config=serverless_config,
             description=description,
             kit=kit,

{beamlit-0.0.34rc78 → beamlit-0.0.34rc80}/src/beamlit/models/integration_connection_spec.py

@@ -20,11 +20,13 @@ class IntegrationConnectionSpec:
     Attributes:
         config (Union[Unset, IntegrationConnectionConfig]): Integration config
         integration (Union[Unset, str]): Integration type
+        sandbox (Union[Unset, bool]): Sandbox mode
         secret (Union[Unset, IntegrationConnectionSecret]): Integration secret
     """
 
     config: Union[Unset, "IntegrationConnectionConfig"] = UNSET
     integration: Union[Unset, str] = UNSET
+    sandbox: Union[Unset, bool] = UNSET
     secret: Union[Unset, "IntegrationConnectionSecret"] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -35,6 +37,8 @@ class IntegrationConnectionSpec:
 
         integration = self.integration
 
+        sandbox = self.sandbox
+
         secret: Union[Unset, dict[str, Any]] = UNSET
         if self.secret and not isinstance(self.secret, Unset):
             secret = self.secret.to_dict()
@@ -46,6 +50,8 @@ class IntegrationConnectionSpec:
             field_dict["config"] = config
         if integration is not UNSET:
             field_dict["integration"] = integration
+        if sandbox is not UNSET:
+            field_dict["sandbox"] = sandbox
         if secret is not UNSET:
             field_dict["secret"] = secret
 
@@ -68,6 +74,8 @@ class IntegrationConnectionSpec:
 
         integration = d.pop("integration", UNSET)
 
+        sandbox = d.pop("sandbox", UNSET)
+
         _secret = d.pop("secret", UNSET)
         secret: Union[Unset, IntegrationConnectionSecret]
         if isinstance(_secret, Unset):
@@ -78,6 +86,7 @@ class IntegrationConnectionSpec:
         integration_connection_spec = cls(
             config=config,
             integration=integration,
+            sandbox=sandbox,
             secret=secret,
         )