kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-14 06:07:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #158 from tillfalko/main

Browse Source 
Avoid side effects if chat message contains images
This commit is contained in:
Michael Yang 2024-06-05 11:28:41 -07:00 committed by GitHub
commit 57c597b60e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ollama/_client.py
View File

@ -7,6 +7,7 @@ import platform
import urllib.parse import urllib.parse
from os import PathLike from os import PathLike
from pathlib import Path from pathlib import Path
from copy import deepcopy
from hashlib import sha256 from hashlib import sha256
from base64 import b64encode, b64decode from base64 import b64encode, b64decode
@ -164,6 +165,8 @@ class Client(BaseClient):
if not model: if not model:
raise RequestError('must provide a model') raise RequestError('must provide a model')
messages = deepcopy(messages)
for message in messages or []: for message in messages or []:
if not isinstance(message, dict): if not isinstance(message, dict):
raise TypeError('messages must be a list of Message or dict-like objects') raise TypeError('messages must be a list of Message or dict-like objects')
@ -449,6 +452,8 @@ class AsyncClient(BaseClient):
if not model: if not model:
raise RequestError('must provide a model') raise RequestError('must provide a model')
messages = deepcopy(messages)
for message in messages or []: for message in messages or []:
if not isinstance(message, dict): if not isinstance(message, dict):
raise TypeError('messages must be a list of strings') raise TypeError('messages must be a list of strings')