From b5fb11af51d80505343a1490ed78490ac4d09c2f Mon Sep 17 00:00:00 2001
From: Michael Ripa <m.ripa123@gmail.com>
Date: Wed, 26 Feb 2025 13:30:26 -0500
Subject: [PATCH] Add client nnsight version to request

---
 src/nnsight/intervention/backends/remote.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/nnsight/intervention/backends/remote.py b/src/nnsight/intervention/backends/remote.py
index e126322..e15e1a1 100755
--- a/src/nnsight/intervention/backends/remote.py
+++ b/src/nnsight/intervention/backends/remote.py
@@ -12,7 +12,7 @@
 import torch
 from tqdm.auto import tqdm
 
-from ... import __IPYTHON__, CONFIG, remote_logger
+from ... import __IPYTHON__, __version__, CONFIG, remote_logger
 from ...schema.request import RequestModel, StreamValueModel
 from ...schema.response import ResponseModel
 from ...schema.result import RESULT, ResultModel
@@ -54,6 +54,7 @@ def __init__(
         self.host = host or CONFIG.API.HOST
         self.address = f"http{'s' if self.ssl else ''}://{self.host}"
         self.ws_address = f"ws{'s' if CONFIG.API.SSL else ''}://{self.host}"
+        self.version = __version__
 
     def request(self, graph: Graph) -> Tuple[bytes, Dict[str, str]]:
 
@@ -65,6 +66,7 @@ def request(self, graph: Graph) -> Tuple[bytes, Dict[str, str]]:
             "zlib": str(self.zlib),
             "ndif-api-key": self.api_key,
             "sent-timestamp": str(time.time()),
+            "version": self.version,
         }
 
         return data, headers
@@ -197,7 +199,10 @@ def submit_request(
             return response
 
         else:
-            msg = response.reason
+            try:
+                msg = response.json()["detail"]
+            except:
+                msg = response.reason
             raise ConnectionError(msg)
 
     def get_response(self) -> Optional[RESULT]: