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

Fix infinite keep_alive (#2480)

Browse source
This commit is contained in:
bnorick 2024-02-13 15:40:32 -08:00 committed by GitHub
parent 1d263449ff
commit caf2b13c10
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
api/types.go
View file

@ -415,8 +415,7 @@ func (d *Duration) UnmarshalJSON(b []byte) (err error) {
switch t := v.(type) { switch t := v.(type) {
case float64: case float64:
if t < 0 { if t < 0 {
t = math.MaxFloat64 d.Duration = time.Duration(math.MaxInt64)
d.Duration = time.Duration(t)
} else { } else {
d.Duration = time.Duration(t * float64(time.Second)) d.Duration = time.Duration(t * float64(time.Second))
} }
@ -426,8 +425,7 @@ func (d *Duration) UnmarshalJSON(b []byte) (err error) {
return err return err
} }
if d.Duration < 0 { if d.Duration < 0 {
mf := math.MaxFloat64 d.Duration = time.Duration(math.MaxInt64)
d.Duration = time.Duration(mf)
} }
} }