github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-23 12:21:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

修改main分支的mlx_adapter.py中get_str_prompt函数的历史消息格式问题

Browse Source
This commit is contained in:
wendell_yi 2025-07-15 18:00:08 +08:00
parent b730fe5720
commit 1f7cd03222
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
packages/dbgpt-core/src/dbgpt/model/adapter/mlx_adapter.py
View File

@ -116,12 +116,11 @@ class MLXModelAdapter(LLMModelAdapter):
messages, convert_to_compatible_format
)
logger.debug(f"The messages after transform: \n{messages}")
#wendell-debug 确保每个 message['content'] 是 str
for msg in messages:
if isinstance(msg.get('content'), list):
msg['content'] = ''.join([str(x) for x in msg['content']])
elif not isinstance(msg.get('content'), str):
msg['content'] = str(msg.get('content'))
if isinstance(msg.get("content"), list):
msg["content"] = "".join([str(x) for x in msg["content"]])
elif not isinstance(msg.get("content"), str):
msg["content"] = str(msg.get("content"))
str_prompt = tokenizer.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True