Added safe_prompt to the API

Author: sam-ulrich1

chat.go

@@ -28,13 +28,15 @@ type ChatRequestParams struct {
 	TopP        float64 `json:"top_p"`       // An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or Temperature but not both.
 	RandomSeed  int     `json:"random_seed"`
 	MaxTokens   int     `json:"max_tokens"`
+	SafePrompt  bool    `json:"safe_prompt"` // Adds a Mistral defined safety message to the system prompt to enforce guardrailing
 }
 
 var DefaultChatRequestParams = ChatRequestParams{
 	Temperature: 1,
 	TopP:        1,
 	RandomSeed:  42069,
 	MaxTokens:   4000,
+	SafePrompt:  false,
 }
 
 // ChatMessage represents a single message in a chat.
@@ -97,6 +99,7 @@ func (c *MistralClient) Chat(model string, messages []ChatMessage, params *ChatR
 		"max_tokens":  params.MaxTokens,
 		"top_p":       params.TopP,
 		"random_seed": params.RandomSeed,
+		"safe_prompt": params.SafePrompt,
 	}
 
 	response, err := c.request(http.MethodPost, requestData, "v1/chat/completions", false, nil)
@@ -133,6 +136,7 @@ func (c *MistralClient) ChatStream(model string, messages []ChatMessage, params
 		"max_tokens":  params.MaxTokens,
 		"top_p":       params.TopP,
 		"random_seed": params.RandomSeed,
+		"safe_prompt": params.SafePrompt,
 		"stream":      true,
 	}