diff --git a/src/openai/__init__.py b/src/openai/__init__.py index fc9675a8b5..dd796a59ce 100644 --- a/src/openai/__init__.py +++ b/src/openai/__init__.py @@ -96,7 +96,7 @@ if not _t.TYPE_CHECKING: from ._utils._resources_proxy import resources as resources -from .lib import azure as _azure, pydantic_function_tool as pydantic_function_tool +from .lib import azure as _azure, pydantic_function_tool as pydantic_function_tool, to_strict_json_schema as to_strict_json_schema from .version import VERSION as VERSION from .lib.azure import AzureOpenAI as AzureOpenAI, AsyncAzureOpenAI as AsyncAzureOpenAI from .lib._old_api import * diff --git a/src/openai/lib/__init__.py b/src/openai/lib/__init__.py index 5c6cb782c0..a5ce7692da 100644 --- a/src/openai/lib/__init__.py +++ b/src/openai/lib/__init__.py @@ -1,2 +1,3 @@ from ._tools import pydantic_function_tool as pydantic_function_tool +from ._pydantic import to_strict_json_schema as to_strict_json_schema from ._parsing import ResponseFormatT as ResponseFormatT diff --git a/tests/lib/test_pydantic.py b/tests/lib/test_pydantic.py index 754a15151c..d6dba7dcc3 100644 --- a/tests/lib/test_pydantic.py +++ b/tests/lib/test_pydantic.py @@ -12,6 +12,11 @@ from .schema_types.query import Query +def test_to_strict_json_schema_public_api() -> None: + """Test that to_strict_json_schema is accessible as a public API via openai.to_strict_json_schema.""" + assert openai.to_strict_json_schema is to_strict_json_schema + + def test_most_types() -> None: if not PYDANTIC_V1: assert openai.pydantic_function_tool(Query)["function"] == snapshot(