update tests

This commit is contained in:
Michael Yang
2025-09-22 12:57:30 -07:00
parent b846eacf42
commit ffaf2e7916
11 changed files with 323 additions and 276 deletions

View File

@@ -80,7 +80,7 @@ func TestGenerateDebugRenderOnly(t *testing.T) {
w := createRequest(t, s.CreateHandler, api.CreateRequest{
Model: "test-model",
Files: map[string]string{"file.gguf": digest},
Files: []api.File{{Name: "file.gguf", Digest: digest}},
Template: "{{ .Prompt }}",
Stream: &stream,
})
@@ -273,7 +273,7 @@ func TestChatDebugRenderOnly(t *testing.T) {
w := createRequest(t, s.CreateHandler, api.CreateRequest{
Model: "test-model",
Files: map[string]string{"file.gguf": digest},
Files: []api.File{{Name: "file.gguf", Digest: digest}},
Template: "{{ if .Tools }}{{ .Tools }}{{ end }}{{ range .Messages }}{{ .Role }}: {{ .Content }}\n{{ end }}",
Stream: &stream,
})