Commit 30194f19
Changed files (3)
src
openai
_utils
src/openai/_utils/_reflection.py
@@ -34,7 +34,7 @@ def assert_signatures_in_sync(
if custom_param.annotation != source_param.annotation:
errors.append(
- f"types for the `{name}` param are do not match; source={repr(source_param.annotation)} checking={repr(source_param.annotation)}"
+ f"types for the `{name}` param are do not match; source={repr(source_param.annotation)} checking={repr(custom_param.annotation)}"
)
continue
tests/test_client.py
@@ -17,6 +17,7 @@ from respx import MockRouter
from pydantic import ValidationError
from openai import OpenAI, AsyncOpenAI, APIResponseValidationError
+from openai._types import Omit
from openai._models import BaseModel, FinalRequestOptions
from openai._constants import RAW_RESPONSE_HEADER
from openai._streaming import Stream, AsyncStream
@@ -328,7 +329,8 @@ class TestOpenAI:
assert request.headers.get("Authorization") == f"Bearer {api_key}"
with pytest.raises(OpenAIError):
- client2 = OpenAI(base_url=base_url, api_key=None, _strict_response_validation=True)
+ with update_env(**{"OPENAI_API_KEY": Omit()}):
+ client2 = OpenAI(base_url=base_url, api_key=None, _strict_response_validation=True)
_ = client2
def test_default_query_option(self) -> None:
@@ -1103,7 +1105,8 @@ class TestAsyncOpenAI:
assert request.headers.get("Authorization") == f"Bearer {api_key}"
with pytest.raises(OpenAIError):
- client2 = AsyncOpenAI(base_url=base_url, api_key=None, _strict_response_validation=True)
+ with update_env(**{"OPENAI_API_KEY": Omit()}):
+ client2 = AsyncOpenAI(base_url=base_url, api_key=None, _strict_response_validation=True)
_ = client2
def test_default_query_option(self) -> None:
tests/utils.py
@@ -8,7 +8,7 @@ from typing import Any, TypeVar, Iterator, cast
from datetime import date, datetime
from typing_extensions import Literal, get_args, get_origin, assert_type
-from openai._types import NoneType
+from openai._types import Omit, NoneType
from openai._utils import (
is_dict,
is_list,
@@ -139,11 +139,15 @@ def _assert_list_type(type_: type[object], value: object) -> None:
@contextlib.contextmanager
-def update_env(**new_env: str) -> Iterator[None]:
+def update_env(**new_env: str | Omit) -> Iterator[None]:
old = os.environ.copy()
try:
- os.environ.update(new_env)
+ for name, value in new_env.items():
+ if isinstance(value, Omit):
+ os.environ.pop(name, None)
+ else:
+ os.environ[name] = value
yield None
finally: