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=cde05cb0964fbc6546e46a3c3c18ba56cbeb7ddb82827c93be7626f82cf5e10c
  Stored in directory: /tmp/pip-ephem-wheel-cache-6klp5j5g/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 244.3/755.1 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 243.6/755.1 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 243.0/755.1 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 243.0/755.1 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 243.5/755.1 GB | DISK 2.1/10.0 GB [ 10%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_embeddings FAILED | MEM 244.0/755.1 GB | DISK 2.1/10.0 GB [ 12%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_capture_exception FAILED | MEM 244.0/755.1 GB | DISK 2.1/10.0 GB [ 14%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_stream FAILED | MEM 244.3/755.1 GB | DISK 2.1/10.0 GB [ 16%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config0] FAILED | MEM 244.2/755.1 GB | DISK 2.1/10.0 GB [ 18%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config1] FAILED | MEM 243.4/755.1 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 243.5/755.1 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 243.9/755.1 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 244.5/755.1 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 244.6/755.1 GB | DISK 2.1/10.0 GB [ 28%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_error FAILED | MEM 245.1/755.1 GB | DISK 2.1/10.0 GB [ 30%]
tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_skip_unsupported_content FAILED | MEM 246.0/755.1 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 247.2/755.1 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 248.4/755.1 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 248.5/755.1 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 248.8/755.1 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 249.1/755.1 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 249.9/755.1 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 250.1/755.1 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 245.9/755.1 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 244.3/755.1 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 244.3/755.1 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 244.4/755.1 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 244.5/755.1 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 243.4/755.1 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 240.5/755.1 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 239.4/755.1 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 239.0/755.1 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 239.2/755.1 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 239.3/755.1 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 239.3/755.1 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 239.4/755.1 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 239.3/755.1 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 239.3/755.1 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 239.1/755.1 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 239.4/755.1 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 239.6/755.1 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 239.5/755.1 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 239.3/755.1 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 239.5/755.1 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 239.5/755.1 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 240.0/755.1 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 240.4/755.1 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 241.0/755.1 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 241.7/755.1 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 238.6/755.1 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:211: in test_bedrock_autolog_invoke_model_llm
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0f8195d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0feb8790>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0f8195d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b102f3d80>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0f8196d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b102f0860>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:32 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:34 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:211: in test_bedrock_autolog_invoke_model_llm
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0f70c650>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0f70c450>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0f70c650>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f5747c0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0f70c810>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b102f1120>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:34 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:36 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:211: in test_bedrock_autolog_invoke_model_llm
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0fd74190>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0fd74210>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0fd74190>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f575260>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b10248b50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f574c20>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:36 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:38 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:211: in test_bedrock_autolog_invoke_model_llm
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0fd77550>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0fd76d50>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0fd77550>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f575d00>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0f55ca90>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f575760>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:38 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:40 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:211: in test_bedrock_autolog_invoke_model_llm
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0f6a1750>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0f6a1650>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0f6a1750>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f574a40>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0f6a18d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f574cc0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:40 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:42 INFO mlflow.tracking.fluent: Active model is cleared
_________________ test_bedrock_autolog_invoke_model_embeddings _________________
tests/bedrock/test_bedrock_autolog.py:246: in test_bedrock_autolog_invoke_model_embeddings
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0f70fe50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0f70dbd0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0f70fe50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f5749a0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b100c4e50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f5759e0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:42 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:44 INFO mlflow.tracking.fluent: Active model is cleared
_____________ test_bedrock_autolog_invoke_model_capture_exception ______________
tests/bedrock/test_bedrock_autolog.py:280: in test_bedrock_autolog_invoke_model_capture_exception
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0f720b90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0f720b50>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0f720b90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f576520>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0f720d50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f574900>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:44 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:46 INFO mlflow.tracking.fluent: Active model is cleared
___________________ test_bedrock_autolog_invoke_model_stream ___________________
tests/bedrock/test_bedrock_autolog.py:320: in test_bedrock_autolog_invoke_model_stream
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0f6ce650>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0f6ce150>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0f6ce650>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0f576980>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0f6ce250>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f5753a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:46 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:48 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 0x789b10489490>
        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 0x789b0fda2500>, '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 0x789b0b3ec890>
        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 0x789b0f5ce250>
        traces     = []
mlflow/tracing/client.py:441: 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 0x789b0f5ce250>
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 0x789b0f5ce910>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:51 INFO mlflow.store.db.utils: Creating initial MLflow database tables...
2026/06/27 08:22:51 INFO mlflow.store.db.utils: Updating database tables
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade  -> 451aebb31d03, add metric step
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 451aebb31d03 -> 90e64c465722, migrate user column to tags
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 90e64c465722 -> 181f10493468, allow nulls for metric values
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 181f10493468 -> df50e92ffc5e, Add Experiment Tags Table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade df50e92ffc5e -> 7ac759974ad8, Update run tags with larger limit
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 7ac759974ad8 -> 89d4b8295536, create latest metrics table
2026-06-27 08:22:51 INFO  [89d4b8295536_create_latest_metrics_table_py] Migration complete!
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 89d4b8295536 -> 2b4d017a5e9b, add model registry tables to db
2026-06-27 08:22:51 INFO  [2b4d017a5e9b_add_model_registry_tables_to_db_py] Adding registered_models and model_versions tables to database.
2026-06-27 08:22:51 INFO  [2b4d017a5e9b_add_model_registry_tables_to_db_py] Migration complete!
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 2b4d017a5e9b -> cfd24bdc0731, Update run status constraint with killed
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade cfd24bdc0731 -> 0a8213491aaa, drop_duplicate_killed_constraint
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 0a8213491aaa -> 728d730b5ebd, add registered model tags table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 728d730b5ebd -> 27a6a02d2cf1, add model version tags table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 27a6a02d2cf1 -> 84291f40a231, add run_link to model_version
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 84291f40a231 -> a8c4a736bde6, allow nulls for run_id
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade a8c4a736bde6 -> 39d1c3be5f05, add_is_nan_constraint_for_metrics_tables_if_necessary
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 39d1c3be5f05 -> c48cb773bb87, reset_default_value_for_is_nan_in_metrics_table_for_mysql
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade c48cb773bb87 -> bd07f7e963c5, create index on run_uuid
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade bd07f7e963c5 -> 0c779009ac13, add deleted_time field to runs table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 0c779009ac13 -> cc1f77228345, change param value length to 500
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade cc1f77228345 -> 97727af70f4d, Add creation_time and last_update_time to experiments table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 97727af70f4d -> 3500859a5d39, Add Model Aliases table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 3500859a5d39 -> 7f2a7d5fae7d, add datasets inputs input_tags tables
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 7f2a7d5fae7d -> 2d6e25af4d3e, increase max param val length from 500 to 8000
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 2d6e25af4d3e -> acf3f17fdcc7, add storage location field to model versions
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade acf3f17fdcc7 -> 867495a8f9d4, add trace tables
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 867495a8f9d4 -> 5b0e9adcef9c, add cascade deletion to trace tables foreign keys
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 5b0e9adcef9c -> 4465047574b1, increase max dataset schema size
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 4465047574b1 -> f5a4f2784254, increase run tag value limit to 8000
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade f5a4f2784254 -> 0584bdc529eb, add cascading deletion to datasets from experiments
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 0584bdc529eb -> 400f98739977, add logged model tables
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 400f98739977 -> 6953534de441, add step to inputs table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 6953534de441 -> bda7b8c39065, increase_model_version_tag_value_limit
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade bda7b8c39065 -> cbc13b556ace, add V3 trace schema columns
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade cbc13b556ace -> 770bee3ae1dd, add assessments table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 770bee3ae1dd -> a1b2c3d4e5f6, add spans table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade a1b2c3d4e5f6 -> de4033877273, create entity_associations table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade de4033877273 -> 1a0cddfcaa16, Add webhooks and webhook_events tables
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 1a0cddfcaa16 -> 534353b11cbc, add scorer tables
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 534353b11cbc -> 71994744cf8e, add evaluation datasets
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 71994744cf8e -> 3da73c924c2f, add outputs to dataset record
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Running upgrade 3da73c924c2f -> bf29a5ff90ea, add jobs table
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
2026-06-27 08:22:51 INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:51 INFO mlflow.tracking.fluent: Active model is cleared
_________________ test_bedrock_autolog_trace_disabled[config1] _________________
tests/bedrock/test_bedrock_autolog.py:413: in test_bedrock_autolog_trace_disabled
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        config     = {'log_traces': False}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b129190>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b129150>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b129190>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b336f20>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b128090>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b5087c0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:53 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:763: in test_bedrock_autolog_converse
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        _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}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b054f50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b1044cb90>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b054f50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b336840>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b055010>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b336fc0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:54 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:55 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:763: in test_bedrock_autolog_converse
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        _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}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b03cb50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b03d110>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b03cb50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b335300>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b03cd50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b336020>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:55 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:57 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:763: in test_bedrock_autolog_converse
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        _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}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b03c450>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b03fad0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b03c450>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b334a40>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b03d510>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b335260>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:57 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:22:59 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:763: in test_bedrock_autolog_converse
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        _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}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0dbc8850>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0dc87910>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0dbc8850>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b102f0180>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b5754d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f577f60>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:22:59 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:01 INFO mlflow.tracking.fluent: Active model is cleared
_____________________ test_bedrock_autolog_converse_error ______________________
tests/bedrock/test_bedrock_autolog.py:788: in test_bedrock_autolog_converse_error
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b095110>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0f6cc850>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b095110>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b336660>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b096190>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b335ee0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:01 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:03 INFO mlflow.tracking.fluent: Active model is cleared
____________ test_bedrock_autolog_converse_skip_unsupported_content ____________
tests/bedrock/test_bedrock_autolog.py:809: in test_bedrock_autolog_converse_skip_unsupported_content
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b097e90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b097cd0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b097e90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b3365c0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b097dd0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b336200>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:03 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:05 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:860: in test_bedrock_autolog_converse_stream
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        _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}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b32ea50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b32e590>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b32ea50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b336020>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b32e490>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b335bc0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:05 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:07 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:860: in test_bedrock_autolog_converse_stream
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        _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}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b0e0490>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b0e3910>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b0e0490>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b3368e0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b0e1e10>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b3371a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:07 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:09 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'expected_usage': None, 'name': 'invoke_model_missing_usage_field', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b1140d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b114790>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b1140d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b335bc0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b114490>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0f5a4040>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:10 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:12 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'expected_usage': None, 'name': 'invoke_model_empty_usage_dict', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b023d90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b021c10>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b023d90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b336ca0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b0226d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b3374c0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:12 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:14 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'expected_usage': None, 'name': 'invoke_model_only_input_tokens', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b32cb10>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b32cfd0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b32cb10>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b335120>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b32cad0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b336200>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:14 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:16 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'expected_usage': None, 'name': 'invoke_model_only_output_tokens', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b01f590>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b01f690>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b01f590>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b334400>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b01c590>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b334b80>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:16 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:18 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b043110>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b040ed0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b043110>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34c540>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b041450>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b334cc0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:18 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:21 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b0e2bd0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b0e2310>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b0e2bd0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b336160>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b0e3890>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b334c20>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:21 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:23 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'expected_usage': None, 'name': 'invoke_model_none_token_values', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b103d5e90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b103d5710>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b103d5e90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b334540>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b103d7f10>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b3344a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:23 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:24 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'expected_usage': None, 'name': 'invoke_model_string_token_values', 'response': {'content': [{'text': 'Hello!', 'type': 'text'}], 'id': 'id-123', 'role': 'assistant', 'type': 'message', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b2e21d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b2e3e10>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b2e21d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34e5c0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b2e2510>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b335bc0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:24 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:25 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:1060: in test_bedrock_autolog_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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', ...}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0afc5f50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0afc4410>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0afc5f50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34eca0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0afc72d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34c400>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:25 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:27 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:1244: in test_bedrock_autolog_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b163d50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b163450>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b163d50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34ed40>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b163b50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b336a20>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:27 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:29 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:1244: in test_bedrock_autolog_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b054110>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b055ed0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b054110>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34e840>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b054a50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b335d00>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:29 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:31 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:1244: in test_bedrock_autolog_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b05da90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b05c610>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b05da90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34e980>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b05dcd0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34cb80>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:31 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:33 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:1244: in test_bedrock_autolog_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b099690>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b098e10>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b099690>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34f060>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b09b8d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34f1a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:33 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:35 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:1244: in test_bedrock_autolog_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b019c10>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b018bd0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b019c10>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34ea20>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b01a190>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34cc20>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:35 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:37 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:1244: in test_bedrock_autolog_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b042790>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b040ad0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b042790>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34fba0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b040d90>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34c540>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:37 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:39 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b019f50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b01bad0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b019f50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34c4a0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b546490>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b336020>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:39 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:41 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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': {}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b0e3e10>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b0e1850>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b0e3e10>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34f6a0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0dcc8410>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34cc20>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:41 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:42 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0afc58d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0afc5ad0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0afc58d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34e8e0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0afc4c10>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34f1a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:42 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:44 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b134c10>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b137690>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b134c10>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058400>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b137c10>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34c2c0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:44 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:46 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b329010>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b32a590>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b329010>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b34efc0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b328c10>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34f1a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:46 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:48 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b3930d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b392910>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b3930d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058860>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b3925d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34e660>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:48 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:50 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b0ebd50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b0e9b50>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b0ebd50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058540>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b0ead90>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34cb80>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:50 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:52 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:1280: in test_bedrock_autolog_converse_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}}}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b109690>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b108110>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b109690>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058a40>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b109b10>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b058180>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:52 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:55 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...}, {'messageStop': {'stopReason': 'end_turn'}}], 'expected_usage': None, 'name': 'converse_stream_missing_usage_field'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0afe6110>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0afe55d0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0afe6110>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b0591c0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0afe48d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b34f1a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:55 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:57 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b176f90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b175450>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b176f90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058d60>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b176450>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b059760>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:57 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:23:59 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...}}, {'metadata': {'usage': {'inputTokens': 10}}}], 'expected_usage': None, 'name': 'converse_stream_only_input_tokens'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b065f50>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b064c90>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b065f50>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b0589a0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b066310>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b0598a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:23:59 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:01 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...}, {'metadata': {'usage': {'outputTokens': 5}}}], 'expected_usage': None, 'name': 'converse_stream_only_output_tokens'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0affdb10>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0affe290>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0affdb10>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b059e40>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0affd5d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b0593a0>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:24:01 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:03 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b05cd90>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b05c990>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b05cd90>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058860>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b05fe90>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b059a80>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:24:03 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:05 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b00c950>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b00fb50>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b00c950>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b0598a0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b00c5d0>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b05a840>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:24:05 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:07 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...': {'inputTokens': None, 'outputTokens': None}}}], 'expected_usage': None, 'name': 'converse_stream_none_token_values'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b0e30d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b0e0110>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b0e30d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058ea0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b0e2890>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b058d60>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:24:07 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:09 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        test_case  = {'events': [{'messageStart': {'role': 'assistant'}}, {'contentBlockDelta': {'delta': {'text': 'Hello!'}}}, {'contentBl...: {'inputTokens': '10', 'outputTokens': '5'}}}], 'expected_usage': None, 'name': 'converse_stream_string_token_values'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b18fa10>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b18fcd0>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b18fa10>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b05a7a0>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b18f390>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b05a020>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:24:09 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:11 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:1345: in test_bedrock_autolog_converse_stream_token_usage_edge_cases
    client = boto3.client("bedrock-runtime", region_name="us-west-2")
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        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'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/__init__.py:93: in client
    return _get_default_session().client(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        args       = ('bedrock-runtime',)
        kwargs     = {'region_name': 'us-west-2'}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/boto3/session.py:337: in client
    return self._session.create_client(
        api_version = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        config     = None
        create_client_kwargs = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
        endpoint_url = None
        region_name = 'us-west-2'
        self       = Session(region_name=None)
        service_name = 'bedrock-runtime'
        use_ssl    = True
        verify     = None
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/context.py:123: in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
        args       = (<botocore.session.Session object at 0x789b1044e0d0>, 'bedrock-runtime')
        func       = <function Session.create_client at 0x789b10084220>
        hook       = None
        kwargs     = {'api_version': None, 'aws_access_key_id': None, 'aws_secret_access_key': None, 'aws_session_token': None, ...}
/opt/miniconda3/envs/testbed/lib/python3.11/site-packages/botocore/session.py:1031: in create_client
    client = client_creator.create_client(
        api_version = None
        auth_token = None
        aws_access_key_id = None
        aws_account_id = None
        aws_secret_access_key = None
        aws_session_token = None
        client_creator = <botocore.client.ClientCreator object at 0x789b0b1287d0>
        config     = None
        config_store = <botocore.configprovider.ConfigValueStore object at 0x789b0b129790>
        credentials = None
        default_client_config = None
        defaults_mode = 'legacy'
        endpoint_resolver = <botocore.regions.EndpointResolver object at 0x789b1023fad0>
        endpoint_url = None
        event_emitter = <botocore.hooks.EventAliaser object at 0x789b10188b10>
        exceptions_factory = <botocore.errorfactory.ClientExceptionsFactory object at 0x789b1034a590>
        ignored_credentials = None
        loader     = <botocore.loaders.Loader object at 0x789b1037e750>
        region_name = 'us-west-2'
        response_parser_factory = <botocore.parsers.ResponseParserFactory object at 0x789b1034a410>
        self       = <botocore.session.Session object at 0x789b1044e0d0>
        service_name = 'bedrock-runtime'
        use_ssl    = True
        user_agent_creator = <botocore.useragent.UserAgentString object at 0x789b103d5150>
        verify     = None
mlflow/utils/autologging_utils/safety.py:780: in safe_patch_function
    _validate_args(autologging_integration, function_name, args, kwargs)
E   TypeError: _validate_args() missing 2 required positional arguments: 'autologging_call_args' and 'autologging_call_kwargs'
        args       = (<botocore.client.ClientCreator object at 0x789b0b1287d0>,)
        autologging_integration = 'bedrock'
        call_original = <function safe_patch.<locals>.safe_patch_function.<locals>.call_original at 0x789b0b058360>
        call_state = {'called': False, 'result': None}
        destination = <class 'botocore.client.ClientCreator'>
        event_logger = <mlflow.utils.autologging_utils.events.AutologgingEventLoggerWrapper object at 0x789b0b12bf50>
        function_name = 'create_client'
        kwargs     = {'api_version': None, 'auth_token': None, 'client_config': None, 'credentials': None, ...}
        original_fn = <function ClientCreator.create_client at 0x789b0b059120>
        patch_function = <function patched_create_client at 0x789b102f0e00>
----------------------------- Captured stderr call -----------------------------
2026/06/27 08:24:11 INFO mlflow.bedrock: Enabled auto-tracing for Bedrock. Note that MLflow can only trace boto3 service clients that are created after this call. If you have already created one, please recreate the client by calling `boto3.client`.
--------------------------- Captured stderr teardown ---------------------------
2026/06/27 08:24:13 INFO mlflow.tracking.fluent: Active model is cleared
============================= slowest 10 durations =============================
3.03s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config0]
2.27s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream[_request1-expected_response1-expected_chat_attr1-expected_tool_attr1-expected_usage1]
2.23s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_empty_usage_dict]
2.05s 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]
2.01s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_token_usage_edge_cases[invoke_model_zero_token_values]
2.00s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_input_tokens]
2.00s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_error
2.00s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_stream_token_usage_edge_cases[converse_stream_only_output_tokens]
2.00s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_token_usage_edge_cases[converse_only_output_tokens]
2.00s call     tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_stream_token_usage_edge_cases[usage_with_none_values]

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

| Rank | File                                    | Duration | Tests | Min    | Max    | Avg    |
| ---- | --------------------------------------- | -------- | ----- | ------ | ------ | ------ |
| 1    | `tests/bedrock/test_bedrock_autolog.py` | 101.12s  | 50    | 1.027s | 3.055s | 2.022s |

::endgroup::

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

=========================== short test summary info ============================
FAILED | MEM 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_embeddings
FAILED | MEM 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_capture_exception
FAILED | MEM 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_invoke_model_stream
FAILED | MEM 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config0]
FAILED | MEM 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_trace_disabled[config1]
FAILED | MEM 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_error
FAILED | MEM 238.5/755.1 GB | DISK 2.1/10.0 GB tests/bedrock/test_bedrock_autolog.py::test_bedrock_autolog_converse_skip_unsupported_content
FAILED | MEM 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.5/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 238.4/755.1 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 101.29s (0:01:41) ========================
Reward: 0
