LLM Custom Attributes Context Manager API

[umaannamalai]

Add new WithLlmCustomAttributes context manager API

[github-actions]

🦙 MegaLinter status: ✅ SUCCESS

Descriptor	Linter	Files	Fixed	Errors	Elapsed time
✅ PYTHON	bandit	5		0	6.63s
✅ PYTHON	black	20	0	0	3.11s
✅ PYTHON	flake8	20		0	1.66s
✅ PYTHON	isort	20	0	0	0.29s
✅ PYTHON	pylint	20		0	11.3s

See detailed report in MegaLinter reports
Set VALIDATE_ALL_CODEBASE: true in mega-linter.yml to validate all sources, not only the diff

MegaLinter is graciously provided by

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 88.23529% with 4 lines in your changes missing coverage. Please review.

Project coverage is 81.24%. Comparing base (2d17237) to head (182cc71).

Files with missing lines	Patch %	Lines
newrelic/api/llm_custom_attributes.py	80.00%	2 Missing and 2 partials ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1214      +/-   ##
==========================================
+ Coverage   81.20%   81.24%   +0.03%     
==========================================
  Files         197      198       +1     
  Lines       21644    21675      +31     
  Branches     3428     3434       +6     
==========================================
+ Hits        17576    17609      +33     
+ Misses       2937     2935       -2     
  Partials     1131     1131              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[hmstepanek]

I'm going to let you fix the implementation issues first and then I'll take a look at the tests. It looks like there's just some small but large impact issues with the implementation and I think once we fix those the tests will probably be in better shape too.

[hmstepanek]

I think we should just use our regular approach of attaching the attrs to the transaction without using contextvars. I don't see that it's needed in this case and I think simpler is better here. Other than that, I just had one minor suggestion for one of the tests. I also committed a small change to undo some unnecessary formatting changes.

[hmstepanek]

Just a couple small suggestions about the tests-otherwise looks good!