From ba27e71a20c42a0804cdb874498d4265f5d914db Mon Sep 17 00:00:00 2001 From: ParthSareen Date: Mon, 13 Jan 2025 17:12:25 -0800 Subject: [PATCH] make 3.8 happy --- ollama/_client.py | 48 +++++++++++++++++------------------------------ 1 file changed, 17 insertions(+), 31 deletions(-) diff --git a/ollama/_client.py b/ollama/_client.py index 0bd8598..50e7ec4 100644 --- a/ollama/_client.py +++ b/ollama/_client.py @@ -18,6 +18,8 @@ from typing import ( TypeVar, Union, overload, + Dict, + List, ) import sys @@ -476,10 +478,10 @@ class Client(BaseClient): model: str, quantize: Optional[str] = None, from_: Optional[str] = None, - files: Optional[dict[str, str]] = None, - adapters: Optional[dict[str, str]] = None, + files: Optional[Dict[str, str]] = None, + adapters: Optional[Dict[str, str]] = None, template: Optional[str] = None, - license: Optional[Union[str, list[str]]] = None, + license: Optional[Union[str, List[str]]] = None, system: Optional[str] = None, parameters: Optional[Union[Mapping[str, Any], Options]] = None, messages: Optional[Sequence[Union[Mapping[str, Any], Message]]] = None, @@ -493,26 +495,10 @@ class Client(BaseClient): model: str, quantize: Optional[str] = None, from_: Optional[str] = None, - files: Optional[dict[str, str]] = None, - adapters: Optional[dict[str, str]] = None, + files: Optional[Dict[str, str]] = None, + adapters: Optional[Dict[str, str]] = None, template: Optional[str] = None, - license: Optional[Union[str, list[str]]] = None, - system: Optional[str] = None, - parameters: Optional[Union[Mapping[str, Any], Options]] = None, - messages: Optional[Sequence[Union[Mapping[str, Any], Message]]] = None, - *, - stream: Literal[True] = True, - ) -> Iterator[ProgressResponse]: ... - - def create( - self, - model: str, - quantize: Optional[str] = None, - from_: Optional[str] = None, - files: Optional[dict[str, str]] = None, - adapters: Optional[dict[str, str]] = None, - template: Optional[str] = None, - license: Optional[Union[str, list[str]]] = None, + license: Optional[Union[str, List[str]]] = None, system: Optional[str] = None, parameters: Optional[Union[Mapping[str, Any], Options]] = None, messages: Optional[Sequence[Union[Mapping[str, Any], Message]]] = None, @@ -978,10 +964,10 @@ class AsyncClient(BaseClient): model: str, quantize: Optional[str] = None, from_: Optional[str] = None, - files: Optional[dict[str, str]] = None, - adapters: Optional[dict[str, str]] = None, + files: Optional[Dict[str, str]] = None, + adapters: Optional[Dict[str, str]] = None, template: Optional[str] = None, - license: Optional[Union[str, list[str]]] = None, + license: Optional[Union[str, List[str]]] = None, system: Optional[str] = None, parameters: Optional[Union[Mapping[str, Any], Options]] = None, messages: Optional[Sequence[Union[Mapping[str, Any], Message]]] = None, @@ -995,10 +981,10 @@ class AsyncClient(BaseClient): model: str, quantize: Optional[str] = None, from_: Optional[str] = None, - files: Optional[dict[str, str]] = None, - adapters: Optional[dict[str, str]] = None, + files: Optional[Dict[str, str]] = None, + adapters: Optional[Dict[str, str]] = None, template: Optional[str] = None, - license: Optional[Union[str, list[str]]] = None, + license: Optional[Union[str, List[str]]] = None, system: Optional[str] = None, parameters: Optional[Union[Mapping[str, Any], Options]] = None, messages: Optional[Sequence[Union[Mapping[str, Any], Message]]] = None, @@ -1011,10 +997,10 @@ class AsyncClient(BaseClient): model: str, quantize: Optional[str] = None, from_: Optional[str] = None, - files: Optional[dict[str, str]] = None, - adapters: Optional[dict[str, str]] = None, + files: Optional[Dict[str, str]] = None, + adapters: Optional[Dict[str, str]] = None, template: Optional[str] = None, - license: Optional[Union[str, list[str]]] = None, + license: Optional[Union[str, List[str]]] = None, system: Optional[str] = None, parameters: Optional[Union[Mapping[str, Any], Options]] = None, messages: Optional[Sequence[Union[Mapping[str, Any], Message]]] = None,