Update llama_types

abetlen · abetlen · commit 428b64ec5012 · 2023-09-20T03:30:04.000-04:00
diff --git a/llama_cpp/llama_types.py b/llama_cpp/llama_types.py
@@ -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):
