letta-nightly 0.11.7.dev20251007104119__py3-none-any.whl → 0.11.7.dev20251008104128__py3-none-any.whl

letta/services/run_manager.py

@@ -0,0 +1,301 @@
+from datetime import datetime
+from pickletools import pyunicode
+from typing import List, Literal, Optional
+
+from httpx import AsyncClient
+from sqlalchemy import select
+from sqlalchemy.orm import Session
+
+from letta.helpers.datetime_helpers import get_utc_time
+from letta.log import get_logger
+from letta.orm.errors import NoResultFound
+from letta.orm.message import Message as MessageModel
+from letta.orm.run import Run as RunModel
+from letta.orm.sqlalchemy_base import AccessType
+from letta.orm.step import Step as StepModel
+from letta.otel.tracing import log_event, trace_method
+from letta.schemas.enums import AgentType, MessageRole, RunStatus
+from letta.schemas.job import LettaRequestConfig
+from letta.schemas.letta_message import LettaMessage, LettaMessageUnion
+from letta.schemas.letta_response import LettaResponse
+from letta.schemas.letta_stop_reason import LettaStopReason, StopReasonType
+from letta.schemas.message import Message as PydanticMessage
+from letta.schemas.run import Run as PydanticRun, RunUpdate
+from letta.schemas.step import Step as PydanticStep
+from letta.schemas.usage import LettaUsageStatistics
+from letta.schemas.user import User as PydanticUser
+from letta.server.db import db_registry
+from letta.services.agent_manager import AgentManager
+from letta.services.helpers.agent_manager_helper import validate_agent_exists_async
+from letta.services.message_manager import MessageManager
+from letta.services.step_manager import StepManager
+from letta.utils import enforce_types
+
+logger = get_logger(__name__)
+
+
+class RunManager:
+    """Manager class to handle business logic related to Runs."""
+
+    def __init__(self):
+        """Initialize the RunManager."""
+        self.step_manager = StepManager()
+        self.message_manager = MessageManager()
+        self.agent_manager = AgentManager()
+
+    @enforce_types
+    async def create_run(self, pydantic_run: PydanticRun, actor: PydanticUser) -> PydanticRun:
+        """Create a new run."""
+        async with db_registry.async_session() as session:
+            # Get agent_id from the pydantic object
+            agent_id = pydantic_run.agent_id
+
+            # Verify agent exists before creating the run
+            await validate_agent_exists_async(session, agent_id, actor)
+            organization_id = actor.organization_id
+
+            run_data = pydantic_run.model_dump(exclude_none=True)
+            # Handle metadata field mapping (Pydantic uses 'metadata', ORM uses 'metadata_')
+            if "metadata" in run_data:
+                run_data["metadata_"] = run_data.pop("metadata")
+
+            run = RunModel(**run_data)
+            run.organization_id = organization_id
+            run = await run.create_async(session, actor=actor, no_commit=True, no_refresh=True)
+            await session.commit()
+
+        return run.to_pydantic()
+
+    @enforce_types
+    async def get_run_by_id(self, run_id: str, actor: PydanticUser) -> PydanticRun:
+        """Get a run by its ID."""
+        async with db_registry.async_session() as session:
+            run = await RunModel.read_async(db_session=session, identifier=run_id, actor=actor, access_type=AccessType.ORGANIZATION)
+            if not run:
+                raise NoResultFound(f"Run with id {run_id} not found")
+            return run.to_pydantic()
+
+    @enforce_types
+    async def list_runs(
+        self,
+        actor: PydanticUser,
+        agent_id: Optional[str] = None,
+        agent_ids: Optional[List[str]] = None,
+        statuses: Optional[List[RunStatus]] = None,
+        limit: Optional[int] = 50,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        ascending: bool = False,
+        stop_reason: Optional[str] = None,
+        background: Optional[bool] = None,
+    ) -> List[PydanticRun]:
+        """List runs with filtering options."""
+        async with db_registry.async_session() as session:
+            from sqlalchemy import select
+
+            query = select(RunModel).filter(RunModel.organization_id == actor.organization_id)
+
+            # Handle agent filtering
+            if agent_id:
+                agent_ids = [agent_id]
+            if agent_ids:
+                query = query.filter(RunModel.agent_id.in_(agent_ids))
+
+            # Filter by status
+            if statuses:
+                query = query.filter(RunModel.status.in_(statuses))
+
+            # Filter by stop reason
+            if stop_reason:
+                query = query.filter(RunModel.stop_reason == stop_reason)
+
+            # Filter by background
+            if background is not None:
+                query = query.filter(RunModel.background == background)
+
+            # Apply pagination
+            from letta.services.helpers.run_manager_helper import _apply_pagination_async
+
+            query = await _apply_pagination_async(query, before, after, session, ascending=ascending)
+
+            # Apply limit
+            if limit:
+                query = query.limit(limit)
+
+            result = await session.execute(query)
+            runs = result.scalars().all()
+            return [run.to_pydantic() for run in runs]
+
+    @enforce_types
+    async def delete_run(self, run_id: str, actor: PydanticUser) -> PydanticRun:
+        """Delete a run by its ID."""
+        async with db_registry.async_session() as session:
+            run = await RunModel.read_async(db_session=session, identifier=run_id, actor=actor, access_type=AccessType.ORGANIZATION)
+            if not run:
+                raise NoResultFound(f"Run with id {run_id} not found")
+
+            pydantic_run = run.to_pydantic()
+            await run.hard_delete_async(db_session=session, actor=actor)
+
+        return pydantic_run
+
+    @enforce_types
+    async def update_run_by_id_async(
+        self, run_id: str, update: RunUpdate, actor: PydanticUser, refresh_result_messages: bool = True
+    ) -> PydanticRun:
+        """Update a run using a RunUpdate object."""
+
+        async with db_registry.async_session() as session:
+            run = await RunModel.read_async(db_session=session, identifier=run_id, actor=actor)
+
+            # Check if this is a terminal update and whether we should dispatch a callback
+            needs_callback = False
+            callback_url = None
+            not_completed_before = not bool(run.completed_at)
+            is_terminal_update = update.status in {RunStatus.completed, RunStatus.failed}
+            if is_terminal_update and not_completed_before and run.callback_url:
+                needs_callback = True
+                callback_url = run.callback_url
+
+            # Housekeeping only when the run is actually completing
+            if not_completed_before and is_terminal_update:
+                if not update.stop_reason:
+                    logger.warning(f"Run {run_id} completed without a stop reason")
+                if not update.completed_at:
+                    logger.warning(f"Run {run_id} completed without a completed_at timestamp")
+                    update.completed_at = get_utc_time().replace(tzinfo=None)
+
+            # Update job attributes with only the fields that were explicitly set
+            update_data = update.model_dump(to_orm=True, exclude_unset=True, exclude_none=True)
+
+            # Automatically update the completion timestamp if status is set to 'completed'
+            for key, value in update_data.items():
+                # Ensure completed_at is timezone-naive for database compatibility
+                if key == "completed_at" and value is not None and hasattr(value, "replace"):
+                    value = value.replace(tzinfo=None)
+                setattr(run, key, value)
+
+            await run.update_async(db_session=session, actor=actor, no_commit=True, no_refresh=True)
+            final_metadata = run.metadata_
+            pydantic_run = run.to_pydantic()
+            await session.commit()
+
+        # Dispatch callback outside of database session if needed
+        if needs_callback:
+            if refresh_result_messages:
+                result = LettaResponse(
+                    messages=await self.get_run_messages(run_id=run_id, actor=actor),
+                    stop_reason=LettaStopReason(stop_reason=pydantic_run.stop_reason),
+                    usage=await self.get_run_usage(run_id=run_id, actor=actor),
+                )
+                final_metadata["result"] = result.model_dump()
+            callback_info = {
+                "run_id": run_id,
+                "callback_url": callback_url,
+                "status": update.status,
+                "completed_at": get_utc_time().replace(tzinfo=None),
+                "metadata": final_metadata,
+            }
+            callback_result = await self._dispatch_callback_async(callback_info)
+
+            # Update callback status in a separate transaction
+            async with db_registry.async_session() as session:
+                run = await RunModel.read_async(db_session=session, identifier=run_id, actor=actor)
+                run.callback_sent_at = callback_result["callback_sent_at"]
+                run.callback_status_code = callback_result.get("callback_status_code")
+                run.callback_error = callback_result.get("callback_error")
+                pydantic_run = run.to_pydantic()
+                await run.update_async(db_session=session, actor=actor, no_commit=True, no_refresh=True)
+                await session.commit()
+
+        return pydantic_run
+
+    @trace_method
+    async def _dispatch_callback_async(self, callback_info: dict) -> dict:
+        """
+        POST a standard JSON payload to callback_url and return callback status asynchronously.
+        """
+        payload = {
+            "run_id": callback_info["run_id"],
+            "status": callback_info["status"],
+            "completed_at": callback_info["completed_at"].isoformat() if callback_info["completed_at"] else None,
+            "metadata": callback_info["metadata"],
+        }
+
+        callback_sent_at = get_utc_time().replace(tzinfo=None)
+        result = {"callback_sent_at": callback_sent_at}
+
+        try:
+            async with AsyncClient() as client:
+                log_event("POST callback dispatched", payload)
+                resp = await client.post(callback_info["callback_url"], json=payload, timeout=5.0)
+                log_event("POST callback finished")
+                result["callback_status_code"] = resp.status_code
+        except Exception as e:
+            error_message = f"Failed to dispatch callback for run {callback_info['run_id']} to {callback_info['callback_url']}: {e!s}"
+            logger.error(error_message)
+            result["callback_error"] = error_message
+            # Continue silently - callback failures should not affect run completion
+        finally:
+            return result
+
+    @enforce_types
+    async def get_run_usage(self, run_id: str, actor: PydanticUser) -> LettaUsageStatistics:
+        """Get usage statistics for a run."""
+        async with db_registry.async_session() as session:
+            run = await RunModel.read_async(db_session=session, identifier=run_id, actor=actor, access_type=AccessType.ORGANIZATION)
+            if not run:
+                raise NoResultFound(f"Run with id {run_id} not found")
+
+        steps = await self.step_manager.list_steps_async(run_id=run_id, actor=actor)
+        total_usage = LettaUsageStatistics()
+        for step in steps:
+            total_usage.prompt_tokens += step.prompt_tokens
+            total_usage.completion_tokens += step.completion_tokens
+            total_usage.total_tokens += step.total_tokens
+            total_usage.step_count += 1
+        return total_usage
+
+    @enforce_types
+    async def get_run_messages(
+        self,
+        run_id: str,
+        actor: PydanticUser,
+        limit: Optional[int] = 100,
+        before: Optional[str] = None,
+        after: Optional[str] = None,
+        order: Literal["asc", "desc"] = "asc",
+    ) -> List[LettaMessage]:
+        """Get the result of a run."""
+        run = await self.get_run_by_id(run_id=run_id, actor=actor)
+        request_config = run.request_config
+        agent = await self.agent_manager.get_agent_by_id_async(agent_id=run.agent_id, actor=actor, include_relationships=[])
+        text_is_assistant_message = agent.agent_type == AgentType.letta_v1_agent
+
+        messages = await self.message_manager.list_messages(
+            actor=actor,
+            run_id=run_id,
+            limit=limit,
+            before=before,
+            after=after,
+            ascending=(order == "asc"),
+        )
+        letta_messages = PydanticMessage.to_letta_messages_from_list(
+            messages, reverse=(order != "asc"), text_is_assistant_message=text_is_assistant_message
+        )
+
+        if request_config and request_config.include_return_message_types:
+            include_return_message_types_set = set(request_config.include_return_message_types)
+            letta_messages = [msg for msg in letta_messages if msg.message_type in include_return_message_types_set]
+
+        return letta_messages
+
+    @enforce_types
+    async def get_run_request_config(self, run_id: str, actor: PydanticUser) -> Optional[LettaRequestConfig]:
+        """Get the letta request config from a run."""
+        async with db_registry.async_session() as session:
+            run = await RunModel.read_async(db_session=session, identifier=run_id, actor=actor, access_type=AccessType.ORGANIZATION)
+            if not run:
+                raise NoResultFound(f"Run with id {run_id} not found")
+            pydantic_run = run.to_pydantic()
+            return pydantic_run.request_config

