LocalAI/pkg/functions
Ettore Di Giacinto 491e1d752b
feat(functions): relax mixedgrammars (#2365)
* feat(functions): relax mixedgrammars

Extend even more the functionalities and when mixed mode is enabled,
tolerate also both strings and JSON in the result - in this case we make
sure that the JSON can be correctly parsed.

This also updates the examples and the gallery model to configure the
grammar.

The changeset also breaks current function/grammar configuration as it
reserves now a stanza in the YAML config.

For example:

```yaml
function:
  grammar:
    # This allows the grammar to also return messages
    mixed_mode: true
    # Suffix to add to the grammar
    # prefix: '<tool_call>\n'
    # Force parallel calls in the grammar
    # parallel_calls: true
```

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

* refactor, add a way to disable mixed json and freestring

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

* Fix linting issues

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

---------

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
2024-05-22 00:14:16 +02:00
..
functions_suite_test.go feat(functions): support models with no grammar, add tests (#2068) 2024-04-18 22:43:12 +02:00
functions_test.go feat(grammar): support models with specific construct (#2291) 2024-05-12 01:13:22 +02:00
functions.go feat(grammar): support models with specific construct (#2291) 2024-05-12 01:13:22 +02:00
grammar_json_schema_test.go feat(functions): relax mixedgrammars (#2365) 2024-05-22 00:14:16 +02:00
grammar_json_schema.go feat(functions): relax mixedgrammars (#2365) 2024-05-22 00:14:16 +02:00
options.go feat(functions): relax mixedgrammars (#2365) 2024-05-22 00:14:16 +02:00
parse_test.go feat(functions): relax mixedgrammars (#2365) 2024-05-22 00:14:16 +02:00
parse.go feat(functions): relax mixedgrammars (#2365) 2024-05-22 00:14:16 +02:00