include modelfile messages

This commit is contained in:
Michael Yang
2024-06-19 14:14:28 -07:00
parent f5e3939220
commit 15af558423
3 changed files with 19 additions and 20 deletions

View File

@@ -70,7 +70,7 @@ type Model struct {
License []string
Digest string
Options map[string]interface{}
Messages []Message
Messages []api.Message
Template *template.Template
}
@@ -191,11 +191,6 @@ func (m *Model) String() string {
return modelfile.String()
}
type Message struct {
Role string `json:"role"`
Content string `json:"content"`
}
type ConfigV2 struct {
ModelFormat string `json:"model_format"`
ModelFamily string `json:"model_family"`