fix input postprocess

wandb · Jan 3, 2025 · 4129525 · 4129525
1 parent 8c07bd9
commit 4129525
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/weave/integrations/langchain_nvidia_ai_endpoints/langchain_nv_ai_endpoints.py b/weave/integrations/langchain_nvidia_ai_endpoints/langchain_nv_ai_endpoints.py
@@ -142,6 +142,8 @@ def postprocess_inputs_to_openai_format(
         "n": n,
         "stream": stream,
     }
+
+    weave_report.update(chat_nvidia_obj.model_dump(exclude_unset=True, exclude_none=True))
 
     return ProcessedInputs(
         original_args=original_args,