Obtaining file:///testbed
  Checking if build backend supports build_editable: started
  Checking if build backend supports build_editable: finished with status 'done'
  Preparing editable metadata (pyproject.toml): started
  Preparing editable metadata (pyproject.toml): finished with status 'done'
Building wheels for collected packages: mlflow
  Building editable for mlflow (pyproject.toml): started
  Building editable for mlflow (pyproject.toml): finished with status 'done'
  Created wheel for mlflow: filename=mlflow-3.6.1.dev0-0.editable-py3-none-any.whl size=17164 sha256=367553715eb35839c9fe737cc14b15c9eaa7e9b0560b06c68da8448ea66fa5d2
  Stored in directory: /tmp/pip-ephem-wheel-cache-5sa7wzbt/wheels/49/5f/4f/01b7d31ea6ddac9643da788242d33e5fd3467d95d17b032c6b
Successfully built mlflow
Installing collected packages: mlflow
  Attempting uninstall: mlflow
    Found existing installation: mlflow 3.6.1.dev0
    Uninstalling mlflow-3.6.1.dev0:
      Successfully uninstalled mlflow-3.6.1.dev0
Successfully installed mlflow-3.6.1.dev0
WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning.
=== Running FAIL_TO_PASS tests ===
============================= test session starts ==============================
platform linux -- Python 3.11.14, pytest-9.0.1, pluggy-1.6.0 -- /opt/miniconda3/envs/testbed/bin/python3.11
cachedir: .pytest_cache
rootdir: /testbed
configfile: pyproject.toml
plugins: anyio-4.11.0, asyncio-1.3.0, cov-7.0.0, mock-3.15.1, timeout-2.4.0, xdist-3.8.0
asyncio: mode=Mode.STRICT, debug=False, asyncio_default_fixture_loop_scope=None, asyncio_default_test_loop_scope=function
timeout: 1200.0s
timeout method: signal
timeout func_only: False
collecting ... collected 50 items

tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[anthropic.claude-3-5-sonnet-20241022-v2:0-llm_request0-llm_response0-expected_usage0] FAILED | MEM 147.1/755.2 GB | DISK 2.1/10.0 GB [  2%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[ai21.jamba-instruct-v1:0-llm_request1-llm_response1-expected_usage1] FAILED | MEM 147.1/755.2 GB | DISK 2.1/10.0 GB [  4%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[us.amazon.nova-lite-v1:0-llm_request2-llm_response2-expected_usage2] FAILED | MEM 147.2/755.2 GB | DISK 2.1/10.0 GB [  6%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[cohere.command-r-plus-v1:0-llm_request3-llm_response3-None] FAILED | MEM 147.2/755.2 GB | DISK 2.1/10.0 GB [  8%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[meta.llama3-8b-instruct-v1:0-llm_request4-llm_response4-expected_usage4] FAILED | MEM 147.2/755.2 GB | DISK 2.1/10.0 GB [ 10%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_embeddings FAILED | MEM 147.2/755.2 GB | DISK 2.1/10.0 GB [ 12%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_capture_exception FAILED | MEM 147.2/755.2 GB | DISK 2.1/10.0 GB [ 14%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_stream FAILED | MEM 147.2/755.2 GB | DISK 2.1/10.0 GB [ 16%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config0] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 18%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config1] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 20%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request0-response0-expected_chat_attr0-None-expected_usage0] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 22%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request1-response1-expected_chat_attr1-expected_tool_attr1-expected_usage1] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 24%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request2-response2-expected_chat_attr2-None-expected_usage2] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 26%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request3-response3-expected_chat_attr3-None-expected_usage3] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 28%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_error FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 30%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_skip_unsupported_content FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 32%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream[_request0-expected_response0-expected_chat_attr0-None-expected_usage0] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 34%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream[_request1-expected_response1-expected_chat_attr1-expected_tool_attr1-expected_usage1] FAILED | MEM 147.3/755.2 GB | DISK 2.1/10.0 GB [ 36%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_missing_usage_field] FAILED | MEM 147.1/755.2 GB | DISK 2.1/10.0 GB [ 38%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_empty_usage_dict] FAILED | MEM 147.1/755.2 GB | DISK 2.1/10.0 GB [ 40%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_only_input_tokens] FAILED | MEM 147.1/755.2 GB | DISK 2.1/10.0 GB [ 42%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_only_output_tokens] FAILED | MEM 147.1/755.2 GB | DISK 2.1/10.0 GB [ 44%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_zero_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 46%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_large_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 48%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_none_token_values] FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB [ 50%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_string_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 52%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_duplicate_keys_in_usage] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 54%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[no_usage_in_stream] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 56%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_only_in_first_chunk] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 58%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_updated_in_later_chunk] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 60%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[malformed_usage_in_chunk] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 62%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_with_none_values] FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB [ 64%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[same_key_in_multiple_chunks] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 66%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_missing_usage_field] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 68%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_empty_usage_dict] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 70%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_input_tokens] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 72%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_output_tokens] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 74%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_zero_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 76%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_large_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 78%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_none_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 80%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_string_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 82%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_missing_usage_field] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 84%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_empty_usage_dict] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 86%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_input_tokens] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 88%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_output_tokens] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 90%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_zero_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 92%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_large_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 94%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_none_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 96%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_string_token_values] FAILED | MEM 147.0/755.2 GB | DISK 2.1/10.0 GB [ 98%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_malformed_usage_in_metadata] FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB [100%]

