97 lines
2.3 KiB
Python
97 lines
2.3 KiB
Python
"""port/fix of ../langgraph-demo, with all the bullshit that it has.
|
|
|
|
Constraints:
|
|
no tapping wf_core
|
|
try to use builtin wf_authoring.ops
|
|
"""
|
|
|
|
from itertools import islice
|
|
from pprint import pprint
|
|
from typing import Any
|
|
|
|
from tests.rewrite.context import context
|
|
from tests.rewrite.models import (
|
|
Context,
|
|
Input,
|
|
State,
|
|
)
|
|
from tests.rewrite.workflow import gacha
|
|
from wf_core.run_state import RunStatus
|
|
|
|
|
|
def build_input_lite(
|
|
rolling: int,
|
|
rolled_previously: int = 0,
|
|
until_5: int = 10,
|
|
until_6: int = 80,
|
|
good_stuff: bool = False,
|
|
) -> dict[str, Any]:
|
|
if not 0 < until_5 <= 10:
|
|
print(f"{until_5 = } invalid, idc")
|
|
if not 0 < until_6 <= 80:
|
|
print(f"{until_6 = } invalid, running anyways")
|
|
return {
|
|
"countdown": rolling,
|
|
"simple_counter": rolled_previously,
|
|
"counter": {
|
|
"c_10": 10 - until_5,
|
|
"c_80": 80 - until_6,
|
|
},
|
|
"pity_120_available": not good_stuff,
|
|
}
|
|
|
|
|
|
def build_input(context: Context):
|
|
def dec(
|
|
rolling: int,
|
|
rolled_previously: int = 0,
|
|
until_5: int = 10,
|
|
until_6: int = 80,
|
|
good_stuff: bool = False,
|
|
) -> Input:
|
|
r = build_input_lite(rolling, rolled_previously, until_5, until_6, good_stuff)
|
|
return Input.model_validate(r | {"context": context})
|
|
|
|
return dec
|
|
|
|
|
|
# twice in a row! it took 100+ and a miss tho
|
|
|
|
|
|
def test():
|
|
assert 240 - 135 + 20 >= 120, "my math!"
|
|
d = gacha.execute(
|
|
build_input(context)(
|
|
20, # lets be optimistic
|
|
rolled_previously=240 - 135, # 15
|
|
until_5=5, # 5
|
|
until_6=73, # 7
|
|
good_stuff=False, # lets pretend
|
|
).model_dump()
|
|
)
|
|
assert d.status == RunStatus.COMPLETED, "oops"
|
|
state = State.model_validate(d.state)
|
|
pprint(state.storage)
|
|
pprint(
|
|
[
|
|
t
|
|
for t in d.trace
|
|
if t.node_id
|
|
in (
|
|
"counter_up",
|
|
"tick",
|
|
)
|
|
]
|
|
)
|
|
pprint(d.state)
|
|
assert any(
|
|
i["name"] in context["pool"]["n_240"]
|
|
for i in islice(state.storage, 120 - (240 - 135))
|
|
), "pity logic failed"
|
|
# pprint(gacha.compile().edges)
|
|
# pprint(gacha.compile().nodes)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test()
|