|
28 | 28 | from ..._streaming import Stream, AsyncStream |
29 | 29 | from ..._base_client import make_request_options |
30 | 30 | from ...types.responses import response_create_params, response_retrieve_params |
31 | | -from ...types.shared.chat_model import ChatModel |
32 | 31 | from ...types.responses.response import Response |
33 | 32 | from ...types.responses.tool_param import ToolParam |
34 | 33 | from ...types.shared_params.metadata import Metadata |
35 | 34 | from ...types.shared_params.reasoning import Reasoning |
36 | 35 | from ...types.responses.response_includable import ResponseIncludable |
| 36 | +from ...types.shared_params.responses_model import ResponsesModel |
37 | 37 | from ...types.responses.response_input_param import ResponseInputParam |
38 | 38 | from ...types.responses.response_stream_event import ResponseStreamEvent |
39 | 39 | from ...types.responses.response_text_config_param import ResponseTextConfigParam |
@@ -70,7 +70,7 @@ def create( |
70 | 70 | self, |
71 | 71 | *, |
72 | 72 | input: Union[str, ResponseInputParam], |
73 | | - model: Union[str, ChatModel], |
| 73 | + model: ResponsesModel, |
74 | 74 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
75 | 75 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
76 | 76 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
@@ -235,7 +235,7 @@ def create( |
235 | 235 | self, |
236 | 236 | *, |
237 | 237 | input: Union[str, ResponseInputParam], |
238 | | - model: Union[str, ChatModel], |
| 238 | + model: ResponsesModel, |
239 | 239 | stream: Literal[True], |
240 | 240 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
241 | 241 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -400,7 +400,7 @@ def create( |
400 | 400 | self, |
401 | 401 | *, |
402 | 402 | input: Union[str, ResponseInputParam], |
403 | | - model: Union[str, ChatModel], |
| 403 | + model: ResponsesModel, |
404 | 404 | stream: bool, |
405 | 405 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
406 | 406 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -565,7 +565,7 @@ def create( |
565 | 565 | self, |
566 | 566 | *, |
567 | 567 | input: Union[str, ResponseInputParam], |
568 | | - model: Union[str, ChatModel], |
| 568 | + model: ResponsesModel, |
569 | 569 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
570 | 570 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
571 | 571 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
@@ -727,7 +727,7 @@ async def create( |
727 | 727 | self, |
728 | 728 | *, |
729 | 729 | input: Union[str, ResponseInputParam], |
730 | | - model: Union[str, ChatModel], |
| 730 | + model: ResponsesModel, |
731 | 731 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
732 | 732 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
733 | 733 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
@@ -892,7 +892,7 @@ async def create( |
892 | 892 | self, |
893 | 893 | *, |
894 | 894 | input: Union[str, ResponseInputParam], |
895 | | - model: Union[str, ChatModel], |
| 895 | + model: ResponsesModel, |
896 | 896 | stream: Literal[True], |
897 | 897 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
898 | 898 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -1057,7 +1057,7 @@ async def create( |
1057 | 1057 | self, |
1058 | 1058 | *, |
1059 | 1059 | input: Union[str, ResponseInputParam], |
1060 | | - model: Union[str, ChatModel], |
| 1060 | + model: ResponsesModel, |
1061 | 1061 | stream: bool, |
1062 | 1062 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
1063 | 1063 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -1222,7 +1222,7 @@ async def create( |
1222 | 1222 | self, |
1223 | 1223 | *, |
1224 | 1224 | input: Union[str, ResponseInputParam], |
1225 | | - model: Union[str, ChatModel], |
| 1225 | + model: ResponsesModel, |
1226 | 1226 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
1227 | 1227 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
1228 | 1228 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
|
0 commit comments