letta/services/sandbox_config_manager.py

@@ -45,40 +45,6 @@ class SandboxConfigManager:
             sandbox_config = self.create_or_update_sandbox_config(SandboxConfigCreate(config=default_config), actor=actor)
         return sandbox_config
 
-    @enforce_types
-    @trace_method
-    def create_or_update_sandbox_config(self, sandbox_config_create: SandboxConfigCreate, actor: PydanticUser) -> PydanticSandboxConfig:
-        """Create or update a sandbox configuration based on the PydanticSandboxConfig schema."""
-        config = sandbox_config_create.config
-        sandbox_type = config.type
-        sandbox_config = PydanticSandboxConfig(
-            type=sandbox_type, config=config.model_dump(exclude_none=True), organization_id=actor.organization_id
-        )
-
-        # Attempt to retrieve the existing sandbox configuration by type within the organization
-        db_sandbox = self.get_sandbox_config_by_type(sandbox_config.type, actor=actor)
-        if db_sandbox:
-            # Prepare the update data, excluding fields that should not be reset
-            update_data = sandbox_config.model_dump(exclude_unset=True, exclude_none=True)
-            update_data = {key: value for key, value in update_data.items() if getattr(db_sandbox, key) != value}
-
-            # If there are changes, update the sandbox configuration
-            if update_data:
-                db_sandbox = self.update_sandbox_config(db_sandbox.id, SandboxConfigUpdate(**update_data), actor)
-            else:
-                printd(
-                    f"`create_or_update_sandbox_config` was called with user_id={actor.id}, organization_id={actor.organization_id}, "
-                    f"type={sandbox_config.type}, but found existing configuration with nothing to update."
-                )
-
-            return db_sandbox
-        else:
-            # If the sandbox configuration doesn't exist, create a new one
-            with db_registry.session() as session:
-                db_sandbox = SandboxConfigModel(**sandbox_config.model_dump(exclude_none=True))
-                db_sandbox.create(session, actor=actor)
-                return db_sandbox.to_pydantic()
-
     @enforce_types
     @trace_method
     async def get_or_create_default_sandbox_config_async(self, sandbox_type: SandboxType, actor: PydanticUser) -> PydanticSandboxConfig:
