baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #799 from deichbewohner/jsonmarshaling

Browse source 
Fix JSON Marshal Escaping for Special Characters
This commit is contained in:
Michael Yang 2023-10-17 08:46:02 -07:00 • committed by GitHub
commit f55bdb6f10
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
llm/llama.go
View file

@ -541,12 +541,16 @@ func (llm *llama) Predict(ctx context.Context, prevContext []int, prompt string,
Stop: llm.Stop,
}
data, err := json.Marshal(predReq)
if err != nil {
return fmt.Errorf("error marshaling data: %v", err)
// Handling JSON marshaling with special characters unescaped.
buffer := &bytes.Buffer{}
enc := json.NewEncoder(buffer)
enc.SetEscapeHTML(false)
if err := enc.Encode(predReq); err != nil {
return fmt.Errorf("failed to marshal data: %v", err)
}
req, err := http.NewRequestWithContext(ctx, http.MethodPost, endpoint, bytes.NewBuffer(data))
req, err := http.NewRequestWithContext(ctx, http.MethodPost, endpoint, buffer)
if err != nil {
return fmt.Errorf("error creating POST request: %v", err)
}