fix: set provider_id and model_id on FauxProvider streamed messages

cubepi/providers/faux.py

@@ -267,7 +267,13 @@ async def _produce() -> None:
                 cache_usage = self._compute_cache_usage(
                     system_prompt, tools, messages, base_usage
                 )
-                resolved = resolved.model_copy(update={"usage": cache_usage})
+                resolved = resolved.model_copy(
+                    update={
+                        "usage": cache_usage,
+                        "provider_id": model.provider,
+                        "model_id": model.id,
+                    }
+                )
 
                 await self._stream_with_deltas(ms, resolved, opts.signal)
             except BaseException as exc:
@@ -297,6 +303,8 @@ async def _stream_with_deltas(
             stop_reason=message.stop_reason,
             usage=message.usage,
             timestamp=message.timestamp,
+            provider_id=message.provider_id,
+            model_id=message.model_id,
         )
 
         if signal and signal.is_set():

tests/providers/test_faux.py

@@ -1,6 +1,7 @@
 import asyncio
 
 from cubepi.providers.base import (
+    Model,
     StreamOptions,
     TextContent,
     ToolDefinition,
@@ -291,6 +292,27 @@ async def test_call_count_tracking(self):
         _ = [e async for e in s2]
         assert provider.call_count == 2
 
+    async def test_provider_metadata_set(self):
+        provider = FauxProvider()
+        provider.set_responses([faux_assistant_message("hello")])
+        model = Model(id="faux-1", provider="faux")
+        stream = await provider.stream(model, [])
+        _ = [e async for e in stream]
+        result = await stream.result()
+        assert result.provider_id == "faux"
+        assert result.model_id == "faux-1"
+
+    async def test_provider_metadata_on_partial_events(self):
+        provider = FauxProvider()
+        provider.set_responses([faux_assistant_message("hello")])
+        model = Model(id="faux-1", provider="faux")
+        stream = await provider.stream(model, [])
+        events = [e async for e in stream]
+        start_event = next(e for e in events if e.type == "start")
+        assert start_event.partial is not None
+        assert start_event.partial.provider_id == "faux"
+        assert start_event.partial.model_id == "faux-1"
+
 
 class TestFauxProviderExtendedFactory:
     """Tests for extended factory signature (messages, model, system_prompt, tools)."""