Sync cloud skyvern to oss skyvern (#55)
This commit is contained in:
@@ -13,6 +13,7 @@ from starlette_context.plugins.base import Plugin
|
||||
from skyvern.forge.sdk.core import skyvern_context
|
||||
from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
|
||||
from skyvern.forge.sdk.routes.agent_protocol import base_router
|
||||
from skyvern.scheduler import SCHEDULER
|
||||
|
||||
LOG = structlog.get_logger()
|
||||
|
||||
@@ -58,6 +59,12 @@ class Agent:
|
||||
),
|
||||
)
|
||||
|
||||
# Register the scheduler on startup so that we can schedule jobs dynamically
|
||||
@app.on_event("startup")
|
||||
def start_scheduler() -> None:
|
||||
LOG.info("Starting the skyvern scheduler.")
|
||||
SCHEDULER.start()
|
||||
|
||||
@app.exception_handler(Exception)
|
||||
async def unexpected_exception(request: Request, exc: Exception) -> JSONResponse:
|
||||
LOG.exception("Unexpected error in agent server.", exc_info=exc)
|
||||
|
||||
Reference in New Issue
Block a user