types

type DatabaseConfig struct {
	DSN         string `` /* 138-byte string literal not displayed */
	AutoMigrate string `usage:"Auto migrate database" default:"true" env:"KNOW_AUTO_MIGRATE"`
}

type Dataset struct {
	// Dataset ID - must be a valid RFC 1123 hostname
	ID             string `json:"id" format:"hostname_rfc1123" binding:"required" example:"asst-12345"`
	EmbedDimension *int   `json:"embed_dim" example:"1536" default:"1536" swaggertype:"integer"`
}

type Ingest struct {
	Filename     *string             `json:"filename" `
	Content      string              `json:"content" binding:"required,base64"`
	FileMetadata *index.FileMetadata `json:"metadata"`
}

type IngestResponse struct {
	Documents []string `json:"documents"`
}

type OpenAIConfig struct {
	APIBase        string `usage:"OpenAI API base" default:"https://api.openai.com/v1" env:"OPENAI_BASE_URL"` // clicky-chats
	APIKey         string `usage:"OpenAI API key (not required if used with clicky-chats)" default:"sk-foo" env:"OPENAI_API_KEY"`
	EmbeddingModel string `usage:"OpenAI Embedding model" default:"text-embedding-ada-002" env:"OPENAI_EMBEDDING_MODEL"`
}

type Query struct {
	Prompt string `json:"prompt" binding:"required"`
	TopK   *int   `json:"topk" example:"5" swaggertype:"integer"`
}

type VectorDBConfig struct {
	VectorDBPath string `` /* 139-byte string literal not displayed */
}