Skip to content

Commit c080458

Browse files
committed
fix: use alias in prompt
1 parent d3e3e23 commit c080458

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

src/server/api/memobase_server/prompts/roleplay/zh_infer_plot.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,12 @@ def get_prompt() -> str:
5151

5252

5353
def pack_messages(messages: list[OpenAICompatibleMessage]):
54-
return "\n".join([f"[{m.role}]: {m.content}" for m in messages])
54+
return "\n".join(
55+
[
56+
f"[{m.role}{f'({m.alias})' if m.alias else ''}]: {m.content}"
57+
for m in messages
58+
]
59+
)
5560

5661

5762
def get_input(role, user, before_plots: list[str], messages: list):

0 commit comments

Comments
 (0)