Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix ollama stream response unmarshall error

Browse Source 
fix ollama stream response body unmarshall error
This commit is contained in:
glc-coder 2024-07-31 00:01:12 +08:00 committed by GitHub
parent c936198ac8
commit 3db8eb6de3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
relay/adaptor/ollama/main.go
View File

@ -119,7 +119,9 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
for scanner.Scan() { for scanner.Scan() {
data := strings.TrimPrefix(scanner.Text(), "}") data := strings.TrimPrefix(scanner.Text(), "}")
data = data + "}" if !strings.HasSuffix(data, "}") {
data = data + "}"
}
var ollamaResponse ChatResponse var ollamaResponse ChatResponse
err := json.Unmarshal([]byte(data), &ollamaResponse) err := json.Unmarshal([]byte(data), &ollamaResponse)