74 lines
2.9 KiB
Python
74 lines
2.9 KiB
Python
"""Risk evidence, identity and model boundaries use synthetic inputs only."""
|
|
import asyncio
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def risk(trading):
|
|
from alphapulse.news_research import model as service
|
|
return service
|
|
|
|
|
|
def test_truncated_completion_is_not_a_safe_label(risk, monkeypatch):
|
|
from alphapulse.news_research.contracts import IndexResult
|
|
from types import SimpleNamespace
|
|
async def stream(*args):
|
|
yield risk.llm.Completion(truncated=True, usage={'output_tokens': 100})
|
|
monkeypatch.setattr(risk.llm, 'stream', stream)
|
|
with pytest.raises(risk.OutputTruncated, match='截断') as exc:
|
|
asyncio.run(risk.complete(SimpleNamespace(max_tokens=100, context_window=10000, reasoning_effort=None), '', {}, IndexResult))
|
|
assert exc.value.usage == {'output_tokens': 100}
|
|
|
|
|
|
def test_model_override_never_changes_shared_provider(risk):
|
|
from types import SimpleNamespace
|
|
provider = SimpleNamespace(model='original')
|
|
class Session:
|
|
async def get(self, *args): return provider
|
|
def expunge(self, obj): self.detached = obj.model
|
|
session = Session()
|
|
result = asyncio.run(risk.provider_for(session, SimpleNamespace(provider_id=1, model='LUNA')))
|
|
assert session.detached == 'original' and result.model == 'LUNA'
|
|
|
|
|
|
def test_context_budget_includes_unicode_and_tool_schema(risk):
|
|
from alphapulse.news_research.contracts import IndexResult
|
|
from types import SimpleNamespace
|
|
provider = SimpleNamespace(max_tokens=256, context_window=3500)
|
|
assert not risk.fits_context(provider, '只使用原文证据', {'content': '风'*1000}, IndexResult)
|
|
|
|
|
|
@pytest.mark.parametrize('context_window', [250000, 1000000])
|
|
def test_news_context_is_capped_at_250k_including_output(risk, monkeypatch, context_window):
|
|
from pydantic import BaseModel
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
class Result(BaseModel):
|
|
facts: str
|
|
|
|
provider = SimpleNamespace(max_tokens=32000, context_window=context_window)
|
|
assert risk.fits_context(provider, '', {'content': 'x' * 200000}, Result)
|
|
stream = AsyncMock()
|
|
monkeypatch.setattr(risk.llm, 'stream', stream)
|
|
with pytest.raises(ValueError, match='超出模型上下文'):
|
|
asyncio.run(risk.complete(provider, '', {'content': 'x' * 220000}, Result))
|
|
stream.assert_not_called()
|
|
|
|
|
|
def test_research_reserves_tool_room_within_250k_cap(risk):
|
|
from alphapulse.news_research.tools import Research
|
|
from pydantic import BaseModel
|
|
from types import SimpleNamespace
|
|
|
|
class Result(BaseModel):
|
|
facts: str
|
|
|
|
reader = Research()
|
|
payload = {'content': 'x' * 190000}
|
|
for context_window in (250000, 1000000):
|
|
provider = SimpleNamespace(max_tokens=32000, context_window=context_window)
|
|
assert risk.fits_context(provider, '', payload, Result)
|
|
assert not reader.fits(provider, '', payload, Result)
|
|
assert provider.context_window == context_window
|