yieldthought
diff --git a/‎llama_cpp/llama_types.py
Copy file name to clipboardExpand all lines: llama_cpp/llama_types.py
+5-2Lines changed: 5 additions & 2 deletions b/‎llama_cpp/llama_types.py
Copy file name to clipboardExpand all lines: llama_cpp/llama_types.py
+5-2Lines changed: 5 additions & 2 deletions
@@ -58,9 +58,11 @@ class Completion(TypedDict):
 
 
 class ChatCompletionMessage(TypedDict):
-    role: Literal["assistant", "user", "system"]
-    content: str
+    role: Literal["assistant", "user", "system", "function"]
+    content: Optional[str]
     user: NotRequired[str]
+    name: NotRequired[str]
+    function_call: NotRequired[str]
 
 
 class ChatCompletionFunction(TypedDict):
@@ -71,6 +73,7 @@ class ChatCompletionFunction(TypedDict):
 
 class ChatCompletionFunctionCall(TypedDict):
     name: str
+    arguments: str
 
 
 class ChatCompletionChoice(TypedDict):