From 3555b3bdc0d8957b8cc5b35b368031b53758fc66 Mon Sep 17 00:00:00 2001 From: Luca Beurer-Kellner Date: Fri, 8 Sep 2023 22:34:58 +0200 Subject: [PATCH] openai: role tags for 'chatgpt' shorthand --- src/lmql/runtime/interpreter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lmql/runtime/interpreter.py b/src/lmql/runtime/interpreter.py index 619c9db2..23d111cf 100644 --- a/src/lmql/runtime/interpreter.py +++ b/src/lmql/runtime/interpreter.py @@ -453,7 +453,7 @@ async def continue_for_more_prompt_stmts(): ) def process_query_string(self, s: str): - if not ("turbo" in self.model_identifier or "gpt-4" in self.model_identifier): + if not ("turbo" in self.model_identifier or "gpt-4" in self.model_identifier or "chatgpt" in self.model_identifier): # replace all r"" tags with ((\1)) s = re.sub(r"", r"((\1)):", s) s = unescape_qstring(s)