Skip to content

Commit 600f499

Browse files
chore: use sorbet union aliases where available
1 parent 15a2b2b commit 600f499

File tree

88 files changed

+784
-1728
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

88 files changed

+784
-1728
lines changed

rbi/openai/models/audio/speech_create_params.rbi

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -18,18 +18,14 @@ module OpenAI
1818

1919
# One of the available [TTS models](https://platform.openai.com/docs/models#tts):
2020
# `tts-1`, `tts-1-hd` or `gpt-4o-mini-tts`.
21-
sig { returns(T.any(String, OpenAI::Audio::SpeechModel::OrSymbol)) }
21+
sig { returns(OpenAI::Audio::SpeechCreateParams::Model::Variants) }
2222
attr_accessor :model
2323

2424
# The voice to use when generating the audio. Supported voices are `alloy`, `ash`,
2525
# `ballad`, `coral`, `echo`, `fable`, `onyx`, `nova`, `sage`, `shimmer`, and
2626
# `verse`. Previews of the voices are available in the
2727
# [Text to speech guide](https://platform.openai.com/docs/guides/text-to-speech#voice-options).
28-
sig do
29-
returns(
30-
T.any(String, OpenAI::Audio::SpeechCreateParams::Voice::OrSymbol)
31-
)
32-
end
28+
sig { returns(OpenAI::Audio::SpeechCreateParams::Voice::Variants) }
3329
attr_accessor :voice
3430

