kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-04-23 15:58:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add model to kwargs

Browse Source
This commit is contained in:
Michael Yang 2023-12-20 12:41:54 -08:00
parent 877ca407b6
commit 21578e3c5e
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ollama/client.py
View File

@ -33,7 +33,10 @@ class Client(BaseClient):
raise Exception(e)
yield part
def generate(self, model, prompt='', system='', template='', context=None, stream=False, raw=False, format='', images=None, options=None):
def generate(self, model='', prompt='', system='', template='', context=None, stream=False, raw=False, format='', images=None, options=None):
if not model:
raise Exception('must provide a model')
fn = self.stream if stream else self._request_json
return fn('POST', '/api/generate', json={
'model': model,
@ -48,7 +51,10 @@ class Client(BaseClient):
'options': options or {},
})
def chat(self, model, messages=None, stream=False, format='', options=None):
def chat(self, model='', messages=None, stream=False, format='', options=None):
if not model:
raise Exception('must provide a model')
for message in messages or []:
if not isinstance(message, dict):
raise TypeError('messages must be a list of strings')