openai: role tags for 'chatgpt' shorthand

eth-sri · Sep 8, 2023 · 3555b3b · 3555b3b
1 parent 6655b17
commit 3555b3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"<lmql:(.*?)\/>" tags with ((\1))
             s = re.sub(r"<lmql:(.*?)\/>", r"((\1)):", s)
         s = unescape_qstring(s)