3531
# Control the voice of your generated audio with additional instructions. Does not
@@ -70,9 +66,8 @@ module OpenAI
7066
sig do
7167
params(
7268
input: String,
73-
model: T.any(String, OpenAI::Audio::SpeechModel::OrSymbol),
74-
voice:
75-
T.any(String, OpenAI::Audio::SpeechCreateParams::Voice::OrSymbol),
69+
model: OpenAI::Audio::SpeechCreateParams::Model::Variants,
70+
voice: OpenAI::Audio::SpeechCreateParams::Voice::Variants,
7671
instructions: String,
7772
response_format:
7873
OpenAI::Audio::SpeechCreateParams::ResponseFormat::OrSymbol,
@@ -108,12 +103,8 @@ module OpenAI
108103
override.returns(
109104
{
110105
input: String,
111-
model: T.any(String, OpenAI::Audio::SpeechModel::OrSymbol),
112-
voice:
113-
T.any(
114-
String,
115-
OpenAI::Audio::SpeechCreateParams::Voice::OrSymbol
116-
),
106+
model: OpenAI::Audio::SpeechCreateParams::Model::Variants,
107+
voice: OpenAI::Audio::SpeechCreateParams::Voice::Variants,
117108
instructions: String,
118109
response_format:
119110
OpenAI::Audio::SpeechCreateParams::ResponseFormat::OrSymbol,

rbi/openai/models/audio/transcription_create_params.rbi

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,9 @@ module OpenAI
2323
# ID of the model to use. The options are `gpt-4o-transcribe`,
2424
# `gpt-4o-mini-transcribe`, and `whisper-1` (which is powered by our open source
2525
# Whisper V2 model).
26-
sig { returns(T.any(String, OpenAI::AudioModel::OrSymbol)) }
26+
sig do
27+
returns(OpenAI::Audio::TranscriptionCreateParams::Model::Variants)
28+
end
2729
attr_accessor :model
2830

2931
# Controls how the audio is cut into chunks. When set to `"auto"`, the server
@@ -131,7 +133,7 @@ module OpenAI
131133
sig do
132134
params(
133135
file: T.any(Pathname, StringIO, IO, OpenAI::FilePart),
134-
model: T.any(String, OpenAI::AudioModel::OrSymbol),
136+
model: OpenAI::Audio::TranscriptionCreateParams::Model::Variants,
135137
chunking_strategy:
136138
T.nilable(
137139
T.any(
@@ -203,7 +205,7 @@ module OpenAI
203205
override.returns(
204206
{
205207
file: T.any(Pathname, StringIO, IO, OpenAI::FilePart),
206-
model: T.any(String, OpenAI::AudioModel::OrSymbol),
208+
model: OpenAI::Audio::TranscriptionCreateParams::Model::Variants,
207209
chunking_strategy:
208210
T.nilable(
209211
T.any(

rbi/openai/models/audio/translation_create_params.rbi

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ module OpenAI
2222

2323
# ID of the model to use. Only `whisper-1` (which is powered by our open source
2424
# Whisper V2 model) is currently available.
25-
sig { returns(T.any(String, OpenAI::AudioModel::OrSymbol)) }
25+
sig { returns(OpenAI::Audio::TranslationCreateParams::Model::Variants) }
2626
attr_accessor :model
2727

2828
# An optional text to guide the model's style or continue a previous audio
@@ -68,7 +68,7 @@ module OpenAI
6868
sig do
6969
params(
7070
file: T.any(Pathname, StringIO, IO, OpenAI::FilePart),
71-
model: T.any(String, OpenAI::AudioModel::OrSymbol),
71+
model: OpenAI::Audio::TranslationCreateParams::Model::Variants,
7272
prompt: String,
7373
response_format:
7474
OpenAI::Audio::TranslationCreateParams::ResponseFormat::OrSymbol,
@@ -105,7 +105,7 @@ module OpenAI
105105
override.returns(
106106
{
107107
file: T.any(Pathname, StringIO, IO, OpenAI::FilePart),
108-
model: T.any(String, OpenAI::AudioModel::OrSymbol),
108+
model: OpenAI::Audio::TranslationCreateParams::Model::Variants,
109109
prompt: String,
110110
response_format:
111111
OpenAI::Audio::TranslationCreateParams::ResponseFormat::OrSymbol,

rbi/openai/models/beta/assistant.rbi

Lines changed: 4 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -54,17 +54,7 @@ module OpenAI
5454
# A list of tool enabled on the assistant. There can be a maximum of 128 tools per
5555
# assistant. Tools can be of types `code_interpreter`, `file_search`, or
5656
# `function`.
57-
sig do
58-
returns(
59-
T::Array[
60-
T.any(
61-
OpenAI::Beta::CodeInterpreterTool,
62-
OpenAI::Beta::FileSearchTool,
63-
OpenAI::Beta::FunctionTool
64-
)
65-
]
66-
)
67-
end
57+
sig { returns(T::Array[OpenAI::Beta::AssistantTool::Variants]) }
6858
attr_accessor :tools
6959

7060
# Specifies the format that the model must output. Compatible with
@@ -89,14 +79,7 @@ module OpenAI
8979
# max context length.
9080
sig do
9181
returns(
92-
T.nilable(
93-
T.any(
94-
Symbol,
95-
OpenAI::ResponseFormatText,
96-
OpenAI::ResponseFormatJSONObject,
97-
OpenAI::ResponseFormatJSONSchema
98-
)
99-
)
82+
T.nilable(OpenAI::Beta::AssistantResponseFormatOption::Variants)
10083
)
10184
end
10285
attr_accessor :response_format
@@ -245,22 +228,10 @@ module OpenAI
245228
model: String,
246229
name: T.nilable(String),
247230
object: Symbol,
248-
tools:
249-
T::Array[
250-
T.any(
251-
OpenAI::Beta::CodeInterpreterTool,
252-
OpenAI::Beta::FileSearchTool,
253-
OpenAI::Beta::FunctionTool
254-
)
255-
],
231+
tools: T::Array[OpenAI::Beta::AssistantTool::Variants],
256232
response_format:
257233
T.nilable(
258-
T.any(
259-
Symbol,
260-
OpenAI::ResponseFormatText,
261-
OpenAI::ResponseFormatJSONObject,
262-
OpenAI::ResponseFormatJSONSchema
263-
)
234+
OpenAI::Beta::AssistantResponseFormatOption::Variants
264235
),
265236
temperature: T.nilable(Float),
266237
tool_resources: T.nilable(OpenAI::Beta::Assistant::ToolResources),

rbi/openai/models/beta/assistant_create_params.rbi

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ module OpenAI
2020
# see all of your available models, or see our
2121
# [Model overview](https://platform.openai.com/docs/models) for descriptions of
2222
# them.
23-
sig { returns(T.any(String, OpenAI::ChatModel::OrSymbol)) }
23+
sig { returns(OpenAI::Beta::AssistantCreateParams::Model::Variants) }
2424
attr_accessor :model
2525

2626
# The description of the assistant. The maximum length is 512 characters.
@@ -155,7 +155,7 @@ module OpenAI
155155

156156
sig do
157157
params(
158-
model: T.any(String, OpenAI::ChatModel::OrSymbol),
158+
model: OpenAI::Beta::AssistantCreateParams::Model::Variants,
159159
description: T.nilable(String),
160160
instructions: T.nilable(String),
161161
metadata: T.nilable(T::Hash[Symbol, String]),
@@ -262,7 +262,7 @@ module OpenAI
262262
sig do
263263
override.returns(
264264
{
265-
model: T.any(String, OpenAI::ChatModel::OrSymbol),
265+
model: OpenAI::Beta::AssistantCreateParams::Model::Variants,
266266
description: T.nilable(String),
267267
instructions: T.nilable(String),
268268
metadata: T.nilable(T::Hash[Symbol, String]),

rbi/openai/models/beta/assistant_update_params.rbi

Lines changed: 4 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -40,23 +40,14 @@ module OpenAI
4040
# them.
4141
sig do
4242
returns(
43-
T.nilable(
44-
T.any(
45-
String,
46-
OpenAI::Beta::AssistantUpdateParams::Model::OrSymbol
47-
)
48-
)
43+
T.nilable(OpenAI::Beta::AssistantUpdateParams::Model::Variants)
4944
)
5045
end
5146
attr_reader :model
5247

5348
sig do
5449
params(
55-
model:
56-
T.any(
57-
String,
58-
OpenAI::Beta::AssistantUpdateParams::Model::OrSymbol
59-
)
50+
model: OpenAI::Beta::AssistantUpdateParams::Model::Variants
6051
).void
6152
end
6253
attr_writer :model
@@ -178,11 +169,7 @@ module OpenAI
178169
description: T.nilable(String),
179170
instructions: T.nilable(String),
180171
metadata: T.nilable(T::Hash[Symbol, String]),
181-
model:
182-
T.any(
183-
String,
184-
OpenAI::Beta::AssistantUpdateParams::Model::OrSymbol
185-
),
172+
model: OpenAI::Beta::AssistantUpdateParams::Model::Variants,
186173
name: T.nilable(String),
187174
reasoning_effort: T.nilable(OpenAI::ReasoningEffort::OrSymbol),
188175
response_format:
@@ -289,11 +276,7 @@ module OpenAI
289276
description: T.nilable(String),
290277
instructions: T.nilable(String),
291278
metadata: T.nilable(T::Hash[Symbol, String]),
292-
model:
293-
T.any(
294-
String,
295-
OpenAI::Beta::AssistantUpdateParams::Model::OrSymbol
296-
),
279+
model: OpenAI::Beta::AssistantUpdateParams::Model::Variants,
297280
name: T.nilable(String),
298281
reasoning_effort: T.nilable(OpenAI::ReasoningEffort::OrSymbol),
299282
response_format:

rbi/openai/models/beta/thread_create_and_run_params.rbi

Lines changed: 17 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,11 @@ module OpenAI
5555
# be used to execute this run. If a value is provided here, it will override the
5656
# model associated with the assistant. If not, the model associated with the
5757
# assistant will be used.
58-
sig { returns(T.nilable(T.any(String, OpenAI::ChatModel::OrSymbol))) }
58+
sig do
59+
returns(
60+
T.nilable(OpenAI::Beta::ThreadCreateAndRunParams::Model::Variants)
61+
)
62+
end
5963
attr_accessor :model
6064

6165
# Whether to enable
@@ -214,7 +218,10 @@ module OpenAI
214218
max_completion_tokens: T.nilable(Integer),
215219
max_prompt_tokens: T.nilable(Integer),
216220
metadata: T.nilable(T::Hash[Symbol, String]),
217-
model: T.nilable(T.any(String, OpenAI::ChatModel::OrSymbol)),
221+
model:
222+
T.nilable(
223+
OpenAI::Beta::ThreadCreateAndRunParams::Model::Variants
224+
),
218225
parallel_tool_calls: T::Boolean,
219226
response_format:
220227
T.nilable(
@@ -357,7 +364,10 @@ module OpenAI
357364
max_completion_tokens: T.nilable(Integer),
358365
max_prompt_tokens: T.nilable(Integer),
359366
metadata: T.nilable(T::Hash[Symbol, String]),
360-
model: T.nilable(T.any(String, OpenAI::ChatModel::OrSymbol)),
367+
model:
368+
T.nilable(
369+
OpenAI::Beta::ThreadCreateAndRunParams::Model::Variants
370+
),
361371
parallel_tool_calls: T::Boolean,
362372
response_format:
363373
T.nilable(
@@ -550,16 +560,7 @@ module OpenAI
550560
# The text contents of the message.
551561
sig do
552562
returns(
553-
T.any(
554-
String,
555-
T::Array[
556-
T.any(
557-
OpenAI::Beta::Threads::ImageFileContentBlock,
558-
OpenAI::Beta::Threads::ImageURLContentBlock,
559-
OpenAI::Beta::Threads::TextContentBlockParam
560-
)
561-
]
562-
)
563+
OpenAI::Beta::ThreadCreateAndRunParams::Thread::Message::Content::Variants
563564
)
564565
end
565566
attr_accessor :content
@@ -601,16 +602,7 @@ module OpenAI
601602
sig do
602603
params(
603604
content:
604-
T.any(
605-
String,
606-
T::Array[
607-
T.any(
608-
OpenAI::Beta::Threads::ImageFileContentBlock::OrHash,
609-
OpenAI::Beta::Threads::ImageURLContentBlock::OrHash,
610-
OpenAI::Beta::Threads::TextContentBlockParam::OrHash
611-
)
612-
]
613-
),
605+
OpenAI::Beta::ThreadCreateAndRunParams::Thread::Message::Content::Variants,
614606
role:
615607
OpenAI::Beta::ThreadCreateAndRunParams::Thread::Message::Role::OrSymbol,
616608
attachments:
@@ -648,16 +640,7 @@ module OpenAI
648640
override.returns(
649641
{
650642
content:
651-
T.any(
652-
String,
653-
T::Array[
654-
T.any(
655-
OpenAI::Beta::Threads::ImageFileContentBlock,
656-
OpenAI::Beta::Threads::ImageURLContentBlock,
657-
OpenAI::Beta::Threads::TextContentBlockParam
658-
)
659-
]
660-
),
643+
OpenAI::Beta::ThreadCreateAndRunParams::Thread::Message::Content::Variants,
661644
role:
662645
OpenAI::Beta::ThreadCreateAndRunParams::Thread::Message::Role::OrSymbol,
663646
attachments:
@@ -682,11 +665,7 @@ module OpenAI
682665
T.any(
683666
String,
684667
T::Array[
685-
T.any(
686-
OpenAI::Beta::Threads::ImageFileContentBlock,
687-
OpenAI::Beta::Threads::ImageURLContentBlock,
688-
OpenAI::Beta::Threads::TextContentBlockParam
689-
)
668+
OpenAI::Beta::Threads::MessageContentPartParam::Variants
690669
]
691670
)
692671
end

0 commit comments

Comments
 (0)