if_lang 0.1.4

Intent-first functional IR language for LLM-friendly logic drafts
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

def list_len(args, ctx):
    if len(args) != 1:
        raise Exception("list_len expects 1 arg")
    return len(args[0])


def list_get(args, ctx):
    if len(args) != 2:
        raise Exception("list_get expects 2 args")
    items = list(args[0])
    idx = int(args[1])
    if idx < 0 or idx >= len(items):
        raise Exception("list_get index out of bounds")
    return items[idx]


def list_push(args, ctx):
    if len(args) != 2:
        raise Exception("list_push expects 2 args")
    items = list(args[0])
    items.append(args[1])
    return items


def str_split_ws(args, ctx):
    if len(args) != 1:
        raise Exception("str_split_ws expects 1 arg")
    return str(args[0]).split()


def str_trim_end_char(args, ctx):
    if len(args) != 2:
        raise Exception("str_trim_end_char expects 2 args")
    value = str(args[0])
    suffix = str(args[1])
    if len(suffix) != 1:
        raise Exception("str_trim_end_char expects 1-char suffix")
    if value.endswith(suffix):
        return value[: -1]
    return value


def str_to_int(args, ctx):
    if len(args) != 1:
        raise Exception("str_to_int expects 1 arg")
    try:
        return int(str(args[0]))
    except Exception:
        raise Exception("str_to_int invalid Int")


def assert_eq(args, ctx):
    if len(args) != 2:
        raise Exception("assert_eq expects 2 args")
    left = args[0]
    right = args[1]
    if left == right:
        return True
    raise Exception(f"assert_eq failed: left={left} right={right}")


def if_lang_register(registry):
    registry["list_len"] = list_len
    registry["list_get"] = list_get
    registry["list_push"] = list_push
    registry["str_split_ws"] = str_split_ws
    registry["str_trim_end_char"] = str_trim_end_char
    registry["str_to_int"] = str_to_int
    registry["assert_eq"] = assert_eq