diff --git a/literalai/instrumentation/openai.py b/literalai/instrumentation/openai.py index a3117ec..8922ced 100644 --- a/literalai/instrumentation/openai.py +++ b/literalai/instrumentation/openai.py @@ -310,6 +310,8 @@ def streaming_response( and generation.model != chunk.model ): generation.model = chunk.model + if generation.settings: + generation.settings["model"] = chunk.model yield chunk