Skip to content

chore: recursively accept AnyHash for BaseModels in arrays and hashes #58

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 21, 2025
Merged

chore: recursively accept AnyHash for BaseModels in arrays and hashes #58

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions rbi/lib/openai/models/audio/transcription.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,16 +21,19 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Audio::Transcription::Logprob])
.returns(T::Array[OpenAI::Models::Audio::Transcription::Logprob])
params(_: T::Array[T.any(OpenAI::Models::Audio::Transcription::Logprob, OpenAI::Util::AnyHash)])
.returns(T::Array[T.any(OpenAI::Models::Audio::Transcription::Logprob, OpenAI::Util::AnyHash)])
end
def logprobs=(_)
end

# Represents a transcription response returned by model, based on the provided
# input.
sig do
params(text: String, logprobs: T::Array[OpenAI::Models::Audio::Transcription::Logprob])
params(
text: String,
logprobs: T::Array[T.any(OpenAI::Models::Audio::Transcription::Logprob, OpenAI::Util::AnyHash)]
)
.returns(T.attached_class)
end
def self.new(text:, logprobs: nil)
Expand Down
10 changes: 7 additions & 3 deletions rbi/lib/openai/models/audio/transcription_text_delta_event.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,12 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Audio::TranscriptionTextDeltaEvent::Logprob])
.returns(T::Array[OpenAI::Models::Audio::TranscriptionTextDeltaEvent::Logprob])
params(
_: T::Array[T.any(OpenAI::Models::Audio::TranscriptionTextDeltaEvent::Logprob, OpenAI::Util::AnyHash)]
)
.returns(
T::Array[T.any(OpenAI::Models::Audio::TranscriptionTextDeltaEvent::Logprob, OpenAI::Util::AnyHash)]
)
end
def logprobs=(_)
end
Expand All @@ -43,7 +47,7 @@ module OpenAI
sig do
params(
delta: String,
logprobs: T::Array[OpenAI::Models::Audio::TranscriptionTextDeltaEvent::Logprob],
logprobs: T::Array[T.any(OpenAI::Models::Audio::TranscriptionTextDeltaEvent::Logprob, OpenAI::Util::AnyHash)],
type: Symbol
)
.returns(T.attached_class)
Expand Down
10 changes: 7 additions & 3 deletions rbi/lib/openai/models/audio/transcription_text_done_event.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,12 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Audio::TranscriptionTextDoneEvent::Logprob])
.returns(T::Array[OpenAI::Models::Audio::TranscriptionTextDoneEvent::Logprob])
params(
_: T::Array[T.any(OpenAI::Models::Audio::TranscriptionTextDoneEvent::Logprob, OpenAI::Util::AnyHash)]
)
.returns(
T::Array[T.any(OpenAI::Models::Audio::TranscriptionTextDoneEvent::Logprob, OpenAI::Util::AnyHash)]
)
end
def logprobs=(_)
end
Expand All @@ -44,7 +48,7 @@ module OpenAI
sig do
params(
text: String,
logprobs: T::Array[OpenAI::Models::Audio::TranscriptionTextDoneEvent::Logprob],
logprobs: T::Array[T.any(OpenAI::Models::Audio::TranscriptionTextDoneEvent::Logprob, OpenAI::Util::AnyHash)],
type: Symbol
)
.returns(T.attached_class)
Expand Down
12 changes: 6 additions & 6 deletions rbi/lib/openai/models/audio/transcription_verbose.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Audio::TranscriptionSegment])
.returns(T::Array[OpenAI::Models::Audio::TranscriptionSegment])
params(_: T::Array[T.any(OpenAI::Models::Audio::TranscriptionSegment, OpenAI::Util::AnyHash)])
.returns(T::Array[T.any(OpenAI::Models::Audio::TranscriptionSegment, OpenAI::Util::AnyHash)])
end
def segments=(_)
end
Expand All @@ -49,8 +49,8 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Audio::TranscriptionWord])
.returns(T::Array[OpenAI::Models::Audio::TranscriptionWord])
params(_: T::Array[T.any(OpenAI::Models::Audio::TranscriptionWord, OpenAI::Util::AnyHash)])
.returns(T::Array[T.any(OpenAI::Models::Audio::TranscriptionWord, OpenAI::Util::AnyHash)])
end
def words=(_)
end
Expand All @@ -62,8 +62,8 @@ module OpenAI
duration: Float,
language: String,
text: String,
segments: T::Array[OpenAI::Models::Audio::TranscriptionSegment],
words: T::Array[OpenAI::Models::Audio::TranscriptionWord]
segments: T::Array[T.any(OpenAI::Models::Audio::TranscriptionSegment, OpenAI::Util::AnyHash)],
words: T::Array[T.any(OpenAI::Models::Audio::TranscriptionWord, OpenAI::Util::AnyHash)]
)
.returns(T.attached_class)
end
Expand Down
6 changes: 3 additions & 3 deletions rbi/lib/openai/models/audio/translation_verbose.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Audio::TranscriptionSegment])
.returns(T::Array[OpenAI::Models::Audio::TranscriptionSegment])
params(_: T::Array[T.any(OpenAI::Models::Audio::TranscriptionSegment, OpenAI::Util::AnyHash)])
.returns(T::Array[T.any(OpenAI::Models::Audio::TranscriptionSegment, OpenAI::Util::AnyHash)])
end
def segments=(_)
end
Expand All @@ -48,7 +48,7 @@ module OpenAI
duration: Float,
language: String,
text: String,
segments: T::Array[OpenAI::Models::Audio::TranscriptionSegment]
segments: T::Array[T.any(OpenAI::Models::Audio::TranscriptionSegment, OpenAI::Util::AnyHash)]
)
.returns(T.attached_class)
end
Expand Down
12 changes: 9 additions & 3 deletions rbi/lib/openai/models/batch.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -201,7 +201,7 @@ module OpenAI
created_at: Integer,
endpoint: String,
input_file_id: String,
status: OpenAI::Models::Batch::Status::TaggedSymbol,
status: OpenAI::Models::Batch::Status::OrSymbol,
cancelled_at: Integer,
cancelling_at: Integer,
completed_at: Integer,
Expand Down Expand Up @@ -301,7 +301,10 @@ module OpenAI
def data
end

sig { params(_: T::Array[OpenAI::Models::BatchError]).returns(T::Array[OpenAI::Models::BatchError]) }
sig do
params(_: T::Array[T.any(OpenAI::Models::BatchError, OpenAI::Util::AnyHash)])
.returns(T::Array[T.any(OpenAI::Models::BatchError, OpenAI::Util::AnyHash)])
end
def data=(_)
end

Expand All @@ -314,7 +317,10 @@ module OpenAI
def object=(_)
end

sig { params(data: T::Array[OpenAI::Models::BatchError], object: String).returns(T.attached_class) }
sig do
params(data: T::Array[T.any(OpenAI::Models::BatchError, OpenAI::Util::AnyHash)], object: String)
.returns(T.attached_class)
end
def self.new(data: nil, object: nil)
end

Expand Down
2 changes: 2 additions & 0 deletions rbi/lib/openai/models/beta/assistant.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -238,6 +238,7 @@ module OpenAI
tools: T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand All @@ -246,6 +247,7 @@ module OpenAI
T.any(
Symbol,
OpenAI::Models::ResponseFormatText,
OpenAI::Util::AnyHash,
OpenAI::Models::ResponseFormatJSONObject,
OpenAI::Models::ResponseFormatJSONSchema
)
Expand Down
32 changes: 29 additions & 3 deletions rbi/lib/openai/models/beta/assistant_create_params.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -196,6 +196,7 @@ module OpenAI
_: T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand All @@ -205,6 +206,7 @@ module OpenAI
T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand Down Expand Up @@ -239,6 +241,7 @@ module OpenAI
T.any(
Symbol,
OpenAI::Models::ResponseFormatText,
OpenAI::Util::AnyHash,
OpenAI::Models::ResponseFormatJSONObject,
OpenAI::Models::ResponseFormatJSONSchema
)
Expand All @@ -248,6 +251,7 @@ module OpenAI
tools: T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand Down Expand Up @@ -429,16 +433,35 @@ module OpenAI
end

sig do
params(_: T::Array[OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore])
.returns(T::Array[OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore])
params(
_: T::Array[
T.any(
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore,
OpenAI::Util::AnyHash
)
]
)
.returns(
T::Array[
T.any(
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore,
OpenAI::Util::AnyHash
)
]
)
end
def vector_stores=(_)
end

sig do
params(
vector_store_ids: T::Array[String],
vector_stores: T::Array[OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore]
vector_stores: T::Array[
T.any(
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore,
OpenAI::Util::AnyHash
)
]
)
.returns(T.attached_class)
end
Expand Down Expand Up @@ -477,12 +500,14 @@ module OpenAI
params(
_: T.any(
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore::ChunkingStrategy::Auto,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore::ChunkingStrategy::Static
)
)
.returns(
T.any(
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore::ChunkingStrategy::Auto,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore::ChunkingStrategy::Static
)
)
Expand Down Expand Up @@ -521,6 +546,7 @@ module OpenAI
params(
chunking_strategy: T.any(
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore::ChunkingStrategy::Auto,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::AssistantCreateParams::ToolResources::FileSearch::VectorStore::ChunkingStrategy::Static
),
file_ids: T::Array[String],
Expand Down
4 changes: 4 additions & 0 deletions rbi/lib/openai/models/beta/assistant_update_params.rbi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -196,6 +196,7 @@ module OpenAI
_: T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand All @@ -205,6 +206,7 @@ module OpenAI
T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand Down Expand Up @@ -239,6 +241,7 @@ module OpenAI
T.any(
Symbol,
OpenAI::Models::ResponseFormatText,
OpenAI::Util::AnyHash,
OpenAI::Models::ResponseFormatJSONObject,
OpenAI::Models::ResponseFormatJSONSchema
)
Expand All @@ -248,6 +251,7 @@ module OpenAI
tools: T::Array[
T.any(
OpenAI::Models::Beta::CodeInterpreterTool,
OpenAI::Util::AnyHash,
OpenAI::Models::Beta::FileSearchTool,
OpenAI::Models::Beta::FunctionTool
)
Expand Down
Loading