Skip to content

Commit fcb5137

Browse files
committed
fix: remove LLMaJ template; uses Requirement template
1 parent 2621825 commit fcb5137

File tree

2 files changed

+11
-16
lines changed

2 files changed

+11
-16
lines changed

mellea/templates/prompts/default/LLMaJRequirement.jinja2

Lines changed: 0 additions & 15 deletions
This file was deleted.

test/stdlib_basics/test_requirement.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import asyncio
22
import pytest
33
from mellea.stdlib.base import ChatContext, ModelOutputThunk
4-
from mellea.stdlib.requirement import Requirement, simple_validate
4+
from mellea.stdlib.requirement import LLMaJRequirement, Requirement, simple_validate
55
from mellea.stdlib.session import start_session
66

77
ctx = ChatContext()
@@ -18,6 +18,16 @@ async def test_llmaj_validation_req_output_field():
1818
"requirement's output shouldn't be updated during/after validation"
1919
)
2020

21+
async def test_llmaj_requirement_uses_requirement_template():
22+
m = start_session(ctx=ctx)
23+
req = LLMaJRequirement("Must output test.")
24+
assert req._output is None
25+
26+
_ = await req.validate(m.backend, ctx=ctx)
27+
assert req._output is None, (
28+
"requirement's output shouldn't be updated during/after validation"
29+
)
30+
2131

2232
def test_simple_validate_bool():
2333
validation_func = simple_validate(lambda x: False, reason="static reason")

0 commit comments

Comments
 (0)