Update Message annotations to support tool calls (#227)

ollama · Jul 29, 2024 · 8b694bb · 8b694bb
1 parent cbf0887
commit 8b694bb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ollama/_types.py b/ollama/_types.py
@@ -79,10 +79,10 @@ class Message(TypedDict):
   Chat message.
   """
 
-  role: Literal['user', 'assistant', 'system']
-  "Assumed role of the message. Response messages always has role 'assistant'."
+  role: Literal['user', 'assistant', 'system', 'tool']
+  "Assumed role of the message. Response messages always has role 'assistant' or 'tool'."
 
-  content: str
+  content: NotRequired[str]
   'Content of the message. Response messages contains message fragments when streaming.'
 
   images: NotRequired[Sequence[Any]]