Skip to content

Commit 9f95eda

Browse files
committed
Rename other bedrock test files
1 parent 6df1aa5 commit 9f95eda

File tree

4 files changed

+14
-16
lines changed

4 files changed

+14
-16
lines changed

tests/external_aiobotocore/test_bedrock_chat_completion_invoke_model.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -15,23 +15,10 @@
1515
import os
1616
from io import BytesIO
1717

18-
import botocore.errorfactory
1918
import botocore.eventstream
2019
import botocore.exceptions
2120
import pytest
2221
from conftest import BOTOCORE_VERSION
23-
from external_botocore._test_bedrock_chat_completion import (
24-
chat_completion_expected_events,
25-
chat_completion_expected_malformed_request_body_events,
26-
chat_completion_expected_malformed_response_body_events,
27-
chat_completion_expected_malformed_response_streaming_body_events,
28-
chat_completion_expected_malformed_response_streaming_chunk_events,
29-
chat_completion_expected_streaming_error_events,
30-
chat_completion_invalid_access_key_error_events,
31-
chat_completion_invalid_model_error_events,
32-
chat_completion_payload_templates,
33-
chat_completion_streaming_expected_events,
34-
)
3522
from testing_support.fixtures import override_llm_token_callback_settings, reset_core_stats_engine, validate_attributes
3623
from testing_support.ml_testing_utils import (
3724
add_token_count_to_events,
@@ -54,6 +41,18 @@
5441
from newrelic.api.transaction import add_custom_attribute
5542
from newrelic.common.object_names import callable_name
5643
from newrelic.hooks.external_botocore import MODEL_EXTRACTORS
44+
from tests.external_botocore._test_bedrock_chat_completion_invoke_model import (
45+
chat_completion_expected_events,
46+
chat_completion_expected_malformed_request_body_events,
47+
chat_completion_expected_malformed_response_body_events,
48+
chat_completion_expected_malformed_response_streaming_body_events,
49+
chat_completion_expected_malformed_response_streaming_chunk_events,
50+
chat_completion_expected_streaming_error_events,
51+
chat_completion_invalid_access_key_error_events,
52+
chat_completion_invalid_model_error_events,
53+
chat_completion_payload_templates,
54+
chat_completion_streaming_expected_events,
55+
)
5756

5857

5958
@pytest.fixture(scope="session", params=[False, True], ids=["ResponseStandard", "ResponseStreaming"])

tests/external_botocore/test_bedrock_chat_completion_invoke_model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,10 @@
1616
from io import BytesIO
1717

1818
import boto3
19-
import botocore.errorfactory
2019
import botocore.eventstream
2120
import botocore.exceptions
2221
import pytest
23-
from _test_bedrock_chat_completion import (
22+
from _test_bedrock_chat_completion_invoke_model import (
2423
chat_completion_expected_events,
2524
chat_completion_expected_malformed_request_body_events,
2625
chat_completion_expected_malformed_response_body_events,

tests/external_botocore/test_bedrock_chat_completion_via_langchain.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# limitations under the License.
1414

1515
import pytest
16-
from _test_bedrock_chat_completion import (
16+
from _test_bedrock_chat_completion_invoke_model import (
1717
chat_completion_langchain_expected_events,
1818
chat_completion_langchain_expected_streaming_events,
1919
)

0 commit comments

Comments
 (0)