@@ -133,34 +99,6 @@ class SandboxConfigManager:
                 await db_sandbox.create_async(session, actor=actor)
                 return db_sandbox.to_pydantic()
 
-    @enforce_types
-    @trace_method
-    def update_sandbox_config(
-        self, sandbox_config_id: str, sandbox_update: SandboxConfigUpdate, actor: PydanticUser
-    ) -> PydanticSandboxConfig:
-        """Update an existing sandbox configuration."""
-        with db_registry.session() as session:
-            sandbox = SandboxConfigModel.read(db_session=session, identifier=sandbox_config_id, actor=actor)
-            # We need to check that the sandbox_update provided is the same type as the original sandbox
-            if sandbox.type != sandbox_update.config.type:
-                raise ValueError(
-                    f"Mismatched type for sandbox config update: tried to update sandbox_config of type {sandbox.type} with config of type {sandbox_update.config.type}"
-                )
-
-            update_data = sandbox_update.model_dump(exclude_unset=True, exclude_none=True)
-            update_data = {key: value for key, value in update_data.items() if getattr(sandbox, key) != value}
-
-            if update_data:
-                for key, value in update_data.items():
-                    setattr(sandbox, key, value)
-                sandbox.update(db_session=session, actor=actor)
-            else:
-                printd(
-                    f"`update_sandbox_config` called with user_id={actor.id}, organization_id={actor.organization_id}, "
-                    f"name={sandbox.type}, but nothing to update."
-                )
-            return sandbox.to_pydantic()
-
     @enforce_types
     @trace_method
     async def update_sandbox_config_async(
@@ -189,15 +127,6 @@ class SandboxConfigManager:
                 )
             return sandbox.to_pydantic()
 
