Pass meter provider to Pydantic AI (#1136)

Author: alexmojaki

logfire/_internal/formatter.py

@@ -179,7 +179,7 @@ def _fstring_chunks(
         for node_value in arg_node.values:
             if isinstance(node_value, ast.Constant):
                 # These are the parts of the f-string not enclosed by `{}`, e.g. 'foo ' in f'foo {bar}'
-                value: str = node_value.value
+                value: str = node_value.value  # type: ignore
                 result.append({'v': value, 't': 'lit'})
                 new_template += value
             else:

logfire/_internal/integrations/pydantic_ai.py

@@ -18,12 +18,22 @@ def instrument_pydantic_ai(
 ) -> None | InstrumentedModel:
     if event_mode is None:
         event_mode = InstrumentationSettings.event_mode
-    settings = InstrumentationSettings(
+    kwargs = dict(
         tracer_provider=logfire_instance.config.get_tracer_provider(),
         event_logger_provider=logfire_instance.config.get_event_logger_provider(),
         event_mode=event_mode,
         **kwargs,
     )
+    try:
+        settings = InstrumentationSettings(
+            meter_provider=logfire_instance.config.get_meter_provider(),
+            **kwargs,
+        )
+    except TypeError:  # pragma: no cover
+        # Handle older pydantic-ai versions that do not support meter_provider.
+        settings = InstrumentationSettings(
+            **kwargs,
+        )
     if isinstance(obj, Agent):
         obj.instrument = settings
     elif isinstance(obj, Model):

tests/test_configure.py

@@ -575,9 +575,9 @@ def test_logfire_config_console_options() -> None:
 def get_batch_span_exporter(processor: SpanProcessor) -> SpanExporter:
     assert isinstance(processor, BatchSpanProcessor)
     try:
+        exporter = processor.span_exporter
+    except AttributeError:  # pragma: no cover
         exporter = processor._batch_processor._exporter  # type: ignore
-    except AttributeError:
-        exporter = processor.span_exporter  # type: ignore
     return exporter  # type: ignore