core

type Content struct {
	Title       string  `json:"title"`
	Description *string `json:"description"`
}

type Core struct {
}

type DeleteDocumentArgs struct {
	URLPrefix string `json:"url_prefix" jsonschema:"required"`
	Context   string `json:"context,omitempty"`
}

type DocumentOutput struct {
	URL         string `json:"url"`
	Title       string `json:"title"`
	Description string `json:"description"`
	Content     string `json:"content"`
	Checksum    string `json:"checksum"`
}

type GetContextArgs struct {
	Context string `json:"context,omitempty"`
}

type GetDocumentArgs struct {
	URL     string `json:"url" jsonschema:"required"`
	Context string `json:"context,omitempty"`
}

type LearnApiArgs struct {
	Api     string `json:"api" jsonschema:"required"`
	Context string `json:"context,omitempty"`
}

type ListDocumentsArgs struct {
	Limit   int    `json:"limit,omitempty"`
	Offset  int    `json:"offset,omitempty"`
	Context string `json:"context,omitempty"`
}

type SearchDatasetTopKArgs struct {
	Query     string  `json:"query" jsonschema:"required"`
	TopK      int     `json:"top_k" jsonschema:"required"`
	Threshold float64 `json:"threshold,omitempty"`
}

type SearchDocChunks struct {
	Query   string `json:"query" jsonschema:"required"`
	Context string `json:"context,omitempty"`
}

type SearchOutput struct {
	URL         string  `json:"url"`
	Title       string  `json:"title"`
	Description string  `json:"description"`
	Content     string  `json:"content"`
	Checksum    string  `json:"checksum"`
	Score       float64 `json:"score"`
}

type UpsertDocumentArgs struct {
	URL         string `json:"url" jsonschema:"required"`
	Content     string `json:"content" jsonschema:"required"`
	Title       string `json:"title,omitempty"`
	Description string `json:"description,omitempty"`
	Context     string `json:"context,omitempty"`
}