tuoling805 1 year ago
parent
commit
5237ea70a9
1 changed files with 11 additions and 0 deletions
  1. 11 0
      models/db.go

+ 11 - 0
models/db.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"eta/eta_api/models/aimod"
 	"eta/eta_api/models/company"
 	"eta/eta_api/models/data_manage"
 	"eta/eta_api/models/data_manage/excel"
@@ -166,6 +167,9 @@ func init() {
 	// 初始化EXCEL的表
 	initExcel()
 
+	//初始化AI
+	initAi()
+
 	// 初始化部分数据表变量(直接init会有顺序问题=_=!)
 	data_manage.InitEdbSourceVar()
 }
@@ -479,3 +483,10 @@ func initSmartReport() {
 		new(smart_report.SmartReportSaveLog), // 智能研报-保存记录表
 	)
 }
+
+func initAi() {
+	orm.RegisterModel(
+		new(aimod.AiChatTopic),
+		new(aimod.AiChat),
+	)
+}