From 4f12151f4ea085417d56a307e8d1d9359fa1e91b Mon Sep 17 00:00:00 2001
From: Rehan <rehan@rehanrana.com>
Date: Sun, 21 Jul 2024 00:31:56 -0400
Subject: [PATCH] add `prefix` attribute to `ChatMessage` class

This allows the setting of a 'prefix' message when using the `ChatMessage` object.
---
 src/mistralai/models/chat_completion.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/mistralai/models/chat_completion.py b/src/mistralai/models/chat_completion.py
index c5eda5ad..5b569d7e 100644
--- a/src/mistralai/models/chat_completion.py
+++ b/src/mistralai/models/chat_completion.py
@@ -48,6 +48,7 @@ class ChatMessage(BaseModel):
     name: Optional[str] = None
     tool_calls: Optional[List[ToolCall]] = None
     tool_call_id: Optional[str] = None
+    prefix: Optional[bool] = None
 
 
 class DeltaMessage(BaseModel):