obscura/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-09-13 06:29:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

literal

Browse Source
This commit is contained in:
Michael Yang
2024-01-10 16:39:55 -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
ollama/_client.py
+5 -5
View File
@@ -7,7 +7,7 @@ from pathlib import Path
from hashlib import sha256 from hashlib import sha256
from base64 import b64encode 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 import sys
@@ -81,7 +81,7 @@ class Client(BaseClient):
context: Optional[Sequence[int]] = None, context: Optional[Sequence[int]] = None,
stream: bool = False, stream: bool = False,
raw: bool = False, raw: bool = False,
format: str = '', format: Literal['', 'json'] = '',
images: Optional[Sequence[AnyStr]] = None, images: Optional[Sequence[AnyStr]] = None,
options: Optional[Options] = None, options: Optional[Options] = None,
) -> Union[Mapping[str, Any], Iterator[Mapping[str, Any]]]: ) -> Union[Mapping[str, Any], Iterator[Mapping[str, Any]]]:
@@ -121,7 +121,7 @@ class Client(BaseClient):
model: str = '', model: str = '',
messages: Optional[Sequence[Message]] = None, messages: Optional[Sequence[Message]] = None,
stream: bool = False, stream: bool = False,
format: str = '', format: Literal['', 'json'] = '',
options: Optional[Options] = None, options: Optional[Options] = None,
) -> Union[Mapping[str, Any], Iterator[Mapping[str, Any]]]: ) -> Union[Mapping[str, Any], Iterator[Mapping[str, Any]]]:
""" """
@@ -338,7 +338,7 @@ class AsyncClient(BaseClient):
context: Optional[Sequence[int]] = None, context: Optional[Sequence[int]] = None,
stream: bool = False, stream: bool = False,
raw: bool = False, raw: bool = False,
format: str = '', format: Literal['', 'json'] = '',
images: Optional[Sequence[AnyStr]] = None, images: Optional[Sequence[AnyStr]] = None,
options: Optional[Options] = None, options: Optional[Options] = None,
) -> Union[Mapping[str, Any], AsyncIterator[Mapping[str, Any]]]: ) -> Union[Mapping[str, Any], AsyncIterator[Mapping[str, Any]]]:
@@ -377,7 +377,7 @@ class AsyncClient(BaseClient):
model: str = '', model: str = '',
messages: Optional[Sequence[Message]] = None, messages: Optional[Sequence[Message]] = None,
stream: bool = False, stream: bool = False,
format: str = '', format: Literal['', 'json'] = '',
options: Optional[Options] = None, options: Optional[Options] = None,
) -> Union[Mapping[str, Any], AsyncIterator[Mapping[str, Any]]]: ) -> Union[Mapping[str, Any], AsyncIterator[Mapping[str, Any]]]:
""" """
ollama/_types.py
+3 -2
View File
@@ -1,4 +1,4 @@
from typing import Any, TypedDict, Sequence from typing import Any, TypedDict, Sequence, Literal
import sys import sys
@@ -27,7 +27,8 @@ class GenerateResponse(BaseGenerateResponse):
class Message(TypedDict): class Message(TypedDict):
role: str role: Literal['user', 'assistant', 'system']
content: str content: str
images: NotRequired[Sequence[Any]] images: NotRequired[Sequence[Any]]