fix: try except around new isolate version import (#147)

* fix: try except around new isolate version import

* chore: fix tests with newer pip error

---------

Co-authored-by: Ruslan Kuprieiev <[email protected]>

src/isolate/connections/grpc/agent.py

@@ -1,4 +1,12 @@
 # agent-requires: isolate[server]
+"""
+This file contains the implementation of the gRPC agent. The agent is a
+separate process that is responsible for running the user code in a
+sandboxed environment.
+
+This file is referenced by the latest version of the `isolate` package
+but then runs it in the context of the frozen agent built environment.
+"""
 
 from __future__ import annotations
 
@@ -17,7 +25,11 @@
 import grpc
 from grpc import ServicerContext, StatusCode
 
-from isolate import __version__ as agent_version
+try:
+    from isolate import __version__ as agent_version
+except ImportError:
+    agent_version = "UNKNOWN"
+
 from isolate.backends.common import sha256_digest_of
 from isolate.connections.common import SerializationError, serialize_object
 from isolate.connections.grpc import definitions

tests/test_server.py

@@ -217,7 +217,7 @@ def test_server_builder_error(stub: definitions.IsolateStub, monkeypatch: Any) -
     assert "Failure during 'pip install': Command" in exc.value.details()
 
     raw_logs = [log.message for log in build_logs]
-    assert "ERROR: Invalid requirement: '$$$$'" in raw_logs
+    assert any("ERROR: Invalid requirement: '$$$$'" in raw_log for raw_log in raw_logs)
 
 
 def test_user_logs_immediate(stub: definitions.IsolateStub, monkeypatch: Any) -> None:
@@ -402,7 +402,7 @@ def test_agent_show_logs_from_agent_requirements(
     assert "Failure during 'pip install': Command" in exc.value.details()
 
     raw_logs = [log.message for log in build_logs]
-    assert "ERROR: Invalid requirement: '$$$$'" in raw_logs
+    assert any("ERROR: Invalid requirement: '$$$$'" in raw_log for raw_log in raw_logs)
 
 
 def test_bridge_connection_reuse(