feat: set ISOLATE_SERVER_VERSION for agent (#143)

src/isolate/connections/_local/_base.py

@@ -15,6 +15,7 @@
     TypeVar,
 )
 
+from isolate import __version__ as isolate_version
 from isolate.backends.common import get_executable_path, logged_io
 from isolate.connections.common import AGENT_SIGNATURE
 from isolate.logs import LogLevel, LogSource
@@ -138,6 +139,7 @@ def get_env_vars(self) -> dict[str, str]:
         immediately (so that we can seamlessly transfer logs)."""
 
         custom_vars = {}
+        custom_vars["ISOLATE_SERVER_VERSION"] = isolate_version
         custom_vars[AGENT_SIGNATURE] = "1"
         custom_vars["PYTHONUNBUFFERED"] = "1"
 

src/isolate/connections/grpc/agent.py

@@ -17,6 +17,7 @@
 import grpc
 from grpc import ServicerContext, StatusCode
 
+from isolate import __version__ as agent_version
 from isolate.backends.common import sha256_digest_of
 from isolate.connections.common import SerializationError, serialize_object
 from isolate.connections.grpc import definitions
@@ -44,6 +45,8 @@ def Run(
         context: ServicerContext,
     ) -> Iterator[definitions.PartialRunResult]:
         self.log(f"A connection has been established: {context.peer()}!")
+        server_version = os.getenv("ISOLATE_SERVER_VERSION") or "unknown"
+        self.log(f"Isolate info: server {server_version}, agent {agent_version}")
 
         extra_args = []
         if request.HasField("setup_func"):