|
40 | 40 | @pytest.mark.vcr()
|
41 | 41 | @pytest.mark.anyio
|
42 | 42 | async def test_pydantic_ai_mcp_sampling(exporter: TestExporter):
|
43 |
| - logfire.instrument_pydantic_ai(version=2) |
| 43 | + logfire.instrument_pydantic_ai(version=3) |
44 | 44 |
|
45 | 45 | fastmcp = FastMCP()
|
46 | 46 |
|
@@ -403,14 +403,15 @@ async def client_streams(self):
|
403 | 403 | },
|
404 | 404 | },
|
405 | 405 | {
|
406 |
| - 'name': 'agent run', |
| 406 | + 'name': 'invoke_agent agent', |
407 | 407 | 'context': {'trace_id': 2, 'span_id': 19, 'is_remote': False},
|
408 | 408 | 'parent': {'trace_id': 2, 'span_id': 17, 'is_remote': False},
|
409 | 409 | 'start_time': 14000000000,
|
410 | 410 | 'end_time': 23000000000,
|
411 | 411 | 'attributes': {
|
412 | 412 | 'model_name': 'mcp-sampling',
|
413 | 413 | 'agent_name': 'agent',
|
| 414 | + 'gen_ai.agent.name': 'agent', |
414 | 415 | 'logfire.msg': 'agent run',
|
415 | 416 | 'logfire.span_type': 'span',
|
416 | 417 | 'final_result': """\
|
@@ -527,18 +528,18 @@ async def client_streams(self):
|
527 | 528 | },
|
528 | 529 | },
|
529 | 530 | {
|
530 |
| - 'name': 'running tool', |
| 531 | + 'name': 'execute_tool joker', |
531 | 532 | 'context': {'trace_id': 2, 'span_id': 13, 'is_remote': False},
|
532 | 533 | 'parent': {'trace_id': 2, 'span_id': 11, 'is_remote': False},
|
533 | 534 | 'start_time': 11000000000,
|
534 | 535 | 'end_time': 26000000000,
|
535 | 536 | 'attributes': {
|
536 | 537 | 'gen_ai.tool.name': 'joker',
|
537 | 538 | 'gen_ai.tool.call.id': 'call_YWeIZ4oGGwEnk9GIb443ZNys',
|
538 |
| - 'tool_arguments': {'theme': 'socks'}, |
| 539 | + 'gen_ai.tool.call.arguments': {'theme': 'socks'}, |
539 | 540 | 'logfire.msg': 'running tool: joker',
|
540 | 541 | 'logfire.span_type': 'span',
|
541 |
| - 'tool_response': """\ |
| 542 | + 'gen_ai.tool.call.result': """\ |
542 | 543 | Why did the sock break up with the shoe?
|
543 | 544 |
|
544 | 545 | Because it found something more "sole-ful!"\
|
@@ -679,14 +680,15 @@ async def client_streams(self):
|
679 | 680 | },
|
680 | 681 | },
|
681 | 682 | {
|
682 |
| - 'name': 'agent run', |
| 683 | + 'name': 'invoke_agent agent', |
683 | 684 | 'context': {'trace_id': 2, 'span_id': 3, 'is_remote': False},
|
684 | 685 | 'parent': None,
|
685 | 686 | 'start_time': 3000000000,
|
686 | 687 | 'end_time': 34000000000,
|
687 | 688 | 'attributes': {
|
688 | 689 | 'model_name': 'gpt-4o',
|
689 | 690 | 'agent_name': 'agent',
|
| 691 | + 'gen_ai.agent.name': 'agent', |
690 | 692 | 'logfire.msg': 'agent run',
|
691 | 693 | 'logfire.span_type': 'span',
|
692 | 694 | 'final_result': """\
|
|
0 commit comments