GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-20 13:30:13 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

add model field

Browse Source
This commit is contained in:
Zhang Minghan 2023-10-24 20:45:06 +08:00
parent dc96278fda
commit 205cdc1400
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
connection/database.go
View File

@ -92,6 +92,7 @@ func CreateConversationTable(db *sql.DB) {
conversation_id INT, conversation_id INT,
conversation_name VARCHAR(255), conversation_name VARCHAR(255),
data TEXT, data TEXT,
model VARCHAR(255) NOT NULL DEFAULT 'gpt-3.5-turbo',
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP, updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
UNIQUE KEY (user_id, conversation_id) UNIQUE KEY (user_id, conversation_id)
); );

12
manager/conversation/storage.go
View File

@ -15,7 +15,10 @@ func (c *Conversation) SaveConversation(db *sql.DB) bool {
} }
data := utils.ToJson(c.GetMessage()) data := utils.ToJson(c.GetMessage())
query := `INSERT INTO conversation (user_id, conversation_id, conversation_name, data) VALUES (?, ?, ?, ?) ON DUPLICATE KEY UPDATE conversation_name = VALUES(conversation_name), data = VALUES(data)` query := `
INSERT INTO conversation (user_id, conversation_id, conversation_name, data, model) VALUES (?, ?, ?, ?, ?)
ON DUPLICATE KEY UPDATE conversation_name = VALUES(conversation_name), data = VALUES(data)
`
stmt, err := db.Prepare(query) stmt, err := db.Prepare(query)
if err != nil { if err != nil {
@ -28,7 +31,7 @@ func (c *Conversation) SaveConversation(db *sql.DB) bool {
} }
}(stmt) }(stmt)
_, err = stmt.Exec(c.UserID, c.Id, c.Name, data) _, err = stmt.Exec(c.UserID, c.Id, c.Name, data, c.Model)
if err != nil { if err != nil {
return false return false
} }
@ -50,7 +53,10 @@ func LoadConversation(db *sql.DB, userId int64, conversationId int64) *Conversat
} }
var data string var data string
err := db.QueryRow("SELECT conversation_name, data FROM conversation WHERE user_id = ? AND conversation_id = ?", userId, conversationId).Scan(&conversation.Name, &data) err := db.QueryRow(`
SELECT conversation_name, model, data FROM conversation
WHERE user_id = ? AND conversation_id = ?
`, userId, conversationId).Scan(&conversation.Name, &data, &conversation.Model)
if err != nil { if err != nil {
return nil return nil
} }