nemotron.gotmpl 773 B

123456789101112131415161718192021222324252627282930313233
  1. {{- if (or .Tools .System) }}<extra_id_0>System
  2. {{ if .System }}{{ .System }}
  3. {{ end }}
  4. {{- if .Tools }}
  5. {{- range .Tools }}<tool> {{ . }} </tool>{{ end }}
  6. {{ end }}
  7. {{- end }}
  8. {{- range $i, $m := .Messages }}
  9. {{- $last := eq (len (slice $.Messages $i)) 1 -}}
  10. {{- if eq .Role "user" }}<extra_id_1>User
  11. {{ .Content }}
  12. {{- if $last }}
  13. <extra_id_1>Assistant
  14. {{- end }}
  15. {{ else if eq .Role "tool" }}<extra_id_1>Tool
  16. {{ .Content }}
  17. {{- if $last }}
  18. <extra_id_1>Assistant
  19. {{- end }}
  20. {{ else if eq .Role "assistant" }}<extra_id_1>Assistant
  21. {{- if .ToolCalls }}
  22. {{ range .ToolCalls }}<toolcall> {"name": "{{ .Function.Name }}", "arguments": {{ .Function.Arguments }}} </toolcall> {{ end }}
  23. {{ else }}
  24. {{ .Content }}
  25. {{- if not $last }}
  26. {{ end }}
  27. {{- end }}
  28. {{- end }}
  29. {{- end }}