vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-08-06 15:05:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

type check hardening

Browse source
This commit is contained in:
Concedo 2026-08-02 10:48:23 +08:00
parent 9e64023c7b
commit bf9b9bd455
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
koboldcpp.py
View file

@ -4262,6 +4262,8 @@ def sweep_media_from_messages(messages_array):
curr_content = message.get("content", None)
if isinstance(curr_content, list):
for item in curr_content:
if not isinstance(item, dict):
continue
if item.get("type") == "image_url":
url = item.get("image_url", {}).get("url", "")
if url.startswith("data:image"):
@ -4485,20 +4487,21 @@ ws ::= | " " | "\n" [ \t]{0,20}
elif isinstance(curr_content, list): #is an array
for item in curr_content:
if isinstance(item, dict):
if item['type']=="text":
item_type = item.get("type", "")
if item_type=="text":
messages_string += item['text']
elif item['type']=="image_url":
elif item_type=="image_url":
if 'image_url' in item and item['image_url'] and item['image_url']['url'] and item['image_url']['url'].startswith("data:image"):
images_added.append(item['image_url']['url'].split(",", 1)[1])
attachedimgid += 1
messages_string += f"\n(Attached Image {attachedimgid})\n"
elif item['type']=="image":
elif item_type=="image":
data = get_base64_from_media_data(item.get("data", ""))
if data:
images_added.append(data)
attachedimgid += 1
messages_string += f"\n(Attached Image {attachedimgid})\n"
elif item['type']=="input_audio":
elif item_type=="input_audio":
if 'input_audio' in item and item['input_audio'] and item['input_audio']['data']:
audio_added.append(item['input_audio']['data'])
attachedaudid += 1