-    @enforce_types
-    @trace_method
-    def delete_sandbox_config(self, sandbox_config_id: str, actor: PydanticUser) -> PydanticSandboxConfig:
-        """Delete a sandbox configuration by its ID."""
-        with db_registry.session() as session:
-            sandbox = SandboxConfigModel.read(db_session=session, identifier=sandbox_config_id, actor=actor)
-            sandbox.hard_delete(db_session=session, actor=actor)
-            return sandbox.to_pydantic()
-
     @enforce_types
     @trace_method
     async def delete_sandbox_config_async(self, sandbox_config_id: str, actor: PydanticUser) -> PydanticSandboxConfig:
@@ -207,24 +136,6 @@ class SandboxConfigManager:
             await sandbox.hard_delete_async(db_session=session, actor=actor)
             return sandbox.to_pydantic()
 
-    @enforce_types
-    @trace_method
-    def list_sandbox_configs(
-        self,
-        actor: PydanticUser,
-        after: Optional[str] = None,
-        limit: Optional[int] = 50,
-        sandbox_type: Optional[SandboxType] = None,
-    ) -> List[PydanticSandboxConfig]:
-        """List all sandbox configurations with optional pagination."""
-        kwargs = {"organization_id": actor.organization_id}
-        if sandbox_type:
-            kwargs.update({"type": sandbox_type})
-
-        with db_registry.session() as session:
-            sandboxes = SandboxConfigModel.list(db_session=session, after=after, limit=limit, **kwargs)
-            return [sandbox.to_pydantic() for sandbox in sandboxes]
-
     @enforce_types
     @trace_method
     async def list_sandbox_configs_async(
@@ -243,35 +154,6 @@ class SandboxConfigManager:
             sandboxes = await SandboxConfigModel.list_async(db_session=session, after=after, limit=limit, **kwargs)
             return [sandbox.to_pydantic() for sandbox in sandboxes]
 
-    @enforce_types
-    @trace_method
-    def get_sandbox_config_by_id(self, sandbox_config_id: str, actor: Optional[PydanticUser] = None) -> Optional[PydanticSandboxConfig]:
-        """Retrieve a sandbox configuration by its ID."""
-        with db_registry.session() as session:
-            try:
-                sandbox = SandboxConfigModel.read(db_session=session, identifier=sandbox_config_id, actor=actor)
-                return sandbox.to_pydantic()
-            except NoResultFound:
-                return None
-
-    @enforce_types
-    @trace_method
-    def get_sandbox_config_by_type(self, type: SandboxType, actor: Optional[PydanticUser] = None) -> Optional[PydanticSandboxConfig]:
-        """Retrieve a sandbox config by its type."""
-        with db_registry.session() as session:
-            try:
-                sandboxes = SandboxConfigModel.list(
-                    db_session=session,
-                    type=type,
-                    organization_id=actor.organization_id,
-                    limit=1,
-                )
-                if sandboxes:
-                    return sandboxes[0].to_pydantic()
-                return None
-            except NoResultFound:
-                return None
-
     @enforce_types
     @trace_method
     async def get_sandbox_config_by_type_async(
@@ -292,34 +174,6 @@ class SandboxConfigManager:
             except NoResultFound:
                 return None
 
-    @enforce_types
-    @trace_method
-    def create_sandbox_env_var(
-        self, env_var_create: SandboxEnvironmentVariableCreate, sandbox_config_id: str, actor: PydanticUser
-    ) -> PydanticEnvVar:
-        """Create a new sandbox environment variable."""
-        env_var = PydanticEnvVar(**env_var_create.model_dump(), sandbox_config_id=sandbox_config_id, organization_id=actor.organization_id)
-
-        db_env_var = self.get_sandbox_env_var_by_key_and_sandbox_config_id(env_var.key, env_var.sandbox_config_id, actor=actor)
-        if db_env_var:
-            update_data = env_var.model_dump(exclude_unset=True, exclude_none=True)
-            update_data = {key: value for key, value in update_data.items() if getattr(db_env_var, key) != value}
-            # If there are changes, update the environment variable
-            if update_data:
-                db_env_var = self.update_sandbox_env_var(db_env_var.id, SandboxEnvironmentVariableUpdate(**update_data), actor)
-            else:
-                printd(
-                    f"`create_or_update_sandbox_env_var` was called with user_id={actor.id}, organization_id={actor.organization_id}, "
-                    f"key={env_var.key}, but found existing variable with nothing to update."
-                )
-
-            return db_env_var
-        else:
-            with db_registry.session() as session:
-                env_var = SandboxEnvVarModel(**env_var.model_dump(to_orm=True, exclude_none=True))
-                env_var.create(session, actor=actor)
-            return env_var.to_pydantic()
-
     @enforce_types
     @trace_method
     async def create_sandbox_env_var_async(
@@ -348,28 +202,6 @@ class SandboxConfigManager:
                 await env_var.create_async(session, actor=actor)
                 return env_var.to_pydantic()
 
-    @enforce_types
-    @trace_method
-    def update_sandbox_env_var(
-        self, env_var_id: str, env_var_update: SandboxEnvironmentVariableUpdate, actor: PydanticUser
-    ) -> PydanticEnvVar:
-        """Update an existing sandbox environment variable."""
-        with db_registry.session() as session:
-            env_var = SandboxEnvVarModel.read(db_session=session, identifier=env_var_id, actor=actor)
-            update_data = env_var_update.model_dump(to_orm=True, exclude_unset=True, exclude_none=True)
-            update_data = {key: value for key, value in update_data.items() if getattr(env_var, key) != value}
-
-            if update_data:
-                for key, value in update_data.items():
-                    setattr(env_var, key, value)
-                env_var.update(db_session=session, actor=actor)
-            else:
-                printd(
-                    f"`update_sandbox_env_var` called with user_id={actor.id}, organization_id={actor.organization_id}, "
-                    f"key={env_var.key}, but nothing to update."
-                )
-            return env_var.to_pydantic()
-
     @enforce_types
     @trace_method
     async def update_sandbox_env_var_async(
@@ -392,15 +224,6 @@ class SandboxConfigManager:
                 )
             return env_var.to_pydantic()
 
-    @enforce_types
-    @trace_method
-    def delete_sandbox_env_var(self, env_var_id: str, actor: PydanticUser) -> PydanticEnvVar:
-        """Delete a sandbox environment variable by its ID."""
-        with db_registry.session() as session:
-            env_var = SandboxEnvVarModel.read(db_session=session, identifier=env_var_id, actor=actor)
-            env_var.hard_delete(db_session=session, actor=actor)
-            return env_var.to_pydantic()
-
     @enforce_types
     @trace_method
     async def delete_sandbox_env_var_async(self, env_var_id: str, actor: PydanticUser) -> PydanticEnvVar:
@@ -410,26 +233,6 @@ class SandboxConfigManager:
             await env_var.hard_delete_async(db_session=session, actor=actor)
             return env_var.to_pydantic()
 
-    @enforce_types
-    @trace_method
-    def list_sandbox_env_vars(
-        self,
-        sandbox_config_id: str,
-        actor: PydanticUser,
-        after: Optional[str] = None,
-        limit: Optional[int] = 50,
-    ) -> List[PydanticEnvVar]:
-        """List all sandbox environment variables with optional pagination."""
-        with db_registry.session() as session:
-            env_vars = SandboxEnvVarModel.list(
-                db_session=session,
-                after=after,
-                limit=limit,
-                organization_id=actor.organization_id,
-                sandbox_config_id=sandbox_config_id,
-            )
-            return [env_var.to_pydantic() for env_var in env_vars]
-
     @enforce_types
     @trace_method
     async def list_sandbox_env_vars_async(
@@ -450,22 +253,6 @@ class SandboxConfigManager:
             )
             return [env_var.to_pydantic() for env_var in env_vars]
 
-    @enforce_types
-    @trace_method
-    def list_sandbox_env_vars_by_key(
-        self, key: str, actor: PydanticUser, after: Optional[str] = None, limit: Optional[int] = 50
-    ) -> List[PydanticEnvVar]:
-        """List all sandbox environment variables with optional pagination."""
-        with db_registry.session() as session:
-            env_vars = SandboxEnvVarModel.list(
-                db_session=session,
-                after=after,
-                limit=limit,
-                organization_id=actor.organization_id,
-                key=key,
-            )
-            return [env_var.to_pydantic() for env_var in env_vars]
-
     @enforce_types
     @trace_method
     async def list_sandbox_env_vars_by_key_async(
@@ -501,27 +288,6 @@ class SandboxConfigManager:
         env_vars = await self.list_sandbox_env_vars_async(sandbox_config_id, actor, after, limit)
         return {env_var.key: env_var.value for env_var in env_vars}
 
-    @enforce_types
-    @trace_method
-    def get_sandbox_env_var_by_key_and_sandbox_config_id(
-        self, key: str, sandbox_config_id: str, actor: Optional[PydanticUser] = None
-    ) -> Optional[PydanticEnvVar]:
-        """Retrieve a sandbox environment variable by its key and sandbox_config_id."""
-        with db_registry.session() as session:
-            try:
-                env_var = SandboxEnvVarModel.list(
-                    db_session=session,
-                    key=key,
-                    sandbox_config_id=sandbox_config_id,
-                    organization_id=actor.organization_id,
-                    limit=1,
-                )
-                if env_var:
-                    return env_var[0].to_pydantic()
-                return None
-            except NoResultFound:
-                return None
-
     @enforce_types
     @trace_method
     async def get_sandbox_env_var_by_key_and_sandbox_config_id_async(