feat: add trusted ticket intake backend
This commit is contained in:
@@ -128,6 +128,37 @@ def test_cancellation_after_model_response_stops_following_steps(db, monkeypatch
|
||||
assert run.final_output is None
|
||||
|
||||
|
||||
def test_multimodal_request_keeps_base64_out_of_database_snapshots(db, monkeypatch):
|
||||
_, version = create_agent(two_steps=False)
|
||||
captured = {}
|
||||
|
||||
def request(deployment, messages, **kwargs):
|
||||
captured["messages"] = messages
|
||||
return CompletionResult('{"summary":"识别到截图"}', 8, 3)
|
||||
|
||||
monkeypatch.setattr("apps.modelhub.services._request_completion", request)
|
||||
run = create_run(version)
|
||||
run.input_payload["image_refs"] = [
|
||||
{"object_path": "ticket-drafts/proof.png", "mime_type": "image/png"}
|
||||
]
|
||||
run.save(update_fields=["input_payload"])
|
||||
|
||||
execute_agent_run(
|
||||
run.id,
|
||||
image_parts=[
|
||||
{
|
||||
"type": "image_url",
|
||||
"image_url": {"url": "data:image/png;base64,c2Vuc2l0aXZl"},
|
||||
}
|
||||
],
|
||||
)
|
||||
|
||||
invocation = run.steps.get().attempts.get(attempt_type="initial").model_invocation
|
||||
assert "base64" in str(captured["messages"])
|
||||
assert "base64" not in str(invocation.input_snapshot)
|
||||
assert invocation.input_snapshot["image_refs"][0]["object_path"].endswith("proof.png")
|
||||
|
||||
|
||||
def test_soft_time_limit_marks_run_timed_out_and_audits(db, monkeypatch):
|
||||
_, version = create_agent()
|
||||
run = create_run(version)
|
||||
|
||||
Reference in New Issue
Block a user