mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-06 23:12:22 +02:00
pewdiepie-archdaemon
2e8413a54a
Snapshot current maintainer-preview application changes and regression fixtures for integration into lab. Excludes local runtime data, evaluation outputs and source backups. Focused Python regression selection: 140 passed; full suite not certified.
42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
import json
|
|
import pytest
|
|
from src.email_reply_stream import reply_body, stream_reply
|
|
|
|
|
|
def test_analysis_is_not_a_reply():
|
|
assert reply_body('The user wants me to draft a reply. Context Analysis: ...', complete=True) == ''
|
|
assert reply_body('<think>analysis</think>', complete=True) == ''
|
|
assert reply_body('<<<REPLY>>>Hi Jonathan', complete=True) == ''
|
|
assert reply_body('<<<REPLY>>><think>analysis</think>Hi<<<END>>>', complete=True) == ''
|
|
assert reply_body('analysis\n<<<REPLY>>>Hi Jonathan,\nThanks.<<<END>>>', complete=True) == 'Hi Jonathan,\nThanks.'
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stream_filters_reasoning_and_split_markers(monkeypatch):
|
|
from src import llm_core
|
|
calls, events = [], []
|
|
async def model(*args, **kwargs):
|
|
calls.append(kwargs)
|
|
for delta in ['Private analysis.', '<<<REP', 'LY>>>Hi Jonathan,', '\nThanks.', '<<<EN', 'D>>>']:
|
|
yield 'data: ' + json.dumps({'delta': delta}) + '\n\n'
|
|
async def emit(event): events.append(event)
|
|
monkeypatch.setattr(llm_core, 'stream_llm', model)
|
|
raw, used = await stream_reply([('http://local', 'Ajax', {})], [], emit)
|
|
assert used == 'Ajax'
|
|
assert calls[0]['thinking_mode'] == 'off'
|
|
assert events[-1]['text'] == 'Hi Jonathan,\nThanks.'
|
|
assert all('analysis' not in e['text'] and '<<<' not in e['text'] for e in events)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_incomplete_stream_clears_partial_draft(monkeypatch):
|
|
from src import llm_core
|
|
events = []
|
|
async def model(*args, **kwargs):
|
|
yield 'data: ' + json.dumps({'delta': '<<<REPLY>>>Partial'}) + '\n\n'
|
|
async def emit(event): events.append(event)
|
|
monkeypatch.setattr(llm_core, 'stream_llm', model)
|
|
with pytest.raises(ValueError, match='incomplete'):
|
|
await stream_reply([('http://local', 'Ajax', {})], [], emit)
|
|
assert events[-1]['text'] == ''
|