From 76466a4d675cd2f95832098fbd8d12da28d77978 Mon Sep 17 00:00:00 2001
From: Chu-Cheng Lin <kitsing@google.com>
Date: Mon, 15 Jul 2024 22:12:04 -0700
Subject: [PATCH] Internal changes.

PiperOrigin-RevId: 652699171
---
 langfun/core/structured/scoring.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/langfun/core/structured/scoring.py b/langfun/core/structured/scoring.py
index 398bab54..5aabc80a 100644
--- a/langfun/core/structured/scoring.py
+++ b/langfun/core/structured/scoring.py
@@ -76,14 +76,19 @@ def score(
       raise ValueError('`lm` must be specified or provided from `lf.context`.')
     lm = lm_override.value
 
+  completion_reprs = []
+  for c in completions:
+    if isinstance(c, mapping.MappingError):
+      rep = c.lm_response
+    else:
+      rep = mapping.MappingExample.value_repr(
+          c, protocol=protocol, compact=False, verbose=False
+      )
+    completion_reprs.append(rep)
+
   results = lm.score(
       input_message,
-      [
-          mapping.MappingExample.value_repr(
-              c, protocol=protocol, compact=False, verbose=False
-          )
-          for c in completions
-      ],
+      completion_reprs,
   )
   if return_scoring_results:
     return results