Add starcoder template support

r2ai/interpreter.py

@@ -105,6 +105,8 @@ def messages_to_prompt(self, messages):
     formatted_messages = template_q4im(self, messages)
   elif "gemma" in lowermodel:
     formatted_messages = template_gemma(self, messages)
+  elif "starcoder" in lowermodel:
+    formatted_messages = template_starcoder(self, messages)
   elif "openchat" in lowermodel:
     formatted_messages = template_openchat(self, messages)
   elif "ferret" in lowermodel:
@@ -281,6 +283,28 @@ def template_phi3(self,messages):
   q += f"<|assistant|>\n"
   return q
 
+def template_starcoder(self,messages):
+  self.terminator = "<|im_end|>"
+  self.terminator = "##"
+  system_prompt = self.system_message # messages[0]['content'].strip()
+  if system_prompt != "":
+      q = f"<|system|>\n{system_prompt}</s>\n"
+  else:
+      q = f""
+  for index, item in enumerate(messages):
+      role = item['role']
+      content = item['content']
+      if role == 'user':
+          q += f"## Question\n{content}"
+      elif role == "hint":
+          q += f"knowledge: {content}\n"
+      elif role == 'function':
+          q += f"## Question\n{content} "
+      elif role == 'assistant' and self.env["chat.reply"] == "true":
+          q += f"## Solution\n{content}\n"
+  q += f"## Solution\n"
+  return q
+
 def template_zephyr(self,messages):
 #<|system|>
 #{system_message}</s>