Module: Rager::Types

Extended by:

T::Sig

Defined in:

lib/rager/types.rb

Constant Summary

ChatInput =

T.type_alias { T::Array[Rager::Chat::Message] }

EmbedInput =

T.type_alias { T::Array[String] }

ImageGenInput =

T.type_alias { String }

MeshGenInput =

T.type_alias { String }

RerankInput =

T.type_alias { {query: String, documents: T.any(T::Array[String], T::Array[T::Hash[Symbol, String]])} }

SearchInput =

T.type_alias { String }

TemplateInput =

T.type_alias { {template: String, variables: T::Hash[Symbol, T.untyped]} }

Input =

T.type_alias {
  T.any(
    ChatInput,
    EmbedInput,
    ImageGenInput,
    MeshGenInput,
    RerankInput,
    SearchInput,
    TemplateInput,
    Rager::Result
  )
}

ChatStream =

T.type_alias { T::Enumerator[Rager::Chat::MessageDelta] }

Stream =

T.type_alias { ChatStream }

ChatBuffer =

T.type_alias { T::Array[Rager::Chat::MessageDelta] }

Buffer =

T.type_alias { ChatBuffer }

ChatNonStream =

T.type_alias { T.any(String, T::Array[String]) }

ChatOutput =

T.type_alias { T.any(ChatNonStream, ChatStream) }

EmbedOutput =

T.type_alias { T::Array[T::Array[Float]] }

ImageGenOutput =

T.type_alias { String }

MeshGenOutput =

T.type_alias { String }

RerankOutput =

T.type_alias { Rager::Rerank::Output }

SearchOutput =

T.type_alias { Rager::Search::Output }

TemplateOutput =

T.type_alias { String }

Output =

T.type_alias {
  T.any(
    ChatOutput,
    EmbedOutput,
    ImageGenOutput,
    MeshGenOutput,
    RerankOutput,
    SearchOutput,
    TemplateOutput
  )
}

ChatNonStreamOutput =

T.type_alias { ChatNonStream }

NonStreamOutput =

T.type_alias {
  T.any(
    ChatNonStreamOutput,
    EmbedOutput,
    ImageGenOutput,
    MeshGenOutput,
    RerankOutput,
    SearchOutput,
    TemplateOutput
  )
}