Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ollama
Commits
a38d88d8
Commit
a38d88d8
authored
Jan 11, 2024
by
Michael Yang
Browse files
api: add model for all requests
prefer using req.Model and fallback to req.Name
parent
abec7f06
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
59 additions
and
26 deletions
+59
-26
api/types.go
api/types.go
+18
-4
server/routes.go
server/routes.go
+41
-22
No files found.
api/types.go
View file @
a38d88d8
...
...
@@ -137,23 +137,31 @@ type EmbeddingResponse struct {
}
type
CreateRequest
struct
{
Name
string
`json:"
name
"`
Model
string
`json:"
model
"`
Path
string
`json:"path"`
Modelfile
string
`json:"modelfile"`
Stream
*
bool
`json:"stream,omitempty"`
// Name is deprecated, see Model
Name
string
`json:"name"`
}
type
DeleteRequest
struct
{
Model
string
`json:"model"`
// Name is deprecated, see Model
Name
string
`json:"name"`
}
type
ShowRequest
struct
{
Name
string
`json:"name"`
Model
string
`json:"model"`
System
string
`json:"system"`
Template
string
`json:"template"`
Options
map
[
string
]
interface
{}
`json:"options"`
// Name is deprecated, see Model
Name
string
`json:"name"`
}
type
ShowResponse
struct
{
...
...
@@ -171,11 +179,14 @@ type CopyRequest struct {
}
type
PullRequest
struct
{
Name
string
`json:"
name
"`
Model
string
`json:"
model
"`
Insecure
bool
`json:"insecure,omitempty"`
Username
string
`json:"username"`
Password
string
`json:"password"`
Stream
*
bool
`json:"stream,omitempty"`
// Name is deprecated, see Model
Name
string
`json:"name"`
}
type
ProgressResponse
struct
{
...
...
@@ -186,11 +197,14 @@ type ProgressResponse struct {
}
type
PushRequest
struct
{
Name
string
`json:"
name
"`
Model
string
`json:"
model
"`
Insecure
bool
`json:"insecure,omitempty"`
Username
string
`json:"username"`
Password
string
`json:"password"`
Stream
*
bool
`json:"stream,omitempty"`
// Name is deprecated, see Model
Name
string
`json:"name"`
}
type
ListResponse
struct
{
...
...
server/routes.go
View file @
a38d88d8
...
...
@@ -414,8 +414,13 @@ func PullModelHandler(c *gin.Context) {
return
}
if
req
.
Name
==
""
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"name is required"
})
var
model
string
if
req
.
Model
!=
""
{
model
=
req
.
Model
}
else
if
req
.
Name
!=
""
{
model
=
req
.
Name
}
else
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"model is required"
})
return
}
...
...
@@ -433,7 +438,7 @@ func PullModelHandler(c *gin.Context) {
ctx
,
cancel
:=
context
.
WithCancel
(
c
.
Request
.
Context
())
defer
cancel
()
if
err
:=
PullModel
(
ctx
,
req
.
Name
,
regOpts
,
fn
);
err
!=
nil
{
if
err
:=
PullModel
(
ctx
,
model
,
regOpts
,
fn
);
err
!=
nil
{
ch
<-
gin
.
H
{
"error"
:
err
.
Error
()}
}
}()
...
...
@@ -458,8 +463,13 @@ func PushModelHandler(c *gin.Context) {
return
}
if
req
.
Name
==
""
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"name is required"
})
var
model
string
if
req
.
Model
!=
""
{
model
=
req
.
Model
}
else
if
req
.
Name
!=
""
{
model
=
req
.
Name
}
else
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"model is required"
})
return
}
...
...
@@ -477,7 +487,7 @@ func PushModelHandler(c *gin.Context) {
ctx
,
cancel
:=
context
.
WithCancel
(
c
.
Request
.
Context
())
defer
cancel
()
if
err
:=
PushModel
(
ctx
,
req
.
Name
,
regOpts
,
fn
);
err
!=
nil
{
if
err
:=
PushModel
(
ctx
,
model
,
regOpts
,
fn
);
err
!=
nil
{
ch
<-
gin
.
H
{
"error"
:
err
.
Error
()}
}
}()
...
...
@@ -502,12 +512,17 @@ func CreateModelHandler(c *gin.Context) {
return
}
if
req
.
Name
==
""
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"name is required"
})
var
model
string
if
req
.
Model
!=
""
{
model
=
req
.
Model
}
else
if
req
.
Name
!=
""
{
model
=
req
.
Name
}
else
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"model is required"
})
return
}
if
err
:=
ParseModelPath
(
req
.
Name
)
.
Validate
();
err
!=
nil
{
if
err
:=
ParseModelPath
(
model
)
.
Validate
();
err
!=
nil
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
err
.
Error
()})
return
}
...
...
@@ -545,7 +560,7 @@ func CreateModelHandler(c *gin.Context) {
ctx
,
cancel
:=
context
.
WithCancel
(
c
.
Request
.
Context
())
defer
cancel
()
if
err
:=
CreateModel
(
ctx
,
req
.
Name
,
filepath
.
Dir
(
req
.
Path
),
commands
,
fn
);
err
!=
nil
{
if
err
:=
CreateModel
(
ctx
,
model
,
filepath
.
Dir
(
req
.
Path
),
commands
,
fn
);
err
!=
nil
{
ch
<-
gin
.
H
{
"error"
:
err
.
Error
()}
}
}()
...
...
@@ -570,14 +585,19 @@ func DeleteModelHandler(c *gin.Context) {
return
}
if
req
.
Name
==
""
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"name is required"
})
var
model
string
if
req
.
Model
!=
""
{
model
=
req
.
Model
}
else
if
req
.
Name
!=
""
{
model
=
req
.
Name
}
else
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"model is required"
})
return
}
if
err
:=
DeleteModel
(
req
.
Name
);
err
!=
nil
{
if
err
:=
DeleteModel
(
model
);
err
!=
nil
{
if
os
.
IsNotExist
(
err
)
{
c
.
JSON
(
http
.
StatusNotFound
,
gin
.
H
{
"error"
:
fmt
.
Sprintf
(
"model '%s' not found"
,
req
.
Name
)})
c
.
JSON
(
http
.
StatusNotFound
,
gin
.
H
{
"error"
:
fmt
.
Sprintf
(
"model '%s' not found"
,
model
)})
}
else
{
c
.
JSON
(
http
.
StatusInternalServerError
,
gin
.
H
{
"error"
:
err
.
Error
()})
}
...
...
@@ -610,21 +630,20 @@ func ShowModelHandler(c *gin.Context) {
return
}
switch
{
case
req
.
Model
==
""
&&
req
.
Name
==
""
:
var
model
string
if
req
.
Model
!=
""
{
model
=
req
.
Model
}
else
if
req
.
Name
!=
""
{
model
=
req
.
Name
}
else
{
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"model is required"
})
return
case
req
.
Model
!=
""
&&
req
.
Name
!=
""
:
c
.
AbortWithStatusJSON
(
http
.
StatusBadRequest
,
gin
.
H
{
"error"
:
"both model and name are set"
})
return
case
req
.
Model
==
""
&&
req
.
Name
!=
""
:
req
.
Model
=
req
.
Name
}
resp
,
err
:=
GetModelInfo
(
req
)
if
err
!=
nil
{
if
os
.
IsNotExist
(
err
)
{
c
.
JSON
(
http
.
StatusNotFound
,
gin
.
H
{
"error"
:
fmt
.
Sprintf
(
"model '%s' not found"
,
req
.
Name
)})
c
.
JSON
(
http
.
StatusNotFound
,
gin
.
H
{
"error"
:
fmt
.
Sprintf
(
"model '%s' not found"
,
model
)})
}
else
{
c
.
JSON
(
http
.
StatusInternalServerError
,
gin
.
H
{
"error"
:
err
.
Error
()})
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment