server

type ChatRequest struct {
	Prefix      string   `json:"prefix,omitempty"`
	History     []string `json:"history,omitempty"`
	Text        string   `json:"text"`
	DocumentIDs []uint64 `json:"document_ids,omitempty"`
	Documents   []any    `json:"documents,omitempty"`
}

type DocumentSearch struct {
	DocumentUpload
	DocumentID         uint64  `json:"document_id"`
	DocumentSimilarity float32 `json:"document_similarity"`
}

type DocumentUpload struct {
	Name       string `json:"name,omitempty"`
	ExternalID string `json:"external_id,omitempty"`
	Document   any    `json:"document"`
}

type FetchCategoryNamesRequest struct {
	Owner string `json:"owner"`
}

type FetchCategoryNamesResponse struct {
	CategoryNames []string `json:"category_names"`
}

type SearchRequest struct {
	Owner     string `json:"owner"`
	Category  string `json:"category"`
	Text      string `json:"text"`
	Count     uint   `json:"count"`
	Offset    uint   `json:"offset,omitempty"`
	Centroids int    `json:"centroids,omitempty"`
}

type SearchResponse struct {
	Documents []DocumentSearch `json:"documents"`
}

type Server struct {
	// contains filtered or unexported fields
}

type UploadRequest struct {
	Owner     string           `json:"owner"`
	Category  string           `json:"category"`
	Documents []DocumentUpload `json:"documents"`
}

type UploadResponse struct {
	DocumentIDs []uint64 `json:"document_ids"`
}