|
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