From 4129525488555d376685cd665d9c18e6b2edc81c Mon Sep 17 00:00:00 2001 From: Abraham Leal <45460452+abraham-leal@users.noreply.github.com> Date: Thu, 2 Jan 2025 18:48:24 -0600 Subject: [PATCH] fix input postprocess --- .../langchain_nvidia_ai_endpoints/langchain_nv_ai_endpoints.py | 2 ++ 1 file changed, 2 insertions(+) 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 index 112046627b55..a70424f4070e 100644 --- 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,