From 0f8c20a596c32832a7f6e293523b7c420bcc4172 Mon Sep 17 00:00:00 2001 From: tillfalko Date: Sat, 18 May 2024 16:48:53 +0200 Subject: [PATCH] Avoid side effects if chat message contains images --- ollama/_client.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/ollama/_client.py b/ollama/_client.py index 6e40081..1ad90ae 100644 --- a/ollama/_client.py +++ b/ollama/_client.py @@ -7,6 +7,7 @@ import urllib.parse from os import PathLike from pathlib import Path +from copy import deepcopy from hashlib import sha256 from base64 import b64encode, b64decode @@ -164,6 +165,8 @@ def chat( if not model: raise RequestError('must provide a model') + messages = deepcopy(messages) + for message in messages or []: if not isinstance(message, dict): raise TypeError('messages must be a list of Message or dict-like objects') @@ -446,6 +449,8 @@ async def chat( if not model: raise RequestError('must provide a model') + messages = deepcopy(messages) + for message in messages or []: if not isinstance(message, dict): raise TypeError('messages must be a list of strings')