init: initial commit

2026-04-07 17:35:09 +08:00
commit 680ecc320f
129 changed files with 10562 additions and 0 deletions
@@ -0,0 +1,89 @@
+package models
+
+import (
+	"time"
+)
+
+// --- Global DB Models ---
+
+// LLMProvider stores LLM provider configuration
+type LLMProvider struct {
+	ID        string `gorm:"primaryKey" json:"id"`
+	Name      string `json:"name"`
+	Provider  string `json:"provider"` // Ollama, DeepSeek, OpenAI, Qwen
+	BaseURL   string `json:"url"`
+	APIKey    string `json:"key"`
+	ModelID   string `json:"model"`
+	Enabled   bool   `json:"enabled"`
+	CreatedAt time.Time
+	UpdatedAt time.Time
+}
+
+// VectorDBConfig stores Qdrant connection settings
+type VectorDBConfig struct {
+	ID       uint   `gorm:"primaryKey" json:"id"`
+	Endpoint string `json:"endpoint"`
+	APIKey   string `json:"apiKey"`
+	Status   string `json:"status"` // connected, disconnected
+}
+
+// Project represents a top-level engineering project
+type Project struct {
+	ID        string `gorm:"primaryKey" json:"id"`
+	Name      string `json:"name"`
+	Path      string `json:"path"` // project DB file path
+	CreatedAt time.Time
+	UpdatedAt time.Time
+}
+
+// --- Project-scoped DB Models ---
+
+// SourceFile represents an imported engineering document
+type SourceFile struct {
+	ID            string `gorm:"primaryKey" json:"id"`
+	ProjectID     string `gorm:"index" json:"projectId"`
+	Name          string `json:"name"`
+	Type          string `json:"type"` // pdf, cad, gis, excel, word
+	Category      string `json:"category"`
+	FilePath      string `json:"filePath"`
+	Size          string `json:"size"`
+	ParsedContent string `json:"parsedContent,omitempty"`
+	VectorStatus  string `json:"vectorStatus"` // pending, processing, done, error
+	CreatedAt     time.Time
+	UpdatedAt     time.Time
+}
+
+// ChatMessage stores conversation history per project
+type ChatMessage struct {
+	ID        uint   `gorm:"primaryKey;autoIncrement" json:"id"`
+	ProjectID string `gorm:"index" json:"projectId"`
+	Role      string `json:"role"` // user, assistant
+	Content   string `json:"content"`
+	Sources   string `json:"sources,omitempty"`   // JSON array of source names
+	Citations string `json:"citations,omitempty"` // JSON array of citation objects
+	CreatedAt time.Time
+}
+
+// TemplateChapter stores delivery template chapters per project
+type TemplateChapter struct {
+	ID           string `gorm:"primaryKey" json:"id"`
+	ProjectID    string `gorm:"index" json:"projectId"`
+	TemplateName string `json:"templateName"`
+	Title        string `json:"title"`
+	Status       string `json:"status"` // idle, loading, done
+	Progress     int    `json:"progress"`
+	Content      string `json:"content"`
+	SortOrder    int    `json:"sortOrder"`
+	CreatedAt    time.Time
+	UpdatedAt    time.Time
+}
+
+// TextChunk represents a vectorized text segment
+type TextChunk struct {
+	ID        string `gorm:"primaryKey" json:"id"`
+	ProjectID string `gorm:"index" json:"projectId"`
+	SourceID  string `gorm:"index" json:"sourceId"`
+	Content   string `json:"content"`
+	ChunkIdx  int    `json:"chunkIdx"`
+	CreatedAt time.Time
+}