models

type AskReq struct {
	Message string `json:"message" binding:"required"`
}

type Config struct {
	Server struct {
		Port string `mapstructure:"port"`
	} `mapstructure:"server"`

	Redis struct {
		Address  string `mapstructure:"address"`
		Password string `mapstructure:"password"`
		DB       int    `mapstructure:"db"`
	} `mapstructure:"redis"`

	SQLite struct {
		Path            string `mapstructure:"path"`
		MaxOpenConns    int    `mapstructure:"max_open_conns"`
		MaxIdleConns    int    `mapstructure:"max_idle_conns"`
		ConnMaxLifetime int    `mapstructure:"conn_max_lifetime"` // 秒
	} `mapstructure:"sqlite"`

	JWT struct {
		Secret string `mapstructure:"secret"`
	} `mapstructure:"jwt"`

	RAG struct {
		ServiceAddr string `mapstructure:"service_addr"`
	} `mapstructure:"rag"`
}

type Conversation struct {
	ID          int64     `json:"conversation_id"`
	Title       string    `json:"title"`
	Model       string    `json:"model"`
	ApiKey      string    `json:"api_key"`
	Messages    []Message `json:"messages"`
	CreatedTime int64     `json:"created_time"` // Unix 时间戳
}

type ConversationHistory struct {
	ID       int64     `json:"conversation_id"`
	Title    string    `json:"title"`
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
}

type ConversationReq struct {
	Model string `json:"model" binding:"required"`
	Title string `json:"title" binding:"required"`
}

type ConversationSummary struct {
	ID          int64  `json:"conversation_id"`
	Title       string `json:"title"`
	CreatedTime int64  `json:"created_time"`
}

type CreateConversationReq struct {
	Model  string `json:"model" binding:"required"`
	Title  string `json:"title" binding:"required"`
	ApiKey string `json:"api_key" binding:"required"`
}

type CreateConversationResp struct {
	ID          int64  `json:"conversation_id"`
	Title       string `json:"title"`
	Model       string `json:"model"`
	ApiKey      string `json:"api_key"`
	CreatedTime int64  `json:"created_time"` // Unix 时间戳
}

type CreateKnowledgeBaseRequest struct {
	Name           string `json:"kb_name" binding:"required"`
	EmbeddingModel string `json:"embedding_model,omitempty"`
}

type CreateKnowledgeBaseResponse struct {
	Success bool   `json:"success"`
	KBID    string `json:"kb_id"`
	Message string `json:"message"`
}

type DeleteDocumentResponse struct {
	Success bool   `json:"success"`
	Message string `json:"message"`
}

type DeleteKnowledgeBaseResponse struct {
	Success bool   `json:"success"`
	Message string `json:"message"`
}

type Document struct {
	ID       string `json:"doc_id"`
	KBID     string `json:"kb_id"`
	Name     string `json:"doc_name"`
	FileType string `json:"file_type"`
}

type KnowledgeBase struct {
	ID             string `json:"kb_id"`
	Name           string `json:"kb_name"`
	EmbeddingModel string `json:"embedding_model,omitempty"`
}

type ListDocumentsResponse struct {
	Success bool       `json:"success"`
	Docs    []Document `json:"docs"`
	Message string     `json:"message"`
}

type ListEmbeddingModelsResponse struct {
	Success bool     `json:"success"`
	Models  []string `json:"models"`
	Message string   `json:"message"`
}

type ListKnowledgeBasesResponse struct {
	Success bool            `json:"success"`
	KBs     []KnowledgeBase `json:"kbs"`
	Message string          `json:"message"`
}

type Message struct {
	Role      string `json:"role"`
	Content   string `json:"content"`
	MessageID int32  `json:"message_id"`
}

type RagChatRequest struct {
	ConversationID int64  `json:"conversation_id" binding:"required"`
	KBID           string `json:"kb_id" binding:"required"`
	Message        string `json:"message" binding:"required"`
	TopK           int    `json:"top_k,omitempty"`
}

type RetrieveRequest struct {
	KBID  string `json:"kb_id" binding:"required"`
	Query string `json:"query" binding:"required"`
	TopK  int    `json:"top_k,omitempty"`
}

type RetrieveResponse struct {
	Success bool             `json:"success"`
	Results []RetrieveResult `json:"results"`
	Message string           `json:"message"`
}

type RetrieveResult struct {
	Content string  `json:"content"`
	Score   float32 `json:"score"`
	DocID   string  `json:"doc_id"`
	DocName string  `json:"doc_name"`
}

type SSEChoice struct {
	Delta SSEDelta `json:"delta"`
}

type SSEDelta struct {
	Content string `json:"content"`
}

type SSEResponse struct {
	Choices []SSEChoice `json:"choices"`
}

type UploadDocumentResponse struct {
	Success bool   `json:"success"`
	DocID   string `json:"doc_id"`
	Message string `json:"message"`
}

type User struct {
	ID       int64  `json:"id"`
	Username string `json:"username"`
	Password string `json:"password"`
}