Update mistral data models

miquido · Aug 16, 2024 · 5288d77 · 5288d77
1 parent 5e11a71
commit 5288d77
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git 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
@@ -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