commit
567e74e7d7
1 changed files with 25 additions and 30 deletions
|
@ -58,9 +58,6 @@ func generate(c *gin.Context) {
|
||||||
req.Model = path.Join(cacheDir(), "models", req.Model+".bin")
|
req.Model = path.Join(cacheDir(), "models", req.Model+".bin")
|
||||||
}
|
}
|
||||||
|
|
||||||
ch := make(chan any)
|
|
||||||
go stream(c, ch)
|
|
||||||
|
|
||||||
templateNames := make([]string, 0, len(templates.Templates()))
|
templateNames := make([]string, 0, len(templates.Templates()))
|
||||||
for _, template := range templates.Templates() {
|
for _, template := range templates.Templates() {
|
||||||
templateNames = append(templateNames, template.Name())
|
templateNames = append(templateNames, template.Name())
|
||||||
|
@ -84,7 +81,10 @@ func generate(c *gin.Context) {
|
||||||
}
|
}
|
||||||
defer llm.Close()
|
defer llm.Close()
|
||||||
|
|
||||||
fn := func(r api.GenerateResponse) {
|
ch := make(chan any)
|
||||||
|
go func() {
|
||||||
|
defer close(ch)
|
||||||
|
llm.Predict(req.Context, req.Prompt, func(r api.GenerateResponse) {
|
||||||
r.Model = req.Model
|
r.Model = req.Model
|
||||||
r.CreatedAt = time.Now().UTC()
|
r.CreatedAt = time.Now().UTC()
|
||||||
if r.Done {
|
if r.Done {
|
||||||
|
@ -92,13 +92,10 @@ func generate(c *gin.Context) {
|
||||||
}
|
}
|
||||||
|
|
||||||
ch <- r
|
ch <- r
|
||||||
}
|
})
|
||||||
|
}()
|
||||||
if err := llm.Predict(req.Context, req.Prompt, fn); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
|
streamResponse(c, ch)
|
||||||
}
|
}
|
||||||
|
|
||||||
func pull(c *gin.Context) {
|
func pull(c *gin.Context) {
|
||||||
|
@ -133,20 +130,18 @@ func pull(c *gin.Context) {
|
||||||
}
|
}
|
||||||
|
|
||||||
ch := make(chan any)
|
ch := make(chan any)
|
||||||
go stream(c, ch)
|
go func() {
|
||||||
|
defer close(ch)
|
||||||
fn := func(total, completed int64) {
|
saveModel(remote, func(total, completed int64) {
|
||||||
ch <- api.PullProgress{
|
ch <- api.PullProgress{
|
||||||
Total: total,
|
Total: total,
|
||||||
Completed: completed,
|
Completed: completed,
|
||||||
Percent: float64(completed) / float64(total) * 100,
|
Percent: float64(completed) / float64(total) * 100,
|
||||||
}
|
}
|
||||||
}
|
})
|
||||||
|
}()
|
||||||
|
|
||||||
if err := saveModel(remote, fn); err != nil {
|
streamResponse(c, ch)
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func Serve(ln net.Listener) error {
|
func Serve(ln net.Listener) error {
|
||||||
|
@ -179,7 +174,7 @@ func matchRankOne(source string, targets []string) (bestMatch string, bestRank i
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func stream(c *gin.Context, ch chan any) {
|
func streamResponse(c *gin.Context, ch chan any) {
|
||||||
c.Stream(func(w io.Writer) bool {
|
c.Stream(func(w io.Writer) bool {
|
||||||
val, ok := <-ch
|
val, ok := <-ch
|
||||||
if !ok {
|
if !ok {
|
||||||
|
|
Loading…
Reference in a new issue