zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-22 21:16:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: optimize the messagef content processing

Browse Source
This commit is contained in:
anchen 2025-01-27 09:37:29 +11:00
parent 74ae24b883
commit 07f3d7d6bb
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
llms/mlx_lm/server.py
View File

@ -133,15 +133,15 @@ def process_message_content(messages):
processed_messages = []
for message in messages:
message_copy = message.copy()
if "content" in message_copy:
flattened_text = ""
if isinstance(message_copy["content"], list):
for content_fragment in message_copy["content"]:
if content_fragment["type"] == "text" and "text" in content_fragment:
flattened_text += content_fragment["text"]
else:
if "content" in message_copy and isinstance(message_copy["content"], list):
text_fragments = [
fragment["text"]
for fragment in message_copy["content"]
if fragment.get("type") == "text"
]
if len(text_fragments) != len(message_copy["content"]):
raise ValueError("Only 'text' content type is supported.")
message_copy["content"] = flattened_text
message_copy["content"] = "".join(text_fragments)
processed_messages.append(message_copy)
return processed_messages