kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +08:00
Code Issues Actions 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ollama/_client.py
View File

@ -7,7 +7,7 @@ from pathlib import Path
from hashlib import sha256
from base64 import b64encode
from typing import Any, AnyStr, Union, Optional, Sequence, Mapping
from typing import Any, AnyStr, Union, Optional, Sequence, Mapping, Literal
import sys
@ -81,7 +81,7 @@ class Client(BaseClient):
context: Optional[Sequence[int]] = None,
stream: bool = False,
raw: bool = False,
format: str = '',
format: Literal['', 'json'] = '',
images: Optional[Sequence[AnyStr]] = None,
options: Optional[Options] = None,
) -> Union[Mapping[str, Any], Iterator[Mapping[str, Any]]]:
@ -121,7 +121,7 @@ class Client(BaseClient):
model: str = '',
messages: Optional[Sequence[Message]] = None,
stream: bool = False,
format: str = '',
format: Literal['', 'json'] = '',
options: Optional[Options] = None,
) -> Union[Mapping[str, Any], Iterator[Mapping[str, Any]]]:
"""
@ -338,7 +338,7 @@ class AsyncClient(BaseClient):
context: Optional[Sequence[int]] = None,
stream: bool = False,
raw: bool = False,
format: str = '',
format: Literal['', 'json'] = '',
images: Optional[Sequence[AnyStr]] = None,
options: Optional[Options] = None,
) -> Union[Mapping[str, Any], AsyncIterator[Mapping[str, Any]]]:
@ -377,7 +377,7 @@ class AsyncClient(BaseClient):
model: str = '',
messages: Optional[Sequence[Message]] = None,
stream: bool = False,
format: str = '',
format: Literal['', 'json'] = '',
options: Optional[Options] = None,
) -> Union[Mapping[str, Any], AsyncIterator[Mapping[str, Any]]]:
"""

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]]