feat(langchain): add update_trace argument

[hassiebp]

Important

Adds update_trace parameter to LangchainCallbackHandler for optional trace-level data updates on root-level chains.

• Behavior:
  • Adds update_trace parameter to LangchainCallbackHandler constructor, defaulting to False.
  • Updates trace with input, name, and metadata on root-level chain start (on_chain_start).
  • Updates trace with output and input on root-level chain end (on_chain_end).
• Conditional Logic:
  • Trace updates only occur for root-level chains to prevent duplicate data in nested chains.
  • Uses span.update_trace() method for trace updates.
• Backward Compatibility:
  • Default update_trace is False, maintaining existing behavior unless explicitly enabled.

^{This description was created by for e96f13b. You can customize this summary. It will automatically update as commits are pushed.}

---

Disclaimer: Experimental PR review

Greptile Summary

This PR introduces a new optional update_trace parameter to the LangchainCallbackHandler class, allowing users to automatically populate trace-level input and output data when using Langchain with Langfuse. The implementation adds trace updates at two key points in the chain execution lifecycle:

1. Chain Start: When a root-level chain begins (on_chain_start with parent_run_id of None), the trace is updated with the chain's input, name, and metadata
2. Chain End: When a root-level chain completes (on_chain_end), the trace is updated with both the final output and the original input

The feature is controlled by a boolean flag passed to the constructor (update_trace=False by default) and only activates for root-level chains to avoid duplicate data in nested chain scenarios. This enhancement bridges a gap between span-level observation data (which was already captured) and trace-level data, providing users with a complete picture of what went into and came out of their entire chain workflow.

The change integrates seamlessly with the existing callback handler architecture, using the same span.update_trace() method that's available elsewhere in the codebase. The conditional logic ensures backward compatibility while offering opt-in functionality for users who need trace-level input/output visibility.

Confidence score: 4/5

• This PR is safe to merge with minimal risk as it adds optional functionality without breaking existing behavior
• Score reflects well-structured conditional logic and proper integration with existing patterns, though testing coverage for the new feature isn't visible in the diff
• Pay close attention to the callback handler file to ensure the conditional logic works correctly in all chain execution scenarios

[greptile-apps]

_{1 file reviewed, no comments}

_{Edit Code Review Bot Settings | Greptile}