Set error payload for type inference errors with single spans

PiperOrigin-RevId: 820232578

Author: NL02

xls/dslx/errors.cc

@@ -45,9 +45,13 @@ absl::Status TypeInferenceErrorStatusInternal(
   if (type_or_annotation != nullptr) {
     type_str = type_or_annotation->ToString() + " ";
   }
-  return absl::InvalidArgumentError(
-      absl::StrFormat("TypeInferenceError: %s %s%s", span.ToString(file_table),
-                      type_str, message));
+
+  absl::Status status = absl::InvalidArgumentError(
+      absl::StrFormat("TypeInferenceError: %s%s", type_str, message));
+  StatusPayloadProto payload;
+  *payload.add_spans() = ToProto(span, file_table);
+  SetStatusPayload(status, payload);
+  return status;
 }
 
 }  // namespace

xls/dslx/errors_test.cc

@@ -42,10 +42,12 @@ TEST(ErrorsTest, TypeInferenceErrorMessage) {
   std::unique_ptr<Type> type = BitsType::MakeU32();
   absl::Status status = TypeInferenceErrorStatus(
       span, type.get(), "this is the message!", file_table);
-  EXPECT_EQ(
-      status.ToString(),
-      "INVALID_ARGUMENT: TypeInferenceError: <no-file>:1:1-2:2 uN[32] this "
-      "is the message!");
+  std::optional<StatusPayloadProto> payload = GetStatusPayload(status);
+  ASSERT_TRUE(payload.has_value());
+  EXPECT_THAT(payload->spans(),
+              UnorderedElementsAre(EqualsProto(ToProto(span, file_table))));
+  EXPECT_EQ(status.message(),
+            "TypeInferenceError: uN[32] this is the message!");
 }
 
 TEST(ErrorsTest, SignednessMismatchErrorAnnotationPayload) {