=================================== FAILURES ===================================
_ test_bedrock_autolog_invoke_model_llm[anthropic.claude-3-5-sonnet-20241022-v2:0-llm_request0-llm_response0-expected_usage0] _
tests/bedrock/test_bedrock_autolog.py:209: in test_bedrock_autolog_invoke_model_llm
    mlflow.bedrock.autolog()
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
        llm_request = {'anthropic_version': 'bedrock-2023-05-31', 'max_tokens': 300, 'messages': [{'content': 'Hi', 'role': 'user'}], 'temperature': 0.1, ...}
        llm_response = {'content': [{'text': 'Hello! How can I help you today?', 'type': 'text'}], 'id': 'id-123', 'model': 'claude-3-5-sonnet-20241022', 'role': 'assistant', ...}
        model_id   = 'anthropic.claude-3-5-sonnet-20241022-v2:0'
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3a214cd0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3a214cd0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39e99800>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_invoke_model_llm[ai21.jamba-instruct-v1:0-llm_request1-llm_response1-expected_usage1] _
tests/bedrock/test_bedrock_autolog.py:209: in test_bedrock_autolog_invoke_model_llm
    mlflow.bedrock.autolog()
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
        llm_request = {'max_tokens': 300, 'messages': [{'content': 'Hi', 'role': 'user'}], 'temperature': 0.1}
        llm_response = {'choices': [{'finish_reason': 'stop', 'index': 0, 'message': {'content': 'Hello! How can I help you today?', 'role': ...istant', 'tool_calls': None}}], 'id': 'id-123', 'meta': {'requestDurationMillis': 288}, 'model': 'jamba-instruct', ...}
        model_id   = 'ai21.jamba-instruct-v1:0'
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39d19750>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39d19750>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39f4f2e0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_invoke_model_llm[us.amazon.nova-lite-v1:0-llm_request2-llm_response2-expected_usage2] _
tests/bedrock/test_bedrock_autolog.py:209: in test_bedrock_autolog_invoke_model_llm
    mlflow.bedrock.autolog()
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
        llm_request = {'inferenceConfig': {'max_new_tokens': 512, 'temperature': 0.5}, 'messages': [{'content': [{'text': 'Hi!'}], 'role': 'user'}], 'schemaVersion': 'messages-v1', 'system': [{'text': 'This is a system prompt'}]}
        llm_response = {'message': {'content': [{'text': 'Sure! How can I help you today?'}], 'role': 'assistant'}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 12, 'totalTokens': 20}}
        model_id   = 'us.amazon.nova-lite-v1:0'
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39d12550>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39d12550>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39f4ed40>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_invoke_model_llm[cohere.command-r-plus-v1:0-llm_request3-llm_response3-None] _
tests/bedrock/test_bedrock_autolog.py:209: in test_bedrock_autolog_invoke_model_llm
    mlflow.bedrock.autolog()
        expected_usage = None
        llm_request = {'max_tokens': 512, 'message': 'Hi!', 'temperature': 0.5}
        llm_response = {'chat_history': [{'message': 'Hi', 'role': 'USER'}, {'message': 'Hello! How can I help you today?', 'role': 'CHATBOT'}], 'finish_reason': 'COMPLETE', 'generation_id': 'generation-id-123', 'response_id': 'id-123', ...}
        model_id   = 'cohere.command-r-plus-v1:0'
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39d291d0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39d291d0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da8040>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_invoke_model_llm[meta.llama3-8b-instruct-v1:0-llm_request4-llm_response4-expected_usage4] _
tests/bedrock/test_bedrock_autolog.py:209: in test_bedrock_autolog_invoke_model_llm
    mlflow.bedrock.autolog()
        expected_usage = {'input_tokens': 2, 'output_tokens': 12, 'total_tokens': 14}
        llm_request = {'max_gen_len': 300, 'prompt': '\n<|begin_of_text|><|start_header_id|>user<|end_header_id|>\nHi!\n<|eot_id|>\n<|start_header_id|>assistant<|end_header_id|>\n', 'temperature': 0.1}
        llm_response = {'generation': 'Hello! How can I help you today?', 'generation_token_count': 12, 'prompt_token_count': 2, 'stop_reason': 'stop'}
        model_id   = 'meta.llama3-8b-instruct-v1:0'
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39d2afd0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39d2afd0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da8540>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_________________ test_bedrock_autolog_invoke_model_embeddings _________________
tests/bedrock/test_bedrock_autolog.py:244: in test_bedrock_autolog_invoke_model_embeddings
    mlflow.bedrock.autolog()
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3a215050>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3a215050>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da8860>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_____________ test_bedrock_autolog_invoke_model_capture_exception ______________
tests/bedrock/test_bedrock_autolog.py:278: in test_bedrock_autolog_invoke_model_capture_exception
    mlflow.bedrock.autolog()
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39d772d0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39d772d0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da8ae0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
___________________ test_bedrock_autolog_invoke_model_stream ___________________
tests/bedrock/test_bedrock_autolog.py:318: in test_bedrock_autolog_invoke_model_stream
    mlflow.bedrock.autolog()
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39cb7f10>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39cb7f10>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da9300>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:42 INFO mlflow.tracking.fluent: Active model is cleared
_________________ test_bedrock_autolog_trace_disabled[config0] _________________
tests/bedrock/test_bedrock_autolog.py:427: in test_bedrock_autolog_trace_disabled
    traces = get_traces()
             ^^^^^^^^^^^^
        client     = <botocore.client.BedrockRuntime object at 0x735f39d76090>
        config     = {'disable': True}
        request_body = '{"messages": [{"role": "user", "content": "Hi"}], "max_tokens": 300, "anthropic_version": "bedrock-2023-05-31", "temperature": 0.1, "top_p": 0.9}'
        response   = {'ResponseMetadata': {'HTTPHeaders': {'content-type': 'application/json'}, 'HTTPStatusCode': 200, 'RequestId': 'reques...ryAttempts': 0}, 'body': <botocore.response.StreamingBody object at 0x735f39e90c70>, 'contentType': 'application/json'}
        response_body = {'content': [{'text': 'Hello! How can I help you today?', 'type': 'text'}], 'id': 'id-123', 'model': 'claude-3-5-sonnet-20241022', 'role': 'assistant', ...}
tests/tracing/helper.py:153: in get_traces
    return TracingClient().search_traces(
        experiment_id = None
mlflow/tracing/client.py:329: in search_traces
    trace_infos, next_token = self._search_traces(
        executor   = <concurrent.futures.thread.ThreadPoolExecutor object at 0x735f37a2e1d0>
        experiment_ids = None
        filter_string = None
        include_spans = True
        locations  = ['0']
        max_results = 100
        max_workers = 256
        model_id   = None
        next_max_results = 100
        next_token = None
        order_by   = None
        page_token = None
        run_id     = None
        self       = <mlflow.tracing.client.TracingClient object at 0x735f39acaf90>
        traces     = []
mlflow/tracing/client.py:786: in _search_traces
    return self.store.search_traces(
        experiment_ids = None
        filter_string = None
        locations  = ['0']
        max_results = 100
        model_id   = None
        order_by   = None
        page_token = None
        self       = <mlflow.tracing.client.TracingClient object at 0x735f39acaf90>
mlflow/store/tracking/abstract_store.py:389: in search_traces
    raise NotImplementedError
E   NotImplementedError
        experiment_ids = None
        filter_string = None
        locations  = ['0']
        max_results = 100
        model_id   = None
        order_by   = None
        page_token = None
        self       = <mlflow.store.tracking.sqlalchemy_store.SqlAlchemyStore object at 0x735f39d16a10>
----------------------------- Captured stderr call -----------------------------
2026/06/27 06:23:44 INFO mlflow.store.db.utils: Creating initial MLflow database tables...
2026/06/27 06:23:44 INFO mlflow.store.db.utils: Updating database tables
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade  -> 451aebb31d03, add metric step
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 451aebb31d03 -> 90e64c465722, migrate user column to tags
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 90e64c465722 -> 181f10493468, allow nulls for metric values
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 181f10493468 -> df50e92ffc5e, Add Experiment Tags Table
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade df50e92ffc5e -> 7ac759974ad8, Update run tags with larger limit
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 7ac759974ad8 -> 89d4b8295536, create latest metrics table
2026-06-27 06:23:44 INFO  [89d4b8295536_create_latest_metrics_table_py] Migration complete!
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 89d4b8295536 -> 2b4d017a5e9b, add model registry tables to db
2026-06-27 06:23:44 INFO  [2b4d017a5e9b_add_model_registry_tables_to_db_py] Adding registered_models and model_versions tables to database.
2026-06-27 06:23:44 INFO  [2b4d017a5e9b_add_model_registry_tables_to_db_py] Migration complete!
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 2b4d017a5e9b -> cfd24bdc0731, Update run status constraint with killed
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade cfd24bdc0731 -> 0a8213491aaa, drop_duplicate_killed_constraint
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 0a8213491aaa -> 728d730b5ebd, add registered model tags table
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 728d730b5ebd -> 27a6a02d2cf1, add model version tags table
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 27a6a02d2cf1 -> 84291f40a231, add run_link to model_version
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade 84291f40a231 -> a8c4a736bde6, allow nulls for run_id
2026-06-27 06:23:44 INFO  [alembic.runtime.migration] Running upgrade a8c4a736bde6 -> 39d1c3be5f05, add_is_nan_constraint_for_metrics_tables_if_necessary
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 39d1c3be5f05 -> c48cb773bb87, reset_default_value_for_is_nan_in_metrics_table_for_mysql
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade c48cb773bb87 -> bd07f7e963c5, create index on run_uuid
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade bd07f7e963c5 -> 0c779009ac13, add deleted_time field to runs table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 0c779009ac13 -> cc1f77228345, change param value length to 500
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade cc1f77228345 -> 97727af70f4d, Add creation_time and last_update_time to experiments table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 97727af70f4d -> 3500859a5d39, Add Model Aliases table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 3500859a5d39 -> 7f2a7d5fae7d, add datasets inputs input_tags tables
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 7f2a7d5fae7d -> 2d6e25af4d3e, increase max param val length from 500 to 8000
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 2d6e25af4d3e -> acf3f17fdcc7, add storage location field to model versions
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade acf3f17fdcc7 -> 867495a8f9d4, add trace tables
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 867495a8f9d4 -> 5b0e9adcef9c, add cascade deletion to trace tables foreign keys
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 5b0e9adcef9c -> 4465047574b1, increase max dataset schema size
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 4465047574b1 -> f5a4f2784254, increase run tag value limit to 8000
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade f5a4f2784254 -> 0584bdc529eb, add cascading deletion to datasets from experiments
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 0584bdc529eb -> 400f98739977, add logged model tables
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 400f98739977 -> 6953534de441, add step to inputs table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 6953534de441 -> bda7b8c39065, increase_model_version_tag_value_limit
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade bda7b8c39065 -> cbc13b556ace, add V3 trace schema columns
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade cbc13b556ace -> 770bee3ae1dd, add assessments table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 770bee3ae1dd -> a1b2c3d4e5f6, add spans table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade a1b2c3d4e5f6 -> de4033877273, create entity_associations table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade de4033877273 -> 1a0cddfcaa16, Add webhooks and webhook_events tables
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 1a0cddfcaa16 -> 534353b11cbc, add scorer tables
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 534353b11cbc -> 71994744cf8e, add evaluation datasets
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 71994744cf8e -> 3da73c924c2f, add outputs to dataset record
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Running upgrade 3da73c924c2f -> bf29a5ff90ea, add jobs table
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
2026-06-27 06:23:45 INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_________________ test_bedrock_autolog_trace_disabled[config1] _________________
tests/bedrock/test_bedrock_autolog.py:411: in test_bedrock_autolog_trace_disabled
    mlflow.bedrock.autolog(**config)
        config     = {'log_traces': False}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {'log_traces': False}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': False, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {'log_traces': False}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = False
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fe0250>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fe0250>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f351f1bc0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse[_request0-response0-expected_chat_attr0-None-expected_usage0] _
tests/bedrock/test_bedrock_autolog.py:761: in test_bedrock_autolog_converse
    mlflow.bedrock.autolog()
        _request   = {'inferenceConfig': {'maxTokens': 300, 'temperature': 0.1, 'topP': 0.9}, 'messages': [{'content': [{'text': 'Hi'}], 'role': 'user'}], 'modelId': 'anthropic.claude-3-5-sonnet-20241022-v2:0'}
        expected_chat_attr = [{'content': [{'text': 'Hi', 'type': 'text'}], 'role': 'user'}, {'content': [{'text': 'Hello! How can I help you today?', 'type': 'text'}], 'role': 'assistant'}]
        expected_tool_attr = None
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
        response   = {'metrics': {'latencyMs': 551}, 'output': {'message': {'content': [{'text': 'Hello! How can I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 12}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fe0910>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fe0910>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35205c60>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse[_request1-response1-expected_chat_attr1-expected_tool_attr1-expected_usage1] _
tests/bedrock/test_bedrock_autolog.py:761: in test_bedrock_autolog_converse
    mlflow.bedrock.autolog()
        _request   = {'inferenceConfig': {'maxTokens': 300, 'temperature': 0.1, 'topP': 0.9}, 'messages': [{'content': [{'text': "What's th...description': 'Get the current weather in a given location', 'inputSchema': {'json': {...}}, 'name': 'get_weather'}}]}}
        expected_chat_attr = [{'content': [{'text': "What's the weather like in San Francisco?", 'type': 'text'}], 'role': 'user'}, {'content': [{'..."location": "San Francisco, CA", "unit": "fahrenheit"}', 'name': 'get_weather'}, 'id': 'tool_2', 'type': 'function'}]}]
        expected_tool_attr = [{'function': {'description': 'Get the temperature unit in a given location', 'name': 'get_unit', 'parameters': {'prop...or "fahrenheit"', 'enum': [...], 'type': 'string'}}, 'required': ['location'], 'type': 'object'}}, 'type': 'function'}]
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
        response   = {'metrics': {'latencyMs': 551}, 'output': {'message': {'content': [{'text': "Now I'll check the current weather in San...lUseId': 'tool_2'}}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 12}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fef050>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fef050>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35205b20>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse[_request2-response2-expected_chat_attr2-None-expected_usage2] _
tests/bedrock/test_bedrock_autolog.py:761: in test_bedrock_autolog_converse
    mlflow.bedrock.autolog()
        _request   = {'messages': [{'content': [{'text': 'What text is in this image?'}, {'image': {'format': 'png', 'source': {...}}}], 'role': 'user'}], 'modelId': 'anthropic.claude-3-5-sonnet-20241022-v2:0'}
        expected_chat_attr = [{'content': [{'text': 'What text is in this image?', 'type': 'text'}, {'image_url': {'detail': 'auto', 'url': 'data:i...QmCC'}, 'type': 'image_url'}], 'role': 'user'}, {'content': [{'text': 'MLflow', 'type': 'text'}], 'role': 'assistant'}]
        expected_tool_attr = None
        expected_usage = {'input_tokens': 8, 'output_tokens': 2, 'total_tokens': 10}
        response   = {'output': {'message': {'content': [{'text': 'MLflow'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 2}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fef850>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fef850>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35205300>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse[_request3-response3-expected_chat_attr3-None-expected_usage3] _
tests/bedrock/test_bedrock_autolog.py:761: in test_bedrock_autolog_converse
    mlflow.bedrock.autolog()
        _request   = {'messages': [{'content': [{'text': 'What text is in this image?'}, {'image': {'format': 'png', 'source': {...}}}], 'role': 'user'}], 'modelId': 'anthropic.claude-3-5-sonnet-20241022-v2:0'}
        expected_chat_attr = [{'content': [{'text': 'What text is in this image?', 'type': 'text'}, {'image_url': {'detail': 'auto', 'url': 'data:i...QmCC'}, 'type': 'image_url'}], 'role': 'user'}, {'content': [{'text': 'MLflow', 'type': 'text'}], 'role': 'assistant'}]
        expected_tool_attr = None
        expected_usage = {'input_tokens': 8, 'output_tokens': 2, 'total_tokens': 10}
        response   = {'output': {'message': {'content': [{'text': 'MLflow'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 2}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34ef3690>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34ef3690>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35204040>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_____________________ test_bedrock_autolog_converse_error ______________________
tests/bedrock/test_bedrock_autolog.py:786: in test_bedrock_autolog_converse_error
    mlflow.bedrock.autolog()
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39ab9910>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39ab9910>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35405b20>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
____________ test_bedrock_autolog_converse_skip_unsupported_content ____________
tests/bedrock/test_bedrock_autolog.py:807: in test_bedrock_autolog_converse_skip_unsupported_content
    mlflow.bedrock.autolog()
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35229fd0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35229fd0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f3553e980>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream[_request0-expected_response0-expected_chat_attr0-None-expected_usage0] _
tests/bedrock/test_bedrock_autolog.py:858: in test_bedrock_autolog_converse_stream
    mlflow.bedrock.autolog()
        _request   = {'inferenceConfig': {'maxTokens': 300, 'temperature': 0.1, 'topP': 0.9}, 'messages': [{'content': [{'text': 'Hi'}], 'role': 'user'}], 'modelId': 'anthropic.claude-3-5-sonnet-20241022-v2:0'}
        expected_chat_attr = [{'content': [{'text': 'Hi', 'type': 'text'}], 'role': 'user'}, {'content': [{'text': 'Hello! How can I help you today?', 'type': 'text'}], 'role': 'assistant'}]
        expected_response = {'metrics': {'latencyMs': 551}, 'output': {'message': {'content': [{'text': 'Hello! How can I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 12}}
        expected_tool_attr = None
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35306e10>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35306e10>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da8040>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream[_request1-expected_response1-expected_chat_attr1-expected_tool_attr1-expected_usage1] _
tests/bedrock/test_bedrock_autolog.py:858: in test_bedrock_autolog_converse_stream
    mlflow.bedrock.autolog()
        _request   = {'inferenceConfig': {'maxTokens': 300, 'temperature': 0.1, 'topP': 0.9}, 'messages': [{'content': [{'text': "What's th...description': 'Get the current weather in a given location', 'inputSchema': {'json': {...}}, 'name': 'get_weather'}}]}}
        expected_chat_attr = [{'content': [{'text': "What's the weather like in San Francisco?", 'type': 'text'}], 'role': 'user'}, {'content': [{'..."location": "San Francisco, CA", "unit": "fahrenheit"}', 'name': 'get_weather'}, 'id': 'tool_2', 'type': 'function'}]}]
        expected_response = {'metrics': {'latencyMs': 551}, 'output': {'message': {'content': [{'text': "Now I'll check the current weather in San...lUseId': 'tool_2'}}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 8, 'outputTokens': 12}}
        expected_tool_attr = [{'function': {'description': 'Get the temperature unit in a given location', 'name': 'get_unit', 'parameters': {'prop...or "fahrenheit"', 'enum': [...], 'type': 'string'}}, 'required': ['location'], 'type': 'object'}}, 'type': 'function'}]
        expected_usage = {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35380150>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35380150>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35204c20>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_missing_usage_field] _
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'invoke_model_missing_usage_field', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f37a63f50>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f37a63f50>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35207a60>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
__ test_bedrock_autolog_token_usage_edge_cases[invoke_model_empty_usage_dict] __
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'invoke_model_empty_usage_dict', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3534d150>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3534d150>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa4a40>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_only_input_tokens] __
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'invoke_model_only_input_tokens', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34efafd0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34efafd0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa40e0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_only_output_tokens] _
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'invoke_model_only_output_tokens', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34efbb90>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34efbb90>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa4b80>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_zero_token_values] __
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': {'input_tokens': 0, 'output_tokens': 0, 'total_tokens': 0}, 'name': 'invoke_model_zero_token_values...ponse': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34ef9410>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34ef9410>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa4220>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_large_token_values] _
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': {'input_tokens': 999999, 'output_tokens': 888888, 'total_tokens': 1888887}, 'name': 'invoke_model_l...ponse': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34efad10>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34efad10>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39da8040>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_none_token_values] __
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'invoke_model_none_token_values', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34efb710>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34efb710>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f352076a0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_string_token_values] _
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'invoke_model_string_token_values', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35382310>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35382310>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa5440>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_token_usage_edge_cases[invoke_model_duplicate_keys_in_usage] _
tests/bedrock/test_bedrock_autolog.py:1059: in test_bedrock_autolog_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': {'input_tokens': 10, 'output_tokens': 5, 'total_tokens': 15}, 'name': 'invoke_model_duplicate_keys_...ponse': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35306d10>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35306d10>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa49a0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
____ test_bedrock_autolog_stream_token_usage_edge_cases[no_usage_in_stream] ____
tests/bedrock/test_bedrock_autolog.py:1242: in test_bedrock_autolog_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'chunks': [{'message': {'content': [], 'id': '123', 'role': 'assistant', 'type': 'message'}, 'type': 'message_start'}...x': 0, 'type': 'content_block_delta'}, {'type': 'message_stop'}], 'expected_usage': None, 'name': 'no_usage_in_stream'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35246a90>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35246a90>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa62a0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_stream_token_usage_edge_cases[usage_only_in_first_chunk] _
tests/bedrock/test_bedrock_autolog.py:1242: in test_bedrock_autolog_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'chunks': [{'message': {'content': [], 'id': '123', 'role': 'assistant', 'type': 'message', ...}, 'type': 'message_st...'}], 'expected_usage': {'input_tokens': 8, 'output_tokens': 1, 'total_tokens': 9}, 'name': 'usage_only_in_first_chunk'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35246390>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35246390>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa6f20>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_stream_token_usage_edge_cases[usage_updated_in_later_chunk] _
tests/bedrock/test_bedrock_autolog.py:1242: in test_bedrock_autolog_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'chunks': [{'message': {'content': [], 'id': '123', 'role': 'assistant', 'type': 'message', ...}, 'type': 'message_st...'expected_usage': {'input_tokens': 8, 'output_tokens': 12, 'total_tokens': 20}, 'name': 'usage_updated_in_later_chunk'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f39a9f210>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f39a9f210>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa5800>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_stream_token_usage_edge_cases[malformed_usage_in_chunk] _
tests/bedrock/test_bedrock_autolog.py:1242: in test_bedrock_autolog_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'chunks': [{'message': {'content': [], 'id': '123', 'role': 'assistant', 'type': 'message', ...}, 'type': 'message_st... 'type': 'content_block_delta'}, {'type': 'message_stop'}], 'expected_usage': None, 'name': 'malformed_usage_in_chunk'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3504df10>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3504df10>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa5ee0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
__ test_bedrock_autolog_stream_token_usage_edge_cases[usage_with_none_values] __
tests/bedrock/test_bedrock_autolog.py:1242: in test_bedrock_autolog_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'chunks': [{'message': {'content': [], 'id': '123', 'role': 'assistant', 'type': 'message', ...}, 'type': 'message_st...0, 'type': 'content_block_delta'}, {'type': 'message_stop'}], 'expected_usage': None, 'name': 'usage_with_none_values'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f351eb9d0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f351eb9d0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa7880>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_stream_token_usage_edge_cases[same_key_in_multiple_chunks] _
tests/bedrock/test_bedrock_autolog.py:1242: in test_bedrock_autolog_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'chunks': [{'message': {'content': [], 'id': '123', 'role': 'assistant', 'type': 'message', ...}, 'type': 'message_st...'expected_usage': {'input_tokens': 10, 'output_tokens': 12, 'total_tokens': 22}, 'name': 'same_key_in_multiple_chunks'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34f36290>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34f36290>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f35373e20>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_missing_usage_field] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'converse_missing_usage_field', 'response': {'output': {'message': {'content': [{'text': 'Hello! How can I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': None}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34f35750>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34f35750>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa4540>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_empty_usage_dict] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'converse_empty_usage_dict', 'response': {'output': {'message': {'content': [{'text': 'Hello! How can I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3504cf50>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3504cf50>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa7c40>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_input_tokens] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'converse_only_input_tokens', 'response': {'output': {'message': {'content': [{'text': 'Hello! How can I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 10}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3504d2d0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3504d2d0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa47c0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_output_tokens] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'converse_only_output_tokens', 'response': {'output': {'message': {'content': [{'text': 'Hello! How can I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'outputTokens': 5}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f3504f710>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f3504f710>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa6980>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_zero_token_values] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': {'input_tokens': 0, 'output_tokens': 0, 'total_tokens': 0}, 'name': 'converse_zero_token_values', '...I help you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 0, 'outputTokens': 0}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35229cd0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35229cd0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f680e0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_large_token_values] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': {'input_tokens': 999999, 'output_tokens': 888888, 'total_tokens': 1888887}, 'name': 'converse_large... today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': 999999, 'outputTokens': 888888}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35015ed0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35015ed0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f685e0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_none_token_values] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'converse_none_token_values', 'response': {'output': {'message': {'content': [{'text'... you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': None, 'outputTokens': None}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f350167d0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f350167d0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f68220>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:45 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_token_usage_edge_cases[converse_string_token_values] _
tests/bedrock/test_bedrock_autolog.py:1278: in test_bedrock_autolog_converse_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'expected_usage': None, 'name': 'converse_string_token_values', 'response': {'output': {'message': {'content': [{'tex...p you today?'}], 'role': 'assistant'}}, 'stopReason': 'end_turn', 'usage': {'inputTokens': '10', 'outputTokens': '5'}}}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34f2a990>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34f2a990>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f6a700>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_missing_usage_field] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...}, {'messageStop': {'stopReason': 'end_turn'}}], 'expected_usage': None, 'name': 'converse_stream_missing_usage_field'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34f2bb50>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34f2bb50>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f698a0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_empty_usage_dict] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...eason': 'end_turn'}}, {'metadata': {'usage': {}}}], 'expected_usage': None, 'name': 'converse_stream_empty_usage_dict'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35002990>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35002990>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa49a0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_input_tokens] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...}}, {'metadata': {'usage': {'inputTokens': 10}}}], 'expected_usage': None, 'name': 'converse_stream_only_input_tokens'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f35003a90>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f35003a90>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f39aa7380>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_output_tokens] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...}, {'metadata': {'usage': {'outputTokens': 5}}}], 'expected_usage': None, 'name': 'converse_stream_only_output_tokens'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fa9010>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fa9010>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f6a020>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_zero_token_values] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...pected_usage': {'input_tokens': 0, 'output_tokens': 0, 'total_tokens': 0}, 'name': 'converse_stream_zero_token_values'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34f85ad0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34f85ad0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f68400>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_large_token_values] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...input_tokens': 999999, 'output_tokens': 888888, 'total_tokens': 1888887}, 'name': 'converse_stream_large_token_values'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fc4a10>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fc4a10>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f6a160>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_none_token_values] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...': {'inputTokens': None, 'outputTokens': None}}}], 'expected_usage': None, 'name': 'converse_stream_none_token_values'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fc7590>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fc7590>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f69bc0>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_string_token_values] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...: {'inputTokens': '10', 'outputTokens': '5'}}}], 'expected_usage': None, 'name': 'converse_stream_string_token_values'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f34fb0ed0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f34fb0ed0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f68540>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
_ test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_malformed_usage_in_metadata] _
tests/bedrock/test_bedrock_autolog.py:1343: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    mlflow.bedrock.autolog()
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl... {'metadata': {'usage': 'not_a_dict'}}], 'expected_usage': None, 'name': 'converse_stream_malformed_usage_in_metadata'}
mlflow/utils/autologging_utils/__init__.py:77: in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
        args       = ()
        fn         = <function autologging_integration.<locals>.wrapper.<locals>.autolog at 0x735f39e99a80>
        kwargs     = {}
mlflow/utils/autologging_utils/__init__.py:463: in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
        _autolog   = <function autolog at 0x735f39e99940>
        args       = ()
        config_to_store = {'disable': False, 'log_traces': True, 'silent': False}
        default_params = {'disable': False, 'log_traces': True, 'silent': False}
        is_silent_mode = False
        kwargs     = {}
        name       = 'bedrock'
        param_spec = mappingproxy(OrderedDict([('log_traces', <Parameter "log_traces: bool = True">), ('disable', <Parameter "disable: bool = False">), ('silent', <Parameter "silent: bool = False">)]))
mlflow/bedrock/__init__.py:30: in autolog
    from mlflow.bedrock._autolog import patched_create_client
        ClientCreator = <class 'botocore.client.ClientCreator'>
        disable    = False
        log_traces = True
        silent     = False
mlflow/bedrock/_autolog.py:88: in <module>
    @skip_if_trace_disabled
     ^^^^^^^^^^^^^^^^^^^^^^
E   NameError: name 'skip_if_trace_disabled' is not defined
        Any        = typing.Any
        BaseClient = <class 'botocore.client.BaseClient'>
        ConverseStreamWrapper = <class 'mlflow.bedrock.stream.ConverseStreamWrapper'>
        FLAVOR_NAME = 'bedrock'
        InvokeModelStreamWrapper = <class 'mlflow.bedrock.stream.InvokeModelStreamWrapper'>
        SpanAttributeKey = <class 'mlflow.tracing.constant.SpanAttributeKey'>
        SpanType   = <class 'mlflow.entities.span.SpanType'>
        StreamingBody = <class 'botocore.response.StreamingBody'>
        _BEDROCK_RUNTIME_SERVICE_NAME = 'bedrock-runtime'
        _BEDROCK_SPAN_PREFIX = 'BedrockRuntime.'
        __builtins__ = <builtins>
        __cached__ = '/testbed/mlflow/bedrock/__pycache__/_autolog.cpython-311.pyc'
        __doc__    = None
        __file__   = '/testbed/mlflow/bedrock/_autolog.py'
        __loader__ = <_frozen_importlib_external.SourceFileLoader object at 0x735f37a635d0>
        __name__   = 'mlflow.bedrock._autolog'
        __package__ = 'mlflow.bedrock'
        __spec__   = ModuleSpec(name='mlflow.bedrock._autolog', loader=<_frozen_importlib_external.SourceFileLoader object at 0x735f37a635d0>, origin='/testbed/mlflow/bedrock/_autolog.py')
        _logger    = <Logger mlflow.bedrock._autolog (INFO)>
        _parse_usage_from_response = <function _parse_usage_from_response at 0x735f34f6b740>
        convert_tool_to_mlflow_chat_tool = <function convert_tool_to_mlflow_chat_tool at 0x735f3a2613a0>
        io         = <module 'io' (frozen)>
        json       = <module 'json' from '/opt/miniconda3/envs/testbed/lib/python3.11/json/__init__.py'>
        logging    = <module 'logging' from '/opt/miniconda3/envs/testbed/lib/python3.11/logging/__init__.py'>
        mlflow     = <module 'mlflow' from '/testbed/mlflow/__init__.py'>
        parse_complete_token_usage_from_response = <function parse_complete_token_usage_from_response at 0x735f39e9a700>
        set_span_chat_tools = <function set_span_chat_tools at 0x735feb578900>
        start_span_no_context = <function start_span_no_context at 0x735f3b8c7a60>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 06:23:46 INFO mlflow.tracking.fluent: Active model is cleared
============================= slowest 10 durations =============================
2.69s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config0]
0.02s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[anthropic.claude-3-5-sonnet-20241022-v2:0-llm_request0-llm_response0-expected_usage0]

(8 durations < 0.005s hidden.  Use -vv to show these durations.)

========================= per-file durations (sorted) ==========================
::group::per-file durations (sorted)

| Rank | File                                    | Duration | Tests | Min    | Max    | Avg    |
| ---- | --------------------------------------- | -------- | ----- | ------ | ------ | ------ |
| 1    | `tests/bedrock/test_bedrock_autolog.py` | 4.17s    | 50    | 0.015s | 2.710s | 0.083s |

::endgroup::

========================= command to run failed tests ==========================
pytest tests/bedrock/test_bedrock_autolog.py

=========================== short test summary info ============================
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[anthropic.claude-3-5-sonnet-20241022-v2:0-llm_request0-llm_response0-expected_usage0]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[ai21.jamba-instruct-v1:0-llm_request1-llm_response1-expected_usage1]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[us.amazon.nova-lite-v1:0-llm_request2-llm_response2-expected_usage2]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[cohere.command-r-plus-v1:0-llm_request3-llm_response3-None]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_llm[meta.llama3-8b-instruct-v1:0-llm_request4-llm_response4-expected_usage4]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_embeddings
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_capture_exception
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_stream
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config0]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config1]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request0-response0-expected_chat_attr0-None-expected_usage0]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request1-response1-expected_chat_attr1-expected_tool_attr1-expected_usage1]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request2-response2-expected_chat_attr2-None-expected_usage2]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse[_request3-response3-expected_chat_attr3-None-expected_usage3]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_error
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_skip_unsupported_content
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream[_request0-expected_response0-expected_chat_attr0-None-expected_usage0]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream[_request1-expected_response1-expected_chat_attr1-expected_tool_attr1-expected_usage1]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_missing_usage_field]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_empty_usage_dict]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_only_input_tokens]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_only_output_tokens]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_zero_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_large_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_none_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_string_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_duplicate_keys_in_usage]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[no_usage_in_stream]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_only_in_first_chunk]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_updated_in_later_chunk]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[malformed_usage_in_chunk]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_with_none_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[same_key_in_multiple_chunks]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_missing_usage_field]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_empty_usage_dict]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_input_tokens]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_output_tokens]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_zero_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_large_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_none_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_string_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_missing_usage_field]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_empty_usage_dict]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_input_tokens]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_output_tokens]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_zero_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_large_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_none_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_string_token_values]
FAILED | MEM 146.9/755.2 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_malformed_usage_in_metadata]
============================== 50 failed in 4.28s ==============================
Reward: 0
