Support OpenTelemetry SDK 1.44.0 (#2094)

Author: alexmojaki

logfire/_internal/exporters/console.py

@@ -506,5 +506,8 @@ def export(self, batch: Sequence[ReadableLogRecord]) -> LogRecordExportResult:
 
         return LogRecordExportResult.SUCCESS
 
+    def force_flush(self, timeout_millis: int = 10_000) -> bool:
+        return self.span_exporter.force_flush(timeout_millis)
+
     def shutdown(self):
         self.span_exporter.shutdown()

logfire/_internal/exporters/wrapper.py

@@ -83,6 +83,9 @@ class WrapperLogExporter(LogRecordExporter):
     def export(self, batch: Sequence[ReadableLogRecord]) -> LogRecordExportResult:
         return self.exporter.export(batch)
 
+    def force_flush(self, timeout_millis: int = 10_000) -> bool:
+        return self.exporter.force_flush(timeout_millis)
+
     def shutdown(self):
         return self.exporter.shutdown()
 

pyproject.toml

@@ -46,8 +46,8 @@ classifiers = [
     "Framework :: OpenTelemetry :: Instrumentations",
 ]
 dependencies = [
-    "opentelemetry-sdk >= 1.39.0, < 1.44.0",
-    "opentelemetry-exporter-otlp-proto-http >= 1.39.0, < 1.44.0",
+    "opentelemetry-sdk >= 1.39.0, < 1.45.0",
+    "opentelemetry-exporter-otlp-proto-http >= 1.39.0, < 1.45.0",
     "opentelemetry-instrumentation >= 0.41b0",
     "rich >= 13.4.2",
     "protobuf >= 4.23.4",
@@ -225,7 +225,7 @@ logfire-api = { workspace = true }
 
 [tool.uv]
 default-groups = ["dev"]
-exclude-newer = "1 week"
+exclude-newer = "4 days"
 
 [tool.uv.exclude-newer-package]
 # These are our own packages, not worth defending from supply chain attacks on them.

tests/test_console_exporter.py

@@ -24,6 +24,7 @@
 from logfire import ConsoleOptions
 from logfire._internal.constants import ATTRIBUTES_MESSAGE_KEY
 from logfire._internal.exporters.console import (
+    ConsoleLogExporter,
     IndentedConsoleSpanExporter,
     ShowParentsConsoleSpanExporter,
     SimpleConsoleSpanExporter,
@@ -944,6 +945,11 @@ def test_console_otel_logs(capsys: pytest.CaptureFixture[str]):
     )
 
 
+def test_console_log_exporter_force_flush():
+    exporter = ConsoleLogExporter(SimpleConsoleSpanExporter())
+    assert exporter.force_flush() is True
+
+
 def test_truncated_json(capsys: pytest.CaptureFixture[str]) -> None:
     with mock.patch.dict('os.environ', {'OTEL_ATTRIBUTE_VALUE_LENGTH_LIMIT': '70'}):
         logfire.configure(

tests/test_otel_logs.py

@@ -152,12 +152,17 @@ def shutdown(self):
         def export(self, batch: Sequence[ReadableLogRecord]):
             raise requests.exceptions.ConnectionError()
 
+        def force_flush(self, timeout_millis: int = 10_000) -> bool:
+            return True
+
     connection_error_exporter = ConnectionErrorExporter()
     exporter = QuietLogExporter(connection_error_exporter)
 
     assert exporter.export([]) == LogRecordExportResult.FAILURE
     assert not caplog.messages
 
+    assert exporter.force_flush() is True
+
     assert not connection_error_exporter.shutdown_called
     exporter.shutdown()
     assert connection_error_exporter.shutdown_called