highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-04-11 21:29:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

llm: consider null format same as empty value

Browse Source
This commit is contained in:
jmorganca 2024-12-17 09:16:01 -08:00
parent 2cde4b8817
commit 04314765f2
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llm/server.go
View File

@ -701,7 +701,7 @@ func (s *llmServer) Completion(ctx context.Context, req CompletionRequest, fn fu
if len(req.Format) > 0 {
switch {
case bytes.Equal(req.Format, []byte(`""`)):
case bytes.Equal(req.Format, []byte(`""`)) || bytes.Equal(req.Format, []byte(`null`)):
// fallthrough
case bytes.Equal(req.Format, []byte(`"json"`)):
request["grammar"] = grammarJSON

2
llm/server_test.go
View File

@ -46,7 +46,7 @@ func TestLLMServerCompletionFormat(t *testing.T) {
}
}
valids := []string{`"json"`, `{"type":"object"}`, ``, `""`}
valids := []string{`"json"`, `{"type":"object"}`, ``, `""`, `null`}
for _, valid := range valids {
err := s.Completion(ctx, CompletionRequest{
Options: new(api.Options),