|
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