eliott/ollama-python
1
0
Fork 0
mirror of https://github.com/kharonsec/ollama-python synced 2026-04-25 15:14:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

literal

Browse Source
This commit is contained in:
Michael Yang
2024-01-10 16:28:48 -08:00
parent 7601947a35
commit f7e6980b1b
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ollama/_types.py
View File

@@ -1,4 +1,4 @@
from typing import Any, TypedDict, Sequence
from typing import Any, TypedDict, Sequence, Literal
import sys
@@ -27,7 +27,8 @@ class GenerateResponse(BaseGenerateResponse):
class Message(TypedDict):
role: str
role: Literal['user', 'assistant', 'system']
content: str
images: NotRequired[Sequence[Any]]