diff --git a/skyvern/_version.py b/skyvern/_version.py new file mode 100644 index 00000000..4ab6b1ef --- /dev/null +++ b/skyvern/_version.py @@ -0,0 +1 @@ +__version__ = "development" diff --git a/skyvern/forge/sdk/forge_log.py b/skyvern/forge/sdk/forge_log.py index 375ef739..02efcbf0 100644 --- a/skyvern/forge/sdk/forge_log.py +++ b/skyvern/forge/sdk/forge_log.py @@ -3,6 +3,7 @@ import logging import structlog from structlog.typing import EventDict +from skyvern._version import __version__ from skyvern.config import settings from skyvern.forge.sdk.core import skyvern_context @@ -39,6 +40,7 @@ def add_kv_pairs_to_msg(logger: logging.Logger, method_name: str, event_dict: Ev # Add env to the log event_dict["env"] = settings.ENV + event_dict["version"] = __version__ if method_name not in ["info", "warning", "error", "critical", "exception"]: # Only modify the log for these log levels diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 10e7046c..917ef20c 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -10,6 +10,7 @@ from fastapi import BackgroundTasks, Depends, Header, HTTPException, Path, Query from fastapi.responses import ORJSONResponse from skyvern import analytics +from skyvern._version import __version__ from skyvern.config import settings from skyvern.forge import app from skyvern.forge.prompts import prompt_engine @@ -155,7 +156,7 @@ async def heartbeat() -> Response: """ Check if the server is running. """ - return Response(content="Server is running.", status_code=200) + return Response(content="Server is running.", status_code=200, headers={"X-Skyvern-API-Version": __version__}) @legacy_base_router.post(