From 5288d77a6f14d237e8b6be1865b456fa8f1461a4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kacper=20Kali=C5=84ski?=
 <47140412+KaQuMiQ@users.noreply.github.com>
Date: Fri, 16 Aug 2024 09:35:29 +0200
Subject: [PATCH] Update mistral data models

---
 src/draive/mistral/lmm.py    | 1 -
 src/draive/mistral/models.py | 2 --
 2 files changed, 3 deletions(-)

diff --git a/src/draive/mistral/lmm.py b/src/draive/mistral/lmm.py
index 52d7e78..bbafa68 100644
--- a/src/draive/mistral/lmm.py
+++ b/src/draive/mistral/lmm.py
@@ -163,7 +163,6 @@ def _convert_context_element(
                 tool_calls=[
                     {
                         "id": request.identifier,
-                        "type": "function",
                         "function": {
                             "name": request.tool,
                             "arguments": json.dumps(request.arguments),
diff --git a/src/draive/mistral/models.py b/src/draive/mistral/models.py
index 522d994..cc68114 100644
--- a/src/draive/mistral/models.py
+++ b/src/draive/mistral/models.py
@@ -47,7 +47,6 @@ class ChatFunctionCall(TypedDict, total=False):
 
 class ChatToolCallRequest(TypedDict, total=False):
     id: Required[str]
-    type: Required[Literal["function"]]
     function: Required[ChatFunctionCall]
 
 
@@ -65,7 +64,6 @@ class ChatFunctionCallResponse(DataModel):
 
 class ChatToolCallResponse(DataModel):
     id: str
-    type: Literal["function"]
     function: ChatFunctionCallResponse