|
44 | 44 | from ...types.responses.parsed_response import ParsedResponse |
45 | 45 | from ...lib.streaming.responses._responses import ResponseStreamManager, AsyncResponseStreamManager |
46 | 46 | from ...types.responses.response_includable import ResponseIncludable |
| 47 | +from ...types.shared_params.responses_model import ResponsesModel |
47 | 48 | from ...types.responses.response_input_param import ResponseInputParam |
48 | 49 | from ...types.responses.response_stream_event import ResponseStreamEvent |
49 | 50 | from ...types.responses.response_text_config_param import ResponseTextConfigParam |
@@ -80,7 +81,7 @@ def create( |
80 | 81 | self, |
81 | 82 | *, |
82 | 83 | input: Union[str, ResponseInputParam], |
83 | | - model: Union[str, ChatModel], |
| 84 | + model: ResponsesModel, |
84 | 85 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
85 | 86 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
86 | 87 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
@@ -245,7 +246,7 @@ def create( |
245 | 246 | self, |
246 | 247 | *, |
247 | 248 | input: Union[str, ResponseInputParam], |
248 | | - model: Union[str, ChatModel], |
| 249 | + model: ResponsesModel, |
249 | 250 | stream: Literal[True], |
250 | 251 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
251 | 252 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -410,7 +411,7 @@ def create( |
410 | 411 | self, |
411 | 412 | *, |
412 | 413 | input: Union[str, ResponseInputParam], |
413 | | - model: Union[str, ChatModel], |
| 414 | + model: ResponsesModel, |
414 | 415 | stream: bool, |
415 | 416 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
416 | 417 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -575,7 +576,7 @@ def create( |
575 | 576 | self, |
576 | 577 | *, |
577 | 578 | input: Union[str, ResponseInputParam], |
578 | | - model: Union[str, ChatModel], |
| 579 | + model: ResponsesModel, |
579 | 580 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
580 | 581 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
581 | 582 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
@@ -892,7 +893,7 @@ async def create( |
892 | 893 | self, |
893 | 894 | *, |
894 | 895 | input: Union[str, ResponseInputParam], |
895 | | - model: Union[str, ChatModel], |
| 896 | + model: ResponsesModel, |
896 | 897 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
897 | 898 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
898 | 899 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
@@ -1057,7 +1058,7 @@ async def create( |
1057 | 1058 | self, |
1058 | 1059 | *, |
1059 | 1060 | input: Union[str, ResponseInputParam], |
1060 | | - model: Union[str, ChatModel], |
| 1061 | + model: ResponsesModel, |
1061 | 1062 | stream: Literal[True], |
1062 | 1063 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
1063 | 1064 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -1222,7 +1223,7 @@ async def create( |
1222 | 1223 | self, |
1223 | 1224 | *, |
1224 | 1225 | input: Union[str, ResponseInputParam], |
1225 | | - model: Union[str, ChatModel], |
| 1226 | + model: ResponsesModel, |
1226 | 1227 | stream: bool, |
1227 | 1228 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
1228 | 1229 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
@@ -1387,7 +1388,7 @@ async def create( |
1387 | 1388 | self, |
1388 | 1389 | *, |
1389 | 1390 | input: Union[str, ResponseInputParam], |
1390 | | - model: Union[str, ChatModel], |
| 1391 | + model: ResponsesModel, |
1391 | 1392 | include: Optional[List[ResponseIncludable]] | NotGiven = NOT_GIVEN, |
1392 | 1393 | instructions: Optional[str] | NotGiven = NOT_GIVEN, |
1393 | 1394 | max_output_tokens: Optional[int] | NotGiven = NOT_GIVEN, |
|
0 commit comments