From 4805a5d85a3be4944b39b66620619a49e212e47a Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Thu, 24 Apr 2025 15:22:42 +0800
Subject: [PATCH] add current sha into log (#2227)

---
 skyvern/_version.py                        | 1 +
 skyvern/forge/sdk/forge_log.py             | 2 ++
 skyvern/forge/sdk/routes/agent_protocol.py | 3 ++-
 3 files changed, 5 insertions(+), 1 deletion(-)
 create mode 100644 skyvern/_version.py

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(