diff --git a/src/heretic/main.py b/src/heretic/main.py index 924adbb..3f8d195 100644 --- a/src/heretic/main.py +++ b/src/heretic/main.py @@ -291,6 +291,12 @@ def run(): elif model.response_prefix.startswith("<|channel|>analysis<|message|>"): # gpt-oss. model.response_prefix = "<|channel|>analysis<|message|><|end|><|start|>assistant<|channel|>final<|message|>" + elif model.response_prefix.startswith(""): + # Unknown, suggested by user. + model.response_prefix = "" + elif model.response_prefix.startswith("[THINK]"): + # Unknown, suggested by user. + model.response_prefix = "[THINK][/THINK]" if model.response_prefix: print(f"* Prefix found: [bold]{model.response_prefix!r}[/]")