25 lines
783 B
Plaintext
25 lines
783 B
Plaintext
|
TEMPLATE """{{- if .Messages }}
|
||
|
|
{{- if .System }}<|im_start|>system
|
||
|
|
{{ .System }}<|im_end|>
|
||
|
|
{{ end }}
|
||
|
|
{{- range $i, $_ := .Messages }}
|
||
|
|
{{- $last := eq (len (slice $.Messages $i)) 1 -}}
|
||
|
|
{{- if eq .Role "user" }}<|im_start|>user
|
||
|
|
{{ .Content }}<|im_end|>
|
||
|
|
{{ else if eq .Role "assistant" }}<|im_start|>assistant
|
||
|
|
{{ .Content }}{{ if not $last }}<|im_end|>
|
||
|
|
{{ end }}
|
||
|
|
{{- end }}
|
||
|
|
{{- if and (ne .Role "assistant") $last }}<|im_start|>assistant
|
||
|
|
{{ end }}
|
||
|
|
{{- end }}
|
||
|
|
{{- else }}
|
||
|
|
{{- if .System }}<|im_start|>system
|
||
|
|
{{ .System }}<|im_end|>
|
||
|
|
{{ end }}{{ if .Prompt }}<|im_start|>user
|
||
|
|
{{ .Prompt }}<|im_end|>
|
||
|
|
{{ end }}<|im_start|>assistant
|
||
|
|
{{ end }}{{ .Response }}{{ if .Response }}<|im_end|>{{ end }}"""
|
||
|
|
SYSTEM You are a helpful AI assistant.
|
||
|
|
PARAMETER stop <|im_start|>
|
||
|
|
PARAMETER stop <|im_end|>
|