diff --git a/libraries/python/assistant-extensions/assistant_extensions/attachments/_attachments.py b/libraries/python/assistant-extensions/assistant_extensions/attachments/_attachments.py index 9d7af16f..8656424a 100644 --- a/libraries/python/assistant-extensions/assistant_extensions/attachments/_attachments.py +++ b/libraries/python/assistant-extensions/assistant_extensions/attachments/_attachments.py @@ -186,7 +186,7 @@ async def get_completion_messages_for_attachments( ] messages.append({ - "role": "user", + "role": "system", "content": content, }) diff --git a/libraries/python/assistant-extensions/assistant_extensions/attachments/tests/test_attachments.py b/libraries/python/assistant-extensions/assistant_extensions/attachments/tests/test_attachments.py index ea9330e1..83e7bfaf 100644 --- a/libraries/python/assistant-extensions/assistant_extensions/attachments/tests/test_attachments.py +++ b/libraries/python/assistant-extensions/assistant_extensions/attachments/tests/test_attachments.py @@ -26,11 +26,11 @@ "content": AttachmentsConfigModel().context_description, }, { - "role": "user", + "role": "system", "content": "file1.txtfile 1", }, { - "role": "user", + "role": "system", "content": "file2.txtfile 2", }, ], @@ -46,11 +46,11 @@ "content": AttachmentsConfigModel().context_description, }, { - "role": "user", + "role": "system", "content": "file1.txterror processing file: file 1 error", }, { - "role": "user", + "role": "system", "content": "file2.txtfile 2", }, ], diff --git a/libraries/python/openai-client/openai_client/tokens.py b/libraries/python/openai-client/openai_client/tokens.py index e85e98d0..a09f5762 100644 --- a/libraries/python/openai-client/openai_client/tokens.py +++ b/libraries/python/openai-client/openai_client/tokens.py @@ -183,8 +183,8 @@ def get_image_dims(image_uri: str) -> tuple[int, int]: # From https://github.com/openai/openai-cookbook/pull/881/files if re.match(r"data:image\/\w+;base64", image_uri): image_uri = re.sub(r"data:image\/\w+;base64,", "", image_uri) - image = Image.open(BytesIO(base64.b64decode(image_uri))) - return image.size + with Image.open(BytesIO(base64.b64decode(image_uri))) as image: + return image.size else: raise ValueError("Image must be a base64 string.")