From 06d04ebc6355ec5a7897432026c27548e26b5ad0 Mon Sep 17 00:00:00 2001 From: yucheng Date: Sat, 26 Sep 2026 00:00:21 +0000 Subject: [PATCH] test(langfuse_otel): type the _RecordingSpan and _emitted unit helpers Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/unit/integrations/test_langfuse_otel.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tests/unit/integrations/test_langfuse_otel.py b/tests/unit/integrations/test_langfuse_otel.py index b4f5e6d96dd..ff9acc9085e 100644 --- a/tests/unit/integrations/test_langfuse_otel.py +++ b/tests/unit/integrations/test_langfuse_otel.py @@ -1,5 +1,7 @@ import json import os +from collections.abc import Mapping +from typing import Final from unittest.mock import MagicMock, patch import pytest @@ -951,15 +953,15 @@ if __name__ == "__main__": class _RecordingSpan: - def __init__(self): - self.attributes = {} + def __init__(self) -> None: + self.attributes: dict[str, object] = {} - def set_attribute(self, key, value): + def set_attribute(self, key: str, value: object) -> None: self.attributes[key] = value -def _emitted(kwargs, response_obj=None): - span = _RecordingSpan() +def _emitted(kwargs: Mapping[str, object], response_obj: object = None) -> Mapping[str, object]: + span: Final = _RecordingSpan() LangfuseOtelLogger._set_langfuse_specific_attributes(span, kwargs, response_obj) return span.attributes