outlines-core 0.2.14

Structured Generation
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
70
71
72
73
74
75
76
77

"""Tests for package imports to catch import/module registration issues."""


def test_import_outlines_core():
    import outlines_core

    assert hasattr(outlines_core, "Guide")
    assert hasattr(outlines_core, "Index")
    assert hasattr(outlines_core, "Vocabulary")
    assert hasattr(outlines_core, "json_schema")


def test_import_json_schema_module():
    from outlines_core import json_schema

    assert hasattr(json_schema, "BOOLEAN")
    assert hasattr(json_schema, "build_regex_from_schema")


def test_import_from_json_schema():
    from outlines_core.json_schema import (  # noqa: F401
        BOOLEAN,
        DATE,
        DATE_TIME,
        EMAIL,
        INTEGER,
        NULL,
        NUMBER,
        STRING,
        STRING_INNER,
        TIME,
        URI,
        UUID,
        WHITESPACE,
        build_regex_from_schema,
    )

    assert BOOLEAN == "(true|false)"
    assert callable(build_regex_from_schema)


def test_import_main_classes():
    from outlines_core import Guide, Index, Vocabulary

    assert Guide is not None
    assert Index is not None
    assert Vocabulary is not None


# The tests below ensure that users can import using
# `outlines_core.outlines_core`. This is not the intended usage but we should
# keep supporting it as users may have fallen back to it when the regular
# Python imports were buggy (fixed in v0.2.14)


def test_import_rust_extension_directly():
    from outlines_core import outlines_core

    assert hasattr(outlines_core, "Guide")
    assert hasattr(outlines_core, "Index")
    assert hasattr(outlines_core, "Vocabulary")
    assert hasattr(outlines_core, "json_schema")


def test_import_from_rust_extension():
    from outlines_core.outlines_core import Guide, Index, Vocabulary

    assert Guide is not None
    assert Index is not None
    assert Vocabulary is not None


def test_import_json_schema_from_rust_extension():
    from outlines_core.outlines_core import json_schema

    assert hasattr(json_schema, "BOOLEAN")
    assert hasattr(json_schema, "build_regex_from_schema")