Module: Rager::Types

Extended by:
T::Sig
Defined in:
lib/rager/types.rb

Constant Summary collapse

ChatInput =
T.type_alias { T::Array[Rager::Chat::Message] }
EmbedInput =
T.type_alias { T::Array[String] }
ImageInput =
T.type_alias { String }
MeshInput =
T.type_alias { String }
RerankInput =
T.type_alias { Rager::Rerank::Input }
SearchInput =
T.type_alias { String }
TemplateInput =
T.type_alias { Rager::Template::Input }
Input =
T.type_alias {
  T.any(
    ChatInput,
    EmbedInput,
    ImageInput,
    MeshInput,
    RerankInput,
    SearchInput,
    TemplateInput
  )
}
ChatStreamOutput =
T.type_alias { T::Enumerator[Rager::Chat::MessageDelta] }
Stream =
T.type_alias { ChatStreamOutput }
ChatBuffer =
T.type_alias { T::Array[Rager::Chat::MessageDelta] }
Buffer =
T.type_alias { ChatBuffer }
ChatNonStreamOutput =
T.type_alias { T.any(String, T::Array[String]) }
ChatOutput =
T.type_alias { T.any(ChatNonStreamOutput, ChatStreamOutput) }
EmbedOutput =
T.type_alias { T::Array[T::Array[Float]] }
ImageOutput =
T.type_alias { String }
MeshOutput =
T.type_alias { String }
RerankOutput =
T.type_alias { T::Array[Rager::Rerank::Result] }
SearchOutput =
T.type_alias { T::Array[Rager::Search::Result] }
TemplateOutput =
T.type_alias { String }
Output =
T.type_alias {
  T.any(
    ChatOutput,
    EmbedOutput,
    ImageOutput,
    MeshOutput,
    RerankOutput,
    SearchOutput,
    TemplateOutput
  )
}
NonStreamOutput =
T.type_alias {
  T.any(
    ChatNonStreamOutput,
    EmbedOutput,
    ImageOutput,
    MeshOutput,
    RerankOutput,
    SearchOutput,
    TemplateOutput
  )
}
Options =
T.type_alias {
  T.any(
    Rager::Chat::Options,
    Rager::Embed::Options,
    Rager::Image::Options,
    Rager::Mesh::Options,
    Rager::Rerank::Options,
    Rager::Search::Options,
    Rager::Template::Options
  )
}