Use system message for attachment content

microsoft · Oct 23, 2024 · b5c20d9 · b5c20d9
1 parent f62b63e
commit b5c20d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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/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.")