strip protocol prefix when resolving eval event content (#698)

UKGovernmentBEIS · Oct 14, 2024 · a86be2c · a86be2c
1 parent 95c99f3
commit a86be2c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,7 @@
 
 - Google: compatibility with google-generativeai v0.8.3
 - Open log files in binary mode when reading headers (fixes ijson deprecation warning).
+- Bugfix: strip protocol prefix when resolving eval event content
 
 ## v0.3.41 (11 October 2024)
 

diff --git a/src/inspect_ai/log/_transcript.py b/src/inspect_ai/log/_transcript.py
@@ -391,7 +391,7 @@ def content_fn(text: str) -> str:
 def eval_events_with_content(events: EvalEvents) -> list[Event]:
     def content_fn(text: str) -> str:
         if text.startswith(CONTENT_PROTOCOL):
-            return events.content.get(text, text)
+            return events.content.get(text.replace(CONTENT_PROTOCOL, "", 1), text)
         else:
             return text