diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 10 |
1 files changed, 8 insertions, 2 deletions
@@ -172,8 +172,14 @@ async def ollamachat(websocket): if current_user_lines: messages.append({"role": "user", "content": "\n".join(current_user_lines)}) - # final isolated turn — the specific message she needs to reply to - messages.append({"role": "user", "content": f"[responding to]: {nots(user_input)}"}) + # final isolated turn — the specific message she needs to reply to, with the speaker named + # so she doesn't anchor on whoever talked last in the history above + speaker, text = parse_line(user_input) + if speaker and text: + final_turn = f"[responding to {speaker}]: {text}\n(reply only to {speaker}, the lines before this are other people's messages for context)" + else: + final_turn = f"[responding to]: {nots(user_input)}" # couldn't parse it, old format + messages.append({"role": "user", "content": final_turn}) # ollama only takes images (no audio/video/pdf), and only on the final turn so history stays text images = [base64.b64decode(m["data"]) for m in media if m["mime_type"].startswith("image/")] |