chore: support gemini-2.5-pro

Author: Revimal

migration/collection.go

@@ -21,4 +21,5 @@ var Collection = []database.Migration{
 	sql.MigrationUser013AlterDiagnosisTable,
 	sql.MigrationUser014AlterDiagnosisTable,
 	sql.MigrationUser015AlterChatTable,
+	sql.MigrationUser016UpdateChatHistoryRow,
 }

migration/sql/016_update_chat_history_row.go

@@ -0,0 +1,11 @@
+package sql
+
+import "github.com/solutionchallenge/ondaum-server/pkg/database"
+
+const sqlUser016UpdateChatHistoryRow = `
+UPDATE chat_histories SET role = 'model' WHERE role = 'assistant';`
+
+var MigrationUser016UpdateChatHistoryRow = database.Migration{
+	Name:  "user.016.update_chat_history_row",
+	Query: sqlUser016UpdateChatHistoryRow,
+}

pkg/llm/gemini/client.go

@@ -129,7 +129,7 @@ func (client *Client) RunActionPrompt(ctx context.Context, instructionIdentifier
 
 	return llm.Message{
 		ID:      uuid.New().String(),
-		Role:    llm.RoleAssistant,
+		Role:    llm.RoleModel,
 		Content: response.Text(),
 		Metadata: map[string]any{
 			"feedbacks": feedbacks,

pkg/llm/gemini/conversation.go

@@ -64,7 +64,7 @@ func (conversation *Conversation) Request(ctx context.Context, request llm.Messa
 	}
 	message := llm.Message{
 		ID:      uuid.New().String(),
-		Role:    llm.RoleAssistant,
+		Role:    llm.RoleModel,
 		Content: response.Text(),
 		Metadata: map[string]any{
 			"feedbacks": feedbacks,

pkg/llm/history.go

@@ -2,14 +2,6 @@ package llm
 
 import "context"
 
-type HistoryRole string
-
-const (
-	HistoryRoleUser      HistoryRole = "user"
-	HistoryRoleSystem    HistoryRole = "system"
-	HistoryRoleAssistant HistoryRole = "assistant"
-)
-
 type HistoryManager interface {
 	Add(ctx context.Context, messages ...Message)
 	Get(ctx context.Context, conversationID string) []Message

pkg/llm/type.go

@@ -11,8 +11,8 @@ type Statistics struct {
 type Role string
 
 const (
-	RoleUser      Role = "user"
-	RoleAssistant Role = "assistant"
+	RoleUser  Role = "user"
+	RoleModel Role = "model"
 )
 
 type Message struct {