SYMBOL INDEX (295 symbols across 28 files)

FILE: benchmarks/exllamav2_json.py
  function create_exllamav2_6bpw_llama3_8b (line 16) | def create_exllamav2_6bpw_llama3_8b():
  function create_exllamav2_4bpw_llama2_7b (line 31) | def create_exllamav2_4bpw_llama2_7b():
  function f_get_address_filter (line 46) | def f_get_address_filter():
  function bind (line 54) | def bind(instance, func, as_name):
  function monkey_patch_lfe (line 76) | def monkey_patch_lfe(exllama_filter: ExLlamaV2TokenEnforcerFilter):
  function lfe_get_address_filter (line 85) | def lfe_get_address_filter():
  function f_get_linkedlist_filter (line 92) | def f_get_linkedlist_filter():
  function lfe_get_linkedlist_filter (line 100) | def lfe_get_linkedlist_filter():
  function f_get_order_filter (line 107) | def f_get_order_filter():
  function lfe_get_order_filter (line 115) | def lfe_get_order_filter():
  function execute (line 122) | def execute():
  function warm_up (line 152) | def warm_up(f):
  function bench (line 158) | def bench(result: BenchResult, context: Context, func, bench_name: str, f):

FILE: benchmarks/transformers_json.py
  function get_llama3_8b_tokenizer_and_model (line 16) | def get_llama3_8b_tokenizer_and_model():
  function get_llama2_7b_tokenizer_and_model (line 27) | def get_llama2_7b_tokenizer_and_model():
  function get_address_schema (line 38) | def get_address_schema():
  function lfe_address_prefix (line 44) | def lfe_address_prefix():
  function get_linkedlist_schema (line 48) | def get_linkedlist_schema():
  function lfe_linkedlist_prefix (line 54) | def lfe_linkedlist_prefix():
  function get_order_schema (line 58) | def get_order_schema():
  function lfe_order_prefix (line 64) | def lfe_order_prefix():
  function execute (line 68) | def execute():
  function warm_up (line 87) | def warm_up(f):
  function bench (line 93) | def bench(result: BenchResult, context: Context, func, bench_name: str, f):

FILE: benchmarks/utils.py
  class Address (line 11) | class Address(ClassSchema):
  class LinkedList (line 20) | class LinkedList(ClassSchema):
  class OrderItem (line 26) | class OrderItem(ClassSchema):
  class Customer (line 32) | class Customer(ClassSchema):
  class Order (line 40) | class Order(ClassSchema):
  class BenchResult (line 50) | class BenchResult:
  class Context (line 57) | class Context:
  function log (line 61) | def log(func_name:str, data:BenchResult,f):
  function load_address (line 72) | def load_address()->list[str]:
  function load_linkedlist (line 75) | def load_linkedlist()->list[str]:
  function load_orders (line 78) | def load_orders()->list[str]:
  function force_gc (line 81) | def force_gc():

FILE: benchmarks/vllm_json.py
  function execute (line 20) | def execute():
  function formatron_vllm_address (line 35) | def formatron_vllm_address():
  function lfe_vllm_address (line 44) | def lfe_vllm_address():
  function outlines_address (line 51) | def outlines_address():
  function formatron_vllm_linkedlist (line 58) | def formatron_vllm_linkedlist():
  function lfe_vllm_linkedlist (line 68) | def lfe_vllm_linkedlist():
  function formatron_vllm_order (line 75) | def formatron_vllm_order():
  function lfe_vllm_order (line 84) | def lfe_vllm_order():
  function outlines_order (line 91) | def outlines_order():
  function warm_up (line 98) | def warm_up(f):
  function bench (line 104) | def bench(result: BenchResult, context: Context, func, bench_name: str, f):

FILE: docs/header_gen.py
  function sort_versions_by_latest (line 4) | def sort_versions_by_latest(versions):

FILE: examples/test_readme_examples.py
  function test_readme_example (line 1) | def test_readme_example():
  function test_readme_example2 (line 30) | def test_readme_example2():
  function test_readme_example3 (line 59) | def test_readme_example3():
  function test_readme_example4 (line 94) | def test_readme_example4():
  function test_readme_example5 (line 125) | def test_readme_example5():
  function test_readme_example6 (line 188) | def test_readme_example6():
  function test_readme_example7 (line 231) | def test_readme_example7():
  function test_readme_example8 (line 259) | def test_readme_example8():
  function test_readme_example9 (line 294) | def test_readme_example9():

FILE: src/formatron/config.py
  class EngineGenerationConfig (line 8) | class EngineGenerationConfig:

FILE: src/formatron/extractor.py
  class Extractor (line 10) | class Extractor(abc.ABC):
    method __init__ (line 15) | def __init__(self, capture_name: typing.Optional[str] = None):
    method capture_name (line 24) | def capture_name(self) -> typing.Optional[str]:
    method extract (line 31) | def extract(self, input_str: str) -> typing.Optional[tuple[str, typing...
    method kbnf_reference (line 42) | def kbnf_reference(self) -> str:
    method __str__ (line 48) | def __str__(self):
    method kbnf_definition (line 53) | def kbnf_definition(self) -> str:
  class NonterminalExtractor (line 63) | class NonterminalExtractor(Extractor):
    method __init__ (line 68) | def __init__(self, nonterminal: str, capture_name: typing.Optional[str...
    method nonterminal (line 79) | def nonterminal(self) -> str:
    method kbnf_reference (line 86) | def kbnf_reference(self) -> str:
  class LiteralExtractor (line 90) | class LiteralExtractor(Extractor):
    method __init__ (line 95) | def __init__(self, literal: str):
    method extract (line 105) | def extract(self, input_str: str) -> typing.Optional[tuple[str, str]]:
    method kbnf_reference (line 115) | def kbnf_reference(self) -> str:
    method kbnf_definition (line 119) | def kbnf_definition(self) -> str:
  class ChoiceExtractor (line 123) | class ChoiceExtractor(NonterminalExtractor):
    method __init__ (line 128) | def __init__(self, choices: typing.Iterable[Extractor], capture_name: ...
    method extract (line 140) | def extract(self, input_str: str) -> typing.Optional[tuple[str, typing...
    method kbnf_definition (line 156) | def kbnf_definition(self) -> str:
  class SubstringExtractor (line 160) | class SubstringExtractor(NonterminalExtractor):
    method __init__ (line 165) | def __init__(self, string: str, capture_name: str, nonterminal: str, *...
    method extract (line 179) | def extract(self, input_str: str) -> typing.Optional[tuple[str, str]]:
    method kbnf_definition (line 201) | def kbnf_definition(self) -> str:

FILE: src/formatron/formats/json.py
  function from_str_to_kbnf_str (line 44) | def from_str_to_kbnf_str(s: str) -> str:
  function register_generate_nonterminal_def (line 61) | def register_generate_nonterminal_def(
  function _register_all_predefined_types (line 79) | def _register_all_predefined_types():
  function _generate_kbnf_grammar (line 425) | def _generate_kbnf_grammar(schema: schemas.schema.Schema|collections.abc...
  class JsonExtractor (line 471) | class JsonExtractor(extractor.NonterminalExtractor):
    method __init__ (line 476) | def __init__(self, nonterminal: str, capture_name: typing.Optional[str...
    method extract (line 528) | def extract(self, input_str: str) -> typing.Optional[tuple[str, schema...
    method kbnf_definition (line 584) | def kbnf_definition(self):

FILE: src/formatron/formats/regex.py
  class RegexExtractor (line 9) | class RegexExtractor(NonterminalExtractor):
    method __init__ (line 14) | def __init__(self, regex: str, capture_name: str, nonterminal: str):
    method extract (line 26) | def extract(self, input_str: str) -> typing.Optional[tuple[str, re.Mat...
    method kbnf_definition (line 42) | def kbnf_definition(self) -> str:
  class RegexComplementExtractor (line 46) | class RegexComplementExtractor(NonterminalExtractor):
    method __init__ (line 51) | def __init__(self, regex: str, capture_name: str, nonterminal: str):
    method extract (line 58) | def extract(self, input_str: str) -> typing.Optional[tuple[str, str]]:
    method kbnf_definition (line 71) | def kbnf_definition(self) -> str:

FILE: src/formatron/formats/utils.py
  function escape_identifier (line 3) | def escape_identifier(s: str) -> str:

FILE: src/formatron/formatter.py
  class FormatterBase (line 20) | class FormatterBase(abc.ABC):
    method accept_token (line 26) | def accept_token(self, token_id: int):
    method accept_bytes (line 36) | def accept_bytes(self, _bytes: bytes)->None:
    method compute_allowed_tokens (line 44) | def compute_allowed_tokens(self) -> None:
    method mask_logits (line 50) | def mask_logits(self, logits) -> typing.Any:
    method get_allowed_tokens_since_last_computation (line 60) | def get_allowed_tokens_since_last_computation(self) -> typing.Sequence...
    method is_completed (line 68) | def is_completed(self) -> bool:
    method _on_completion (line 74) | def _on_completion(self, generated_output: str) -> None:
    method captures (line 81) | def captures(self) -> dict[str, typing.Any|None]:
    method reset (line 87) | def reset(self) -> None:
  class Formatter (line 93) | class Formatter(FormatterBase):
    method __init__ (line 100) | def __init__(self, extractors: list[Extractor], engine: kbnf.Engine,
    method grammar_str (line 118) | def grammar_str(self):
    method accept_token (line 124) | def accept_token(self, token_id: int)->kbnf.AcceptTokenResult:
    method _obtain_accepted_output (line 132) | def _obtain_accepted_output(self)->str:
    method accept_bytes (line 161) | def accept_bytes(self, _bytes: bytes)->kbnf.AcceptTokenResult:
    method compute_allowed_tokens (line 169) | def compute_allowed_tokens(self) -> None:
    method mask_logits (line 172) | def mask_logits(self, logits) -> typing.Any:
    method get_allowed_tokens_since_last_computation (line 175) | def get_allowed_tokens_since_last_computation(self) -> typing.Sequence...
    method is_completed (line 178) | def is_completed(self) -> bool:
    method _on_completion (line 186) | def _on_completion(self, generated_output: str) -> None:
    method captures (line 202) | def captures(self) -> dict[str, typing.Any] | None:
    method reset (line 223) | def reset(self) -> None:
    method __str__ (line 228) | def __str__(self):
  class FormatterBuilder (line 237) | class FormatterBuilder:
    method __init__ (line 243) | def __init__(self):
    method _assert_capture_name_valid (line 257) | def _assert_capture_name_valid(self, capture_name: str):
    method append_line (line 263) | def append_line(self, line: str) -> None:
    method append_multiline_str (line 271) | def append_multiline_str(self, lines: str) -> None:
    method append_str (line 286) | def append_str(self, string: str) -> None:
    method _create_nonterminal (line 327) | def _create_nonterminal(self, name: str) -> str:
    method _add_capture_name (line 332) | def _add_capture_name(self, extractor: NonterminalExtractor) -> None:
    method choose (line 338) | def choose(self, *extractors: Extractor | str, capture_name: str = Non...
    method _add_extractor (line 358) | def _add_extractor(self, extractor_type: str, create_extractor: typing...
    method extractor (line 368) | def extractor(self, create_extractor: typing.Callable[[str], Extractor...
    method json (line 377) | def json(self, schema: typing.Type[Schema]|collections.abc.Sequence, *...
    method regex (line 405) | def regex(self, regex: str, *, capture_name: str = None) -> RegexExtra...
    method regex_complement (line 420) | def regex_complement(self, regex: str, *, capture_name: str = None) ->...
    method str (line 435) | def str(self, *, stop: typing.Union[str, list[str]] = None,
    method substr (line 457) | def substr(self, string: str, *, capture_name: str = None, extract_emp...
    method build (line 476) | def build(self, vocabulary: kbnf.Vocabulary,

FILE: src/formatron/integrations/RWKV.py
  class PIPELINE_ARGS (line 12) | class PIPELINE_ARGS(rwkv.utils.PIPELINE_ARGS):
    method __init__ (line 17) | def __init__(self,
  function create_engine_vocabulary (line 33) | def create_engine_vocabulary(WORD_NAME: str, tokenizer) -> kbnf.Vocabula...
  class PIPELINE (line 43) | class PIPELINE(rwkv.utils.PIPELINE):  # NOSONAR
    method __init__ (line 48) | def __init__(self, model, WORD_NAME, formatter_builder: FormatterBuild...
    method generate (line 57) | def generate(self, ctx, token_count=100, args=PIPELINE_ARGS(), callbac...

FILE: src/formatron/integrations/exllamav2.py
  function create_engine_vocabulary (line 16) | def create_engine_vocabulary(tokenizer: ExLlamaV2Tokenizer,
  function create_formatter_filter (line 35) | def create_formatter_filter(model: ExLlamaV2, tokenizer: ExLlamaV2Tokeni...
  class FormatterFilter (line 55) | class FormatterFilter(ExLlamaV2Filter):
    method __init__ (line 60) | def __init__(self, model, tokenizer, formatter: FormatterBase,
    method is_completed (line 72) | def is_completed(self) -> bool:
    method clone (line 78) | def clone(self, c=None) -> "FormatterFilter":
    method begin (line 90) | def begin(self, prefix_str: str) -> None:
    method reset (line 97) | def reset(self) -> None:
    method feed (line 100) | def feed(self, token: int):
    method next_set (line 107) | def next_set(self) -> typing.Tuple[typing.Set[int], typing.Set[int]]:
    method next (line 116) | def next(self) -> typing.Tuple[typing.Sequence[int], typing.Sequence[i...
    method use_background_worker (line 126) | def use_background_worker(self) -> bool:
    method can_mask_logits (line 130) | def can_mask_logits(self) -> bool:
    method prepare_logit_mask (line 133) | def prepare_logit_mask(self):
    method mask_logits (line 137) | def mask_logits(self, logits: torch.Tensor) -> torch.Tensor:
    method formatter_captures (line 148) | def formatter_captures(self) -> dict[str, typing.Any]:

FILE: src/formatron/integrations/transformers.py
  function create_engine_vocabulary (line 16) | def create_engine_vocabulary(tokenizer: PreTrainedTokenizerBase,
  function create_formatter_logits_processor (line 31) | def create_formatter_logits_processor(tokenizer: PreTrainedTokenizerBase,
  function create_formatter_logits_processor_list (line 52) | def create_formatter_logits_processor_list(tokenizer: PreTrainedTokenize...
  class FormattersLogitsProcessor (line 70) | class FormattersLogitsProcessor(LogitsProcessor):
    method __init__ (line 75) | def __init__(self, formatters: typing.Sequence[FormatterBase | None], ...
    method _get_fastest_compatible_logits_mask_fn (line 88) | def _get_fastest_compatible_logits_mask_fn():
    method reset (line 99) | def reset(self) -> None:
    method formatters_captures (line 106) | def formatters_captures(self) -> list[dict[str, typing.Any] | None]:
    method is_completed (line 114) | def is_completed(self) -> list[bool | None]:
    method __call__ (line 122) | def __call__(self, input_ids, scores):

FILE: src/formatron/integrations/utils.py
  function _multiple_replace (line 7) | def _multiple_replace(replacements: typing.Dict[bytes, bytes], regex: re...
  function default_mask_logits_fn (line 10) | def default_mask_logits_fn(bit_mask, formatter, logits):
  function get_fastest_compatible_logits_mask_fn (line 12) | def get_fastest_compatible_logits_mask_fn():
  function get_bit_mask (line 22) | def get_bit_mask(logits):
  function get_original_characters (line 26) | def get_original_characters(vocab: typing.Dict[str, int],
  function autodetect_processors (line 52) | def autodetect_processors(vocab: typing.Dict[str, int]) -> typing.List[t...
  function update_vocab_0xHH (line 69) | def update_vocab_0xHH(token_to_char: typing.Dict[bytes, bytes]):
  function update_vocab_sentencepiece (line 77) | def update_vocab_sentencepiece(token_to_char: typing.Dict[bytes, bytes]):
  function update_vocab_dot_G (line 84) | def update_vocab_dot_G(token_to_char: typing.Dict[bytes, bytes]):
  function _huggingface_bytelevel_decoder (line 92) | def _huggingface_bytelevel_decoder():

FILE: src/formatron/integrations/vllm.py
  class FormattersLogitsProcessor (line 15) | class FormattersLogitsProcessor:
    method __init__ (line 20) | def __init__(self, formatters: typing.Sequence[FormatterBase | None], ...
    method formatters_captures (line 36) | def formatters_captures(self) -> list[dict[str, typing.Any] | None]:
    method is_completed (line 39) | def is_completed(self) -> list[bool | None]:
    method reset (line 46) | def reset(self) -> None:
    method _to_next_batch_step (line 54) | def _to_next_batch_step(self):
    method __call__ (line 59) | def __call__(self, prompt, generated_tokens, logits):
  function create_engine_vocabulary (line 108) | def create_engine_vocabulary(tokenizer: AnyTokenizer,
  function create_formatters_logits_processor (line 123) | def create_formatters_logits_processor(llm: LLM,

FILE: src/formatron/schemas/dict_inference.py
  class FieldInfo (line 13) | class FieldInfo(schemas.schema.FieldInfo):
    method __init__ (line 16) | def __init__(self, annotation: typing.Type):
    method annotation (line 26) | def annotation(self) -> typing.Type[typing.Any] | None:
    method required (line 33) | def required(self) -> bool:
  function _infer_type (line 40) | def _infer_type(value: Any) -> Type[Any]:
  function infer_mapping (line 67) | def infer_mapping(mapping: collections.abc.Mapping[str, Any]) -> typing....

FILE: src/formatron/schemas/json_schema.py
  class FieldInfo (line 17) | class FieldInfo(schemas.schema.FieldInfo):
    method __init__ (line 20) | def __init__(self, annotation: typing.Type, required:bool):
    method annotation (line 32) | def annotation(self) -> typing.Type[typing.Any] | None:
    method required (line 39) | def required(self) -> bool:
  function create_schema (line 47) | def create_schema(schema: dict[str, typing.Any], registry=Registry()) ->...
  function _resolve_new_url (line 105) | def _resolve_new_url(uri: str, ref: str) -> str:
  function _validate_json_schema (line 113) | def _validate_json_schema(schema: dict[str, typing.Any]) -> None:
  function _convert_json_schema_to_our_schema (line 126) | def _convert_json_schema_to_our_schema(schema: dict[str, typing.Any], js...
  function _extract_fields_from_object_type (line 143) | def _extract_fields_from_object_type(object_type:typing.Type):
  function _handle_anyOf (line 151) | def _handle_anyOf(schema: dict[str, typing.Any], json_schema_id_to_schem...
  function _infer_type (line 159) | def _infer_type(schema: dict[str, typing.Any], json_schema_id_to_schema:...
  function _get_literal (line 194) | def _get_literal(schema: dict[str, typing.Any]) -> typing.Any:
  function _handle_literal (line 199) | def _handle_literal(literal: typing.Any, obtained_type: typing.Type, sch...
  function _handle_str_with_metadata (line 207) | def _handle_str_with_metadata(schema: dict[str, typing.Any]) -> typing.T...
  function _handle_numeric_with_metadata (line 225) | def _handle_numeric_with_metadata(schema: dict[str, typing.Any], numeric...
  function _create_custom_type (line 245) | def _create_custom_type(schema: dict[str, typing.Any], json_schema_id_to...
  function _handle_list_metadata (line 256) | def _handle_list_metadata(schema: dict[str, typing.Any], json_schema_id_...
  function _obtain_type (line 282) | def _obtain_type(schema: dict[str, typing.Any], json_schema_id_to_schema...
  function _merge_referenced_schema (line 322) | def _merge_referenced_schema(schema: dict[str, typing.Any], memo: set[in...
  function _merge_key (line 341) | def _merge_key(schema:dict[str, typing.Any], ref_key:str, reference_valu...
  function _recursive_resolve_reference (line 360) | def _recursive_resolve_reference(base_uri: str, schema: typing.Any, regi...
  function _resolve_reference (line 382) | def _resolve_reference(schema: dict[str, typing.Any], key: str, resolver...

FILE: src/formatron/schemas/pydantic.py
  class FieldInfo (line 15) | class FieldInfo(schema.FieldInfo):
    method __init__ (line 21) | def __init__(self, field: pydantic.fields.FieldInfo):
    method annotation (line 37) | def annotation(self) -> typing.Type[typing.Any] | None:
    method required (line 41) | def required(self) -> bool:
    method __repr__ (line 44) | def __repr__(self):
    method __str__ (line 47) | def __str__(self):
  class ClassSchema (line 51) | class ClassSchema(BaseModel, Schema):
    method fields (line 58) | def fields(cls) -> typing.Dict[str, typing.Any]:
    method from_json (line 65) | def from_json(cls, _json: str) -> "ClassSchema":
  function callable_schema (line 75) | def callable_schema(func: CallableT, /, *, config: ConfigDict = None, va...

FILE: src/formatron/schemas/schema.py
  class FieldInfo (line 9) | class FieldInfo(abc.ABC):
    method annotation (line 15) | def annotation(self) -> typing.Type[typing.Any] | None:
    method required (line 23) | def required(self) -> bool:
  class TypeWithMetadata (line 29) | class TypeWithMetadata:
    method __init__ (line 33) | def __init__(self, type: typing.Type[typing.Any], metadata: dict[str, ...
    method type (line 38) | def type(self) -> typing.Type[typing.Any]:
    method metadata (line 45) | def metadata(self) -> dict[str, typing.Any]|None:
  class Schema (line 51) | class Schema(abc.ABC):
    method fields (line 57) | def fields(cls) -> dict[str, FieldInfo]:
    method from_json (line 65) | def from_json(cls, json: str) -> "Schema":
  class SubstringOf (line 72) | class SubstringOf:

FILE: tests/scratch.py
  class StringLengthModel (line 5) | class StringLengthModel(BaseModel):

FILE: tests/test_exllamav2_integration.py
  function test_exllamav2_integration (line 12) | def test_exllamav2_integration(snapshot):
  function test_exllamav2_json_schema (line 41) | def test_exllamav2_json_schema(snapshot):
  function test_exllamav2_utf_8 (line 109) | def test_exllamav2_utf_8(snapshot):
  function test_exllamav2_batched_inference (line 141) | def test_exllamav2_batched_inference(snapshot):

FILE: tests/test_formatter.py
  class Test (line 13) | class Test(formatron.schemas.pydantic.ClassSchema):
  function test_formatter (line 19) | def test_formatter(snapshot):
  function test_formatter_str (line 43) | def test_formatter_str(snapshot):
  function test_formatter_substr (line 56) | def test_formatter_substr(snapshot):
  function test_formatter_dict_inference (line 70) | def test_formatter_dict_inference(snapshot):
  function test_formatter_json_schema (line 84) | def test_formatter_json_schema(snapshot):
  function test_formatter_top_level_array_json_schema (line 113) | def test_formatter_top_level_array_json_schema(snapshot):
  function test_formatter_callable_schema (line 144) | def test_formatter_callable_schema(snapshot):
  function test_grammar_literal (line 162) | def test_grammar_literal(snapshot):
  function test_formatter_alternate_accept (line 179) | def test_formatter_alternate_accept(snapshot):
  function test_formatter_regex_complement (line 203) | def test_formatter_regex_complement(snapshot):
  function test_formatter_json_no_properties (line 229) | def test_formatter_json_no_properties(snapshot):
  function test_utf8_json_key (line 248) | def test_utf8_json_key(snapshot):

FILE: tests/test_grammar_gen.py
  class Test (line 17) | class Test(formatron.schemas.pydantic.ClassSchema):
  class LinkedList (line 26) | class LinkedList(formatron.schemas.pydantic.ClassSchema):
  function test_pydantic_substring_constraint (line 31) | def test_pydantic_substring_constraint(snapshot):
  function test_json_schema_substring_constraint (line 38) | def test_json_schema_substring_constraint(snapshot):
  function test_json_schema_object_without_properties (line 56) | def test_json_schema_object_without_properties(snapshot):
  function test_pydantic_class (line 69) | def test_pydantic_class(snapshot):
  function test_pydantic_string_constraints (line 73) | def test_pydantic_string_constraints(snapshot):
  function test_pydantic_integer_constraints (line 83) | def test_pydantic_integer_constraints(snapshot):
  function test_pydantic_float_constraints (line 97) | def test_pydantic_float_constraints(snapshot):
  function test_pydantic_sequence_constraints (line 111) | def test_pydantic_sequence_constraints(snapshot):
  function test_json_schema_integer_constraints (line 125) | def test_json_schema_integer_constraints(snapshot):
  function test_json_schema_number_constraints (line 154) | def test_json_schema_number_constraints(snapshot):
  function test_json_schema_array_min_max_items_constraints (line 183) | def test_json_schema_array_min_max_items_constraints(snapshot):
  function test_json_schema_array_prefix_items (line 212) | def test_json_schema_array_prefix_items(snapshot):
  function test_json_schema (line 264) | def test_json_schema(snapshot):
  function test_recursive_binary_tree_schema (line 309) | def test_recursive_binary_tree_schema(snapshot):
  function test_recursive_linked_list_schema (line 325) | def test_recursive_linked_list_schema(snapshot):
  function test_schema_with_reference (line 342) | def test_schema_with_reference(snapshot):
  function test_schema_with_anchor_reference (line 371) | def test_schema_with_anchor_reference(snapshot):
  function test_schema_with_embedded_schema (line 398) | def test_schema_with_embedded_schema(snapshot):
  function test_schema_with_reference_to_number (line 431) | def test_schema_with_reference_to_number(snapshot):
  function test_schema_with_top_level_array (line 455) | def test_schema_with_top_level_array(snapshot):
  function test_schema_with_union_array_object (line 476) | def test_schema_with_union_array_object(snapshot):
  function test_schema_with_top_level_anyOf (line 495) | def test_schema_with_top_level_anyOf(snapshot):
  function test_schema_with_string_metadata (line 524) | def test_schema_with_string_metadata(snapshot):
  function test_schema_with_anyOf_inside_array (line 556) | def test_schema_with_anyOf_inside_array(snapshot):
  function test_pydantic_class_linked_list (line 588) | def test_pydantic_class_linked_list(snapshot):
  function test_pydantic_callable (line 593) | def test_pydantic_callable(snapshot):
  function test_infer_mapping (line 602) | def test_infer_mapping(snapshot):
  function test_schema_with_dynamic_ref (line 615) | def test_schema_with_dynamic_ref(snapshot):

FILE: tests/test_rwkv_integration.py
  function test_rwkv_integration (line 7) | def test_rwkv_integration(snapshot):

FILE: tests/test_transformers_integration.py
  function test_transformers_integration (line 6) | def test_transformers_integration(snapshot):
  function test_transformers_batched_inference (line 20) | def test_transformers_batched_inference(snapshot):
  function test_transformers_sparse_formatters (line 39) | def test_transformers_sparse_formatters(snapshot):
  function test_transformers_text_generation_pipeline (line 61) | def test_transformers_text_generation_pipeline(snapshot):

FILE: tests/test_vllm_integration.py
  function test_vllm_integration (line 10) | def test_vllm_integration(snapshot):
  function test_vllm_integration_sparse (line 36) | def test_vllm_integration_sparse(snapshot):