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
3f0b309a
Commit
3f0b309a
authored
Jun 10, 2024
by
Michael Yang
Browse files
remove ManifestV2
parent
e70610ef
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
17 additions
and
22 deletions
+17
-22
server/images.go
server/images.go
+5
-12
server/manifest.go
server/manifest.go
+11
-9
server/manifest_test.go
server/manifest_test.go
+1
-1
No files found.
server/images.go
View file @
3f0b309a
...
...
@@ -135,13 +135,6 @@ type Message struct {
Content
string
`json:"content"`
}
type
ManifestV2
struct
{
SchemaVersion
int
`json:"schemaVersion"`
MediaType
string
`json:"mediaType"`
Config
*
Layer
`json:"config"`
Layers
[]
*
Layer
`json:"layers"`
}
type
ConfigV2
struct
{
ModelFormat
string
`json:"model_format"`
ModelFamily
string
`json:"model_family"`
...
...
@@ -160,7 +153,7 @@ type RootFS struct {
DiffIDs
[]
string
`json:"diff_ids"`
}
func
GetManifest
(
mp
ModelPath
)
(
*
Manifest
V2
,
string
,
error
)
{
func
GetManifest
(
mp
ModelPath
)
(
*
Manifest
,
string
,
error
)
{
fp
,
err
:=
mp
.
GetManifestPath
()
if
err
!=
nil
{
return
nil
,
""
,
err
...
...
@@ -170,7 +163,7 @@ func GetManifest(mp ModelPath) (*ManifestV2, string, error) {
return
nil
,
""
,
err
}
var
manifest
*
Manifest
V2
var
manifest
*
Manifest
bts
,
err
:=
os
.
ReadFile
(
fp
)
if
err
!=
nil
{
...
...
@@ -822,7 +815,7 @@ func PushModel(ctx context.Context, name string, regOpts *registryOptions, fn fu
func
PullModel
(
ctx
context
.
Context
,
name
string
,
regOpts
*
registryOptions
,
fn
func
(
api
.
ProgressResponse
))
error
{
mp
:=
ParseModelPath
(
name
)
var
manifest
*
Manifest
V2
var
manifest
*
Manifest
var
err
error
var
noprune
string
...
...
@@ -929,7 +922,7 @@ func PullModel(ctx context.Context, name string, regOpts *registryOptions, fn fu
return
nil
}
func
pullModelManifest
(
ctx
context
.
Context
,
mp
ModelPath
,
regOpts
*
registryOptions
)
(
*
Manifest
V2
,
error
)
{
func
pullModelManifest
(
ctx
context
.
Context
,
mp
ModelPath
,
regOpts
*
registryOptions
)
(
*
Manifest
,
error
)
{
requestURL
:=
mp
.
BaseURL
()
.
JoinPath
(
"v2"
,
mp
.
GetNamespaceRepository
(),
"manifests"
,
mp
.
Tag
)
headers
:=
make
(
http
.
Header
)
...
...
@@ -940,7 +933,7 @@ func pullModelManifest(ctx context.Context, mp ModelPath, regOpts *registryOptio
}
defer
resp
.
Body
.
Close
()
var
m
*
Manifest
V2
var
m
*
Manifest
if
err
:=
json
.
NewDecoder
(
resp
.
Body
)
.
Decode
(
&
m
);
err
!=
nil
{
return
nil
,
err
}
...
...
server/manifest.go
View file @
3f0b309a
...
...
@@ -14,7 +14,10 @@ import (
)
type
Manifest
struct
{
ManifestV2
SchemaVersion
int
`json:"schemaVersion"`
MediaType
string
`json:"mediaType"`
Config
*
Layer
`json:"config"`
Layers
[]
*
Layer
`json:"layers"`
filepath
string
fi
os
.
FileInfo
...
...
@@ -66,7 +69,7 @@ func ParseNamedManifest(n model.Name) (*Manifest, error) {
p
:=
filepath
.
Join
(
manifests
,
n
.
Filepath
())
var
m
Manifest
V2
var
m
Manifest
f
,
err
:=
os
.
Open
(
p
)
if
err
!=
nil
{
return
nil
,
err
...
...
@@ -83,12 +86,11 @@ func ParseNamedManifest(n model.Name) (*Manifest, error) {
return
nil
,
err
}
return
&
Manifest
{
ManifestV2
:
m
,
filepath
:
p
,
fi
:
fi
,
digest
:
fmt
.
Sprintf
(
"%x"
,
sha256sum
.
Sum
(
nil
)),
},
nil
m
.
filepath
=
p
m
.
fi
=
fi
m
.
digest
=
fmt
.
Sprintf
(
"%x"
,
sha256sum
.
Sum
(
nil
))
return
&
m
,
nil
}
func
WriteManifest
(
name
model
.
Name
,
config
*
Layer
,
layers
[]
*
Layer
)
error
{
...
...
@@ -108,7 +110,7 @@ func WriteManifest(name model.Name, config *Layer, layers []*Layer) error {
}
defer
f
.
Close
()
m
:=
Manifest
V2
{
m
:=
Manifest
{
SchemaVersion
:
2
,
MediaType
:
"application/vnd.docker.distribution.manifest.v2+json"
,
Config
:
config
,
...
...
server/manifest_test.go
View file @
3f0b309a
...
...
@@ -25,7 +25,7 @@ func createManifest(t *testing.T, path, name string) {
}
defer
f
.
Close
()
if
err
:=
json
.
NewEncoder
(
f
)
.
Encode
(
Manifest
V2
{});
err
!=
nil
{
if
err
:=
json
.
NewEncoder
(
f
)
.
Encode
(
Manifest
{});
err
!=
nil
{
t
.
Fatal
(
err
)
}
}
...
...
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