123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852 |
- package server
- import (
- "context"
- "crypto/sha256"
- "encoding/json"
- "errors"
- "fmt"
- "io"
- "io/fs"
- "log"
- "net"
- "net/http"
- "os"
- "os/signal"
- "path/filepath"
- "reflect"
- "runtime"
- "strconv"
- "strings"
- "sync"
- "syscall"
- "time"
- "github.com/gin-contrib/cors"
- "github.com/gin-gonic/gin"
- "github.com/jmorganca/ollama/api"
- "github.com/jmorganca/ollama/llm"
- "github.com/jmorganca/ollama/parser"
- "github.com/jmorganca/ollama/version"
- )
- var mode string = gin.DebugMode
- func init() {
- switch mode {
- case gin.DebugMode:
- case gin.ReleaseMode:
- case gin.TestMode:
- default:
- mode = gin.DebugMode
- }
- gin.SetMode(mode)
- }
- var loaded struct {
- mu sync.Mutex
- runner llm.LLM
- expireAt time.Time
- expireTimer *time.Timer
- *Model
- *api.Options
- }
- var defaultSessionDuration = 5 * time.Minute
- // load a model into memory if it is not already loaded, it is up to the caller to lock loaded.mu before calling this function
- func load(ctx context.Context, workDir string, model *Model, reqOpts map[string]interface{}, sessionDuration time.Duration) error {
- opts := api.DefaultOptions()
- if err := opts.FromMap(model.Options); err != nil {
- log.Printf("could not load model options: %v", err)
- return err
- }
- if err := opts.FromMap(reqOpts); err != nil {
- return err
- }
- // check if the loaded model is still running in a subprocess, in case something unexpected happened
- if loaded.runner != nil {
- if err := loaded.runner.Ping(ctx); err != nil {
- log.Print("loaded llm process not responding, closing now")
- // the subprocess is no longer running, so close it
- loaded.runner.Close()
- loaded.runner = nil
- loaded.Model = nil
- loaded.Options = nil
- }
- }
- needLoad := loaded.runner == nil || // is there a model loaded?
- loaded.ModelPath != model.ModelPath || // has the base model changed?
- !reflect.DeepEqual(loaded.AdapterPaths, model.AdapterPaths) || // have the adapters changed?
- !reflect.DeepEqual(loaded.Options.Runner, opts.Runner) // have the runner options changed?
- if needLoad {
- if loaded.runner != nil {
- log.Println("changing loaded model")
- loaded.runner.Close()
- loaded.runner = nil
- loaded.Model = nil
- loaded.Options = nil
- }
- llmRunner, err := llm.New(workDir, model.ModelPath, model.AdapterPaths, opts)
- if err != nil {
- // some older models are not compatible with newer versions of llama.cpp
- // show a generalized compatibility error until there is a better way to
- // check for model compatibility
- if strings.Contains(err.Error(), "failed to load model") {
- err = fmt.Errorf("%v: this model may be incompatible with your version of Ollama. If you previously pulled this model, try updating it by running `ollama pull %s`", err, model.ShortName)
- }
- return err
- }
- loaded.Model = model
- loaded.runner = llmRunner
- loaded.Options = &opts
- }
- // update options for the loaded llm
- // TODO(mxyng): this isn't thread safe, but it should be fine for now
- loaded.runner.SetOptions(opts)
- loaded.expireAt = time.Now().Add(sessionDuration)
- if loaded.expireTimer == nil {
- loaded.expireTimer = time.AfterFunc(sessionDuration, func() {
- loaded.mu.Lock()
- defer loaded.mu.Unlock()
- if time.Now().Before(loaded.expireAt) {
- return
- }
- if loaded.runner != nil {
- loaded.runner.Close()
- }
- loaded.runner = nil
- loaded.Model = nil
- loaded.Options = nil
- })
- }
- loaded.expireTimer.Reset(sessionDuration)
- return nil
- }
- func GenerateHandler(c *gin.Context) {
- loaded.mu.Lock()
- defer loaded.mu.Unlock()
- checkpointStart := time.Now()
- var req api.GenerateRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- // validate the request
- switch {
- case req.Model == "":
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "model is required"})
- return
- case len(req.Format) > 0 && req.Format != "json":
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "format must be json"})
- return
- case req.Raw && (req.Template != "" || req.System != "" || len(req.Context) > 0):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "raw mode does not support template, system, or context"})
- return
- }
- model, err := GetModel(req.Model)
- if err != nil {
- var pErr *fs.PathError
- if errors.As(err, &pErr) {
- c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found, try pulling it first", req.Model)})
- return
- }
- c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- workDir := c.GetString("workDir")
- // TODO: set this duration from the request if specified
- sessionDuration := defaultSessionDuration
- if err := load(c.Request.Context(), workDir, model, req.Options, sessionDuration); err != nil {
- if errors.Is(err, api.ErrInvalidOpts) {
- c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- checkpointLoaded := time.Now()
- prompt := req.Prompt
- if !req.Raw {
- prompt, err = model.Prompt(req)
- if err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- }
- ch := make(chan any)
- go func() {
- defer close(ch)
- // an empty request loads the model
- if req.Prompt == "" && req.Template == "" && req.System == "" {
- ch <- api.GenerateResponse{CreatedAt: time.Now().UTC(), Model: req.Model, Done: true}
- return
- }
- fn := func(r api.GenerateResponse) {
- loaded.expireAt = time.Now().Add(sessionDuration)
- loaded.expireTimer.Reset(sessionDuration)
- r.Model = req.Model
- r.CreatedAt = time.Now().UTC()
- if r.Done {
- r.TotalDuration = time.Since(checkpointStart)
- r.LoadDuration = checkpointLoaded.Sub(checkpointStart)
- }
- if req.Raw {
- // in raw mode the client must manage history on their own
- r.Context = nil
- }
- ch <- r
- }
- if err := loaded.runner.Predict(c.Request.Context(), req.Context, prompt, req.Format, fn); err != nil {
- ch <- gin.H{"error": err.Error()}
- }
- }()
- if req.Stream != nil && !*req.Stream {
- var response api.GenerateResponse
- generated := ""
- for resp := range ch {
- if r, ok := resp.(api.GenerateResponse); ok {
- generated += r.Response
- response = r
- } else {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- }
- response.Response = generated
- c.JSON(http.StatusOK, response)
- return
- }
- streamResponse(c, ch)
- }
- func EmbeddingHandler(c *gin.Context) {
- loaded.mu.Lock()
- defer loaded.mu.Unlock()
- var req api.EmbeddingRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Model == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "model is required"})
- return
- }
- model, err := GetModel(req.Model)
- if err != nil {
- c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- workDir := c.GetString("workDir")
- if err := load(c.Request.Context(), workDir, model, req.Options, 5*time.Minute); err != nil {
- c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if !loaded.Options.EmbeddingOnly {
- c.JSON(http.StatusBadRequest, gin.H{"error": "embedding option must be set to true"})
- return
- }
- embedding, err := loaded.runner.Embedding(c.Request.Context(), req.Prompt)
- if err != nil {
- log.Printf("embedding generation failed: %v", err)
- c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to generate embedding"})
- return
- }
- resp := api.EmbeddingResponse{
- Embedding: embedding,
- }
- c.JSON(http.StatusOK, resp)
- }
- func PullModelHandler(c *gin.Context) {
- var req api.PullRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Name == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "name is required"})
- return
- }
- ch := make(chan any)
- go func() {
- defer close(ch)
- fn := func(r api.ProgressResponse) {
- ch <- r
- }
- regOpts := &RegistryOptions{
- Insecure: req.Insecure,
- }
- ctx, cancel := context.WithCancel(c.Request.Context())
- defer cancel()
- if err := PullModel(ctx, req.Name, regOpts, fn); err != nil {
- ch <- gin.H{"error": err.Error()}
- }
- }()
- if req.Stream != nil && !*req.Stream {
- waitForStream(c, ch)
- return
- }
- streamResponse(c, ch)
- }
- func PushModelHandler(c *gin.Context) {
- var req api.PushRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Name == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "name is required"})
- return
- }
- ch := make(chan any)
- go func() {
- defer close(ch)
- fn := func(r api.ProgressResponse) {
- ch <- r
- }
- regOpts := &RegistryOptions{
- Insecure: req.Insecure,
- }
- ctx, cancel := context.WithCancel(c.Request.Context())
- defer cancel()
- if err := PushModel(ctx, req.Name, regOpts, fn); err != nil {
- ch <- gin.H{"error": err.Error()}
- }
- }()
- if req.Stream != nil && !*req.Stream {
- waitForStream(c, ch)
- return
- }
- streamResponse(c, ch)
- }
- func CreateModelHandler(c *gin.Context) {
- var req api.CreateRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Name == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "name is required"})
- return
- }
- if req.Path == "" && req.Modelfile == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "path or modelfile are required"})
- return
- }
- var modelfile io.Reader = strings.NewReader(req.Modelfile)
- if req.Path != "" && req.Modelfile == "" {
- mf, err := os.Open(req.Path)
- if err != nil {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("error reading modelfile: %s", err)})
- return
- }
- defer mf.Close()
- modelfile = mf
- }
- commands, err := parser.Parse(modelfile)
- if err != nil {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- ch := make(chan any)
- go func() {
- defer close(ch)
- fn := func(resp api.ProgressResponse) {
- ch <- resp
- }
- ctx, cancel := context.WithCancel(c.Request.Context())
- defer cancel()
- if err := CreateModel(ctx, req.Name, filepath.Dir(req.Path), commands, fn); err != nil {
- ch <- gin.H{"error": err.Error()}
- }
- }()
- if req.Stream != nil && !*req.Stream {
- waitForStream(c, ch)
- return
- }
- streamResponse(c, ch)
- }
- func DeleteModelHandler(c *gin.Context) {
- var req api.DeleteRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Name == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "name is required"})
- return
- }
- if err := DeleteModel(req.Name); err != nil {
- if os.IsNotExist(err) {
- c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found", req.Name)})
- } else {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- }
- return
- }
- manifestsPath, err := GetManifestPath()
- if err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- if err := PruneDirectory(manifestsPath); err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- c.JSON(http.StatusOK, nil)
- }
- func ShowModelHandler(c *gin.Context) {
- var req api.ShowRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Name == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "name is required"})
- return
- }
- resp, err := GetModelInfo(req.Name)
- if err != nil {
- if os.IsNotExist(err) {
- c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found", req.Name)})
- } else {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- }
- return
- }
- c.JSON(http.StatusOK, resp)
- }
- func GetModelInfo(name string) (*api.ShowResponse, error) {
- model, err := GetModel(name)
- if err != nil {
- return nil, err
- }
- resp := &api.ShowResponse{
- License: strings.Join(model.License, "\n"),
- System: model.System,
- Template: model.Template,
- }
- mf, err := ShowModelfile(model)
- if err != nil {
- return nil, err
- }
- resp.Modelfile = mf
- var params []string
- cs := 30
- for k, v := range model.Options {
- switch val := v.(type) {
- case string:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, val))
- case int:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, strconv.Itoa(val)))
- case float64:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, strconv.FormatFloat(val, 'f', 0, 64)))
- case bool:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, strconv.FormatBool(val)))
- case []interface{}:
- for _, nv := range val {
- switch nval := nv.(type) {
- case string:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, nval))
- case int:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, strconv.Itoa(nval)))
- case float64:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, strconv.FormatFloat(nval, 'f', 0, 64)))
- case bool:
- params = append(params, fmt.Sprintf("%-*s %s", cs, k, strconv.FormatBool(nval)))
- }
- }
- }
- }
- resp.Parameters = strings.Join(params, "\n")
- return resp, nil
- }
- func ListModelsHandler(c *gin.Context) {
- models := make([]api.ModelResponse, 0)
- fp, err := GetManifestPath()
- if err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- walkFunc := func(path string, info os.FileInfo, _ error) error {
- if !info.IsDir() {
- dir, file := filepath.Split(path)
- dir = strings.Trim(strings.TrimPrefix(dir, fp), string(os.PathSeparator))
- tag := strings.Join([]string{dir, file}, ":")
- mp := ParseModelPath(tag)
- manifest, digest, err := GetManifest(mp)
- if err != nil {
- log.Printf("skipping file: %s", fp)
- return nil
- }
- models = append(models, api.ModelResponse{
- Name: mp.GetShortTagname(),
- Size: manifest.GetTotalSize(),
- Digest: digest,
- ModifiedAt: info.ModTime(),
- })
- }
- return nil
- }
- if err := filepath.Walk(fp, walkFunc); err != nil {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- c.JSON(http.StatusOK, api.ListResponse{Models: models})
- }
- func CopyModelHandler(c *gin.Context) {
- var req api.CopyRequest
- err := c.ShouldBindJSON(&req)
- switch {
- case errors.Is(err, io.EOF):
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "missing request body"})
- return
- case err != nil:
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if req.Source == "" || req.Destination == "" {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "source add destination are required"})
- return
- }
- if err := CopyModel(req.Source, req.Destination); err != nil {
- if os.IsNotExist(err) {
- c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found", req.Source)})
- } else {
- c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- }
- return
- }
- }
- func HeadBlobHandler(c *gin.Context) {
- path, err := GetBlobsPath(c.Param("digest"))
- if err != nil {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
- return
- }
- if _, err := os.Stat(path); err != nil {
- c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("blob %q not found", c.Param("digest"))})
- return
- }
- c.Status(http.StatusOK)
- }
- func CreateBlobHandler(c *gin.Context) {
- targetPath, err := GetBlobsPath(c.Param("digest"))
- if err != nil {
- c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- hash := sha256.New()
- temp, err := os.CreateTemp(filepath.Dir(targetPath), c.Param("digest")+"-")
- if err != nil {
- c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- defer temp.Close()
- defer os.Remove(temp.Name())
- if _, err := io.Copy(temp, io.TeeReader(c.Request.Body, hash)); err != nil {
- c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- if fmt.Sprintf("sha256:%x", hash.Sum(nil)) != c.Param("digest") {
- c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "digest does not match body"})
- return
- }
- if err := temp.Close(); err != nil {
- c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- if err := os.Rename(temp.Name(), targetPath); err != nil {
- c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
- return
- }
- c.Status(http.StatusCreated)
- }
- var defaultAllowOrigins = []string{
- "localhost",
- "127.0.0.1",
- "0.0.0.0",
- }
- func Serve(ln net.Listener, allowOrigins []string) error {
- if noprune := os.Getenv("OLLAMA_NOPRUNE"); noprune == "" {
- // clean up unused layers and manifests
- if err := PruneLayers(); err != nil {
- return err
- }
- manifestsPath, err := GetManifestPath()
- if err != nil {
- return err
- }
- if err := PruneDirectory(manifestsPath); err != nil {
- return err
- }
- }
- config := cors.DefaultConfig()
- config.AllowWildcard = true
- config.AllowOrigins = allowOrigins
- for _, allowOrigin := range defaultAllowOrigins {
- config.AllowOrigins = append(config.AllowOrigins,
- fmt.Sprintf("http://%s", allowOrigin),
- fmt.Sprintf("https://%s", allowOrigin),
- fmt.Sprintf("http://%s:*", allowOrigin),
- fmt.Sprintf("https://%s:*", allowOrigin),
- )
- }
- workDir, err := os.MkdirTemp("", "ollama")
- if err != nil {
- return err
- }
- defer os.RemoveAll(workDir)
- r := gin.Default()
- r.Use(
- cors.New(config),
- func(c *gin.Context) {
- c.Set("workDir", workDir)
- c.Next()
- },
- )
- r.POST("/api/pull", PullModelHandler)
- r.POST("/api/generate", GenerateHandler)
- r.POST("/api/embeddings", EmbeddingHandler)
- r.POST("/api/create", CreateModelHandler)
- r.POST("/api/push", PushModelHandler)
- r.POST("/api/copy", CopyModelHandler)
- r.DELETE("/api/delete", DeleteModelHandler)
- r.POST("/api/show", ShowModelHandler)
- r.POST("/api/blobs/:digest", CreateBlobHandler)
- r.HEAD("/api/blobs/:digest", HeadBlobHandler)
- for _, method := range []string{http.MethodGet, http.MethodHead} {
- r.Handle(method, "/", func(c *gin.Context) {
- c.String(http.StatusOK, "Ollama is running")
- })
- r.Handle(method, "/api/tags", ListModelsHandler)
- }
- log.Printf("Listening on %s (version %s)", ln.Addr(), version.Version)
- s := &http.Server{
- Handler: r,
- }
- // listen for a ctrl+c and stop any loaded llm
- signals := make(chan os.Signal, 1)
- signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
- go func() {
- <-signals
- if loaded.runner != nil {
- loaded.runner.Close()
- }
- os.RemoveAll(workDir)
- os.Exit(0)
- }()
- if runtime.GOOS == "linux" {
- // check compatibility to log warnings
- if _, err := llm.CheckVRAM(); err != nil {
- log.Printf(err.Error())
- }
- }
- return s.Serve(ln)
- }
- func waitForStream(c *gin.Context, ch chan interface{}) {
- c.Header("Content-Type", "application/json")
- for resp := range ch {
- switch r := resp.(type) {
- case api.ProgressResponse:
- if r.Status == "success" {
- c.JSON(http.StatusOK, r)
- return
- }
- case gin.H:
- if errorMsg, ok := r["error"].(string); ok {
- c.JSON(http.StatusInternalServerError, gin.H{"error": errorMsg})
- return
- } else {
- c.JSON(http.StatusInternalServerError, gin.H{"error": "unexpected error format in progress response"})
- return
- }
- default:
- c.JSON(http.StatusInternalServerError, gin.H{"error": "unexpected progress response"})
- return
- }
- }
- c.JSON(http.StatusInternalServerError, gin.H{"error": "unexpected end of progress response"})
- }
- func streamResponse(c *gin.Context, ch chan any) {
- c.Header("Content-Type", "application/x-ndjson")
- c.Stream(func(w io.Writer) bool {
- val, ok := <-ch
- if !ok {
- return false
- }
- bts, err := json.Marshal(val)
- if err != nil {
- log.Printf("streamResponse: json.Marshal failed with %s", err)
- return false
- }
- // Delineate chunks with new-line delimiter
- bts = append(bts, '\n')
- if _, err := w.Write(bts); err != nil {
- log.Printf("streamResponse: w.Write failed with %s", err)
- return false
- }
- return true
- })
- }
|