30 lines
925 B
Python
30 lines
925 B
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from openai import pydantic_function_tool
|
||
|
|
|
||
|
|
from instructor.providers.openai.utils import (
|
||
|
|
handle_responses_tools,
|
||
|
|
handle_responses_tools_with_inbuilt_tools,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class ResponseToolModel(BaseModel):
|
||
|
|
"""Extract a structured response for the user."""
|
||
|
|
|
||
|
|
name: str
|
||
|
|
|
||
|
|
|
||
|
|
def test_responses_tools_preserves_function_description() -> None:
|
||
|
|
expected_description = pydantic_function_tool(ResponseToolModel)["function"][
|
||
|
|
"description"
|
||
|
|
]
|
||
|
|
|
||
|
|
_, responses_tools_kwargs = handle_responses_tools(ResponseToolModel, {})
|
||
|
|
_, inbuilt_tools_kwargs = handle_responses_tools_with_inbuilt_tools(
|
||
|
|
ResponseToolModel, {}
|
||
|
|
)
|
||
|
|
|
||
|
|
assert responses_tools_kwargs["tools"][0]["description"] == expected_description
|
||
|
|
assert inbuilt_tools_kwargs["tools"][0]["description"] == expected_description
|
||
|
|
assert responses_tools_kwargs["tools"][0] == inbuilt_tools_kwargs["tools"][0]
|