Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: remove token on crewai kickoff chain span #1213

Merged
merged 12 commits into from
Jan 17, 2025
Merged

fix: remove token on crewai kickoff chain span #1213

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
5b9e9ef
Update _wrappers.py
nate-mar Jan 16, 2025
2a195e4
Update test_instrumentor.py
nate-mar Jan 17, 2025
d7a9471
add comment and remove token vars
nate-mar Jan 17, 2025
2de39f7
reenable crewai checks
nate-mar Jan 17, 2025
b492a3f
Merge branch 'main' into remove-token-on-kickoff
nate-mar Jan 17, 2025
27d0160
debug failure
nate-mar Jan 17, 2025
e638953
Merge branch 'remove-token-on-kickoff' of https://github.com/Arize-ai…
nate-mar Jan 17, 2025
1b60cc5
Update tox.ini
nate-mar Jan 17, 2025
a99ef57
Update tox.ini
nate-mar Jan 17, 2025
a050cfa
Update test_instrumentor.py
nate-mar Jan 17, 2025
fbca9d9
Update tox.ini
nate-mar Jan 17, 2025
ab9b6c7
Update tox.ini
nate-mar Jan 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 0 additions & 17 deletions ...peninference-instrumentation-crewai/src/openinference/instrumentation/crewai/_wrappers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -214,20 +214,6 @@ def __call__(
)
try:
crew_output = wrapped(*args, **kwargs)
usage_metrics = crew.usage_metrics
if isinstance(usage_metrics, dict):
if (prompt_tokens := usage_metrics.get("prompt_tokens")) is not None:
span.set_attribute(LLM_TOKEN_COUNT_PROMPT, int(prompt_tokens))
if (completion_tokens := usage_metrics.get("completion_tokens")) is not None:
span.set_attribute(LLM_TOKEN_COUNT_COMPLETION, int(completion_tokens))
if (total_tokens := usage_metrics.get("total_tokens")) is not None:
span.set_attribute(LLM_TOKEN_COUNT_TOTAL, int(total_tokens))
else:
# version 0.51 and onwards
span.set_attribute(LLM_TOKEN_COUNT_PROMPT, usage_metrics.prompt_tokens)
span.set_attribute(LLM_TOKEN_COUNT_COMPLETION, usage_metrics.completion_tokens)
span.set_attribute(LLM_TOKEN_COUNT_TOTAL, usage_metrics.total_tokens)

except Exception as exception:
span.set_status(trace_api.Status(trace_api.StatusCode.ERROR, str(exception)))
span.record_exception(exception)
Expand Down Expand Up @@ -298,6 +284,3 @@ def __call__(
OPENINFERENCE_SPAN_KIND = SpanAttributes.OPENINFERENCE_SPAN_KIND
OUTPUT_VALUE = SpanAttributes.OUTPUT_VALUE
OUTPUT_MIME_TYPE = SpanAttributes.OUTPUT_MIME_TYPE
LLM_TOKEN_COUNT_PROMPT = SpanAttributes.LLM_TOKEN_COUNT_PROMPT
LLM_TOKEN_COUNT_COMPLETION = SpanAttributes.LLM_TOKEN_COUNT_COMPLETION
LLM_TOKEN_COUNT_TOTAL = SpanAttributes.LLM_TOKEN_COUNT_TOTAL
9 changes: 6 additions & 3 deletions ...ce-instrumentation-crewai/tests/openinference/instrumentation/crewai/test_instrumentor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -113,9 +113,12 @@ def test_crewai_instrumentation(
checked_spans += 1
assert attributes.get("openinference.span.kind") == "CHAIN"
assert attributes.get("output.value")
assert attributes.get("llm.token_count.prompt") == 5751
assert attributes.get("llm.token_count.completion") == 1793
assert attributes.get("llm.token_count.total") == 7544
# assert that there are no tokens on the kickoff chain so that we do not
# double count token when a user is also instrumenting with another instrumentor
# that provides token counts via the spans.
assert attributes.get("llm.token_count.prompt") == 0
assert attributes.get("llm.token_count.completion") == 0
assert attributes.get("llm.token_count.total") == 0
assert span.status.is_ok
elif span.name == "ToolUsage._use":
checked_spans += 1
Expand Down
Loading