diff --git a/libs/partners/openai/tests/conftest.py b/libs/partners/openai/tests/conftest.py index 575a647f359..a2edbc3292a 100644 --- a/libs/partners/openai/tests/conftest.py +++ b/libs/partners/openai/tests/conftest.py @@ -1,5 +1,5 @@ import pytest -from langchain_tests.conftest import _base_vcr_config +from langchain_tests.conftest import _base_vcr_config as _base_vcr_config _EXTRA_HEADERS = [ ("openai-organization", "PLACEHOLDER"), @@ -7,13 +7,13 @@ _EXTRA_HEADERS = [ ] -def remove_response_headers(response): +def remove_response_headers(response: dict) -> dict: response["headers"] = {} return response @pytest.fixture(scope="session") -def vcr_config(_base_vcr_config: dict) -> dict: +def vcr_config(_base_vcr_config: dict) -> dict: # noqa: F811 """ Extend the default configuration coming from langchain_tests. """