265: creating mapper for interal role to provider role

pkg/api/schemas/chat.go

@@ -108,3 +108,20 @@ type ChatMessage struct {
 	// The content of the message.
 	Content string `json:"content" validate:"required"`
 }
+
+func MapToProviderRole(provider string, role Role) Role {
+	switch provider {
+	case "cohere":
+		switch role {
+		case RoleAssistant:
+			return "CHATBOT"
+		case RoleSystem:
+			return "SYSTEM"
+		case RoleUser:
+			return "USER"
+		}
+	case "openai":
+		return role
+	}
+	return role
+}

pkg/providers/cohere/chat.go

@@ -127,7 +127,7 @@ func (c *Client) doChatRequest(ctx context.Context, payload *ChatRequest) (*sche
 				"responseId":   cohereCompletion.ResponseID,
 			},
 			Message: schemas.ChatMessage{
-				Role:    schemas.RoleAssistant,
+				Role:    schemas.MapToProviderRole(providerName, schemas.RoleAssistant),
 				Content: cohereCompletion.Text,
 			},
 			TokenUsage: schemas.TokenUsage{