Skip to content

Commit b95770b

Browse files
committed
tests
1 parent e1e1b26 commit b95770b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/tracing/test_decorator.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,6 @@ def test_span_templates_ai_dicts(sentry_init, capture_events):
127127

128128
@sentry_sdk.trace(template=SPANTEMPLATE.AI_TOOL)
129129
def my_tool(arg1, arg2):
130-
"""This is a tool function."""
131130
return {
132131
"output": "my_tool_result",
133132
"usage": {
@@ -189,14 +188,14 @@ def my_agent():
189188
)
190189
assert tool_span["data"] == {
191190
"gen_ai.tool.name": "test_decorator.test_span_templates_ai_dicts.<locals>.my_tool",
192-
"gen_ai.tool.description": "This is a tool function.",
193191
"gen_ai.operation.name": "execute_tool",
194192
"gen_ai.usage.input_tokens": 10,
195193
"gen_ai.usage.output_tokens": 20,
196194
"gen_ai.usage.total_tokens": 30,
197195
"thread.id": mock.ANY,
198196
"thread.name": mock.ANY,
199197
}
198+
assert "gen_ai.tool.description" not in tool_span["data"]
200199

201200
assert chat_span["op"] == "gen_ai.chat"
202201
assert chat_span["description"] == "chat my-gpt-4o-mini"

0 commit comments

Comments
 (0)