From 101e1360ba4b06b2646f53fd8592a109ec5cb0ce Mon Sep 17 00:00:00 2001 From: almerindopaixao Date: Fri, 21 Apr 2023 23:13:21 -0300 Subject: [PATCH] =?UTF-8?q?fix(aula-02):=20corrigir=20persist=C3=AAncia=20?= =?UTF-8?q?do=20ID=20da=20"InitialSystemMessage"?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- chatservice/internal/infra/repository/chat.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/chatservice/internal/infra/repository/chat.go b/chatservice/internal/infra/repository/chat.go index b3d585c..d047abf 100644 --- a/chatservice/internal/infra/repository/chat.go +++ b/chatservice/internal/infra/repository/chat.go @@ -28,7 +28,7 @@ func (r *ChatRepositoryMySQL) CreateChat(ctx context.Context, chat *entity.Chat) db.CreateChatParams{ ID: chat.ID, UserID: chat.UserID, - InitialMessageID: chat.InitialSystemMessage.Content, + InitialMessageID: chat.InitialSystemMessage.ID, Status: chat.Status, TokenUsage: int32(chat.TokenUsage), Model: chat.Config.Model.Name, @@ -128,6 +128,7 @@ func (r *ChatRepositoryMySQL) SaveChat(ctx context.Context, chat *entity.Chat) e ID: chat.ID, UserID: chat.UserID, Status: chat.Status, + InitialMessageID: chat.InitialSystemMessage.ID, TokenUsage: int32(chat.TokenUsage), Model: chat.Config.Model.Name, ModelMaxTokens: int32(chat.Config.Model.MaxTokens),