Skip to content

Commit 00649e5

Browse files
committed
Fix deprecated token callback, add deprecation on cost feature
1 parent 8f8aebe commit 00649e5

File tree

2 files changed

+17
-5
lines changed

2 files changed

+17
-5
lines changed

Diff for: ddtrace/contrib/internal/langchain/patch.py

+12-5
Original file line numberDiff line numberDiff line change
@@ -129,11 +129,18 @@ def _tag_openai_token_usage(
129129
total_cost = span.get_metric(TOTAL_COST) or 0
130130
if not propagate and get_openai_token_cost_for_model:
131131
try:
132-
completion_cost = get_openai_token_cost_for_model(
133-
span.get_tag(MODEL),
134-
span.get_metric(COMPLETION_TOKENS),
135-
is_completion=True,
136-
)
132+
if parse_version(langchain_community.__version__) >= (0, 3, 13):
133+
completion_cost = get_openai_token_cost_for_model(
134+
span.get_tag(MODEL),
135+
span.get_metric(COMPLETION_TOKENS),
136+
token_type=2,
137+
)
138+
else:
139+
completion_cost = get_openai_token_cost_for_model(
140+
span.get_tag(MODEL),
141+
span.get_metric(COMPLETION_TOKENS),
142+
is_completion=True,
143+
)
137144
prompt_cost = get_openai_token_cost_for_model(span.get_tag(MODEL), span.get_metric(PROMPT_TOKENS))
138145
total_cost = completion_cost + prompt_cost
139146
except ValueError:
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
deprecations:
3+
- |
4+
langchain: The `langchain.tokens.total_cost` span metric for OpenAI calls is now deprecated and will be removed in `ddtrace==4.0`.
5+
For cost estimation of OpenAI calls, enable LLM Observability.

0 commit comments

Comments
 (0)