api

Documentation

Index

• type Document
• type Handler
  • func New(cfg *config.Config) (*Handler, error)
  • func (h *Handler) Attach(r chi.Router)
• type RerankRequest
• type RerankResponse
• type Result
• type Segment
• type SegmentRequest
• type SummarizeRequest

Types

type Document

type Document struct {
	Content string `json:"content,omitempty"`

	Segments []Segment `json:"segments,omitempty"`
}

type Handler

type Handler struct {
	*config.Config
	http.Handler
}

func New

func New(cfg *config.Config) (*Handler, error)

func (*Handler) Attach

func (h *Handler) Attach(r chi.Router)

type RerankRequest

type RerankRequest struct {
	Model string `json:"model"`

	Query     string   `json:"query"`
	Documents []string `json:"documents"`

	Limit *int `json:"limit,omitempty"`
}

type RerankResponse

type RerankResponse struct {
	Model string `json:"model"`

	Results []Result `json:"results"`
}

type Result

type Result struct {
	Index    int     `json:"index,omitempty"`
	Score    float64 `json:"score,omitempty"`
	Document `json:",inline"`
}

type Segment

type Segment struct {
	Text string `json:"text"`
}

type SegmentRequest

type SegmentRequest struct {
	Content string `json:"content"`

	SegmentLength  *int `json:"segment_length"`
	SegmentOverlap *int `json:"segment_overlap"`
}

type SummarizeRequest

type SummarizeRequest struct {
	Model string `json:"model"`

	Content string `json:"